Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

AMFT: Aligning LLM Reasoners by Meta-Learning the Optimal Imitation-Exploration Balance

Created by
  • Haebom

저자

Lixuan He, Jie Feng, Yong Li

개요

대규모 언어 모델(LLM)의 추론 능력 향상을 위해 일반적으로 사용되는 지도 미세 조정(SFT)과 강화 학습(RL)의 2단계 파이프라인은 문제점을 지닌다. 이 논문은 SFT와 RL을 별개의 방법론이 아닌 보완적인 보상 신호로 보고, SFT의 암묵적 보상과 RL의 명시적 보상 사이의 최적 균형을 학습하는 단일 단계 알고리즘인 Adaptive Meta Fine-Tuning (AMFT)를 제안한다. AMFT는 메타 기울기 적응 가중치 제어기를 사용하여 SFT-RL 균형을 학습 가능한 파라미터로 취급하며, 장기적인 작업 성능을 극대화하도록 동적으로 최적화한다. 수학적 추론, 추상적 시각적 추론, 시각 언어 내비게이션 벤치마크에서 새로운 SOTA를 달성했으며, OOD 작업에서도 우수한 일반화 성능을 보였다.

시사점, 한계점

시사점:
SFT와 RL의 균형을 학습 가능한 파라미터로 취급하여 효과적인 단일 단계 훈련 가능성을 제시함.
다양한 추론 벤치마크에서 SOTA 달성 및 OOD 일반화 성능 개선.
메타 학습 제어기의 중요성을 입증하여 LLM 정렬의 안정성, 샘플 효율성 및 성능 향상.
한계점:
논문에 명시된 한계점은 없음.
👍