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))의 한계를 극복하기 위해, SFT와 RL을 상호 보완적인 보상 신호로 보는 새로운 관점을 제시합니다. 기존 방법들의 단점인 끔찍한 망각(catastrophic forgetting)과 모방과 탐색 간의 최적화되지 않은 절충을 해결하기 위해, 암시적 보상(implicit rewards) 개념을 도입하여 SFT의 경로 수준 보상과 RL의 결과 기반 보상 간의 최적 균형을 학습하는 단일 단계 알고리즘인 적응형 메타 미세 조정(AMFT)을 제안합니다. AMFT의 핵심은 SFT-RL 균형을 학습 가능한 매개변수로 취급하여 장기적인 작업 성능을 극대화하도록 동적으로 최적화하는 메타-기울기 적응 가중치 제어기입니다. 정책 엔트로피를 사용한 안정성 확보를 통해 효과적인 학습 과정을 자율적으로 발견합니다. 수학적 추론, 추상적 시각적 추론(General Points), 시각-언어 탐색(V-IRL) 등 다양한 벤치마크에서 최첨단 성능을 달성하고, 분포 외(OOD) 작업에서도 우수한 일반화 성능을 보입니다. 에이블레이션 연구와 학습 동적 분석을 통해 메타 학습 제어기가 AMFT의 안정성, 샘플 효율성 및 성능에 중요한 역할을 함을 확인합니다.

시사점, 한계점

시사점:
기존의 SFT와 RL의 두 단계 파이프라인 방식의 한계를 극복하는 새로운 단일 단계 학습 알고리즘 AMFT 제시.
암시적 보상 개념을 도입하여 SFT와 RL의 보상 신호를 효과적으로 통합.
메타-기울기 적응 가중치 제어기를 통해 SFT-RL 균형을 동적으로 최적화하여 장기적인 작업 성능 향상.
다양한 벤치마크에서 최첨단 성능 달성 및 우수한 일반화 성능 확인.
오픈소스 코드 공개를 통한 연구의 재현성 및 확장성 제공.
한계점:
AMFT 알고리즘의 복잡성으로 인한 계산 비용 증가 가능성.
특정 벤치마크에 대한 최적화 가능성, 다른 유형의 작업에 대한 일반화 성능 추가 검증 필요.
메타-기울기 적응 가중치 제어기의 동작에 대한 더 깊이 있는 분석 및 해석 필요.
👍