Daily Arxiv

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

Nested-ReFT: Efficient Reinforcement Learning for Large Language Model Fine-Tuning via Off-Policy Rollouts

Created by
  • Haebom
Category
Empty

저자

Maxime Heuillet, Yufei Cui, Boxing Chen, Audrey Durand, Prasanna Parthasarathi

개요

본 논문은 수학적 추론과 같은 어려운 분야에서 LLM의 고급 추론 능력을 향상시키기 위해 검증 가능한 보상을 기반으로 하는 강화 미세 조정(ReFT)을 활용하는 새로운 프레임워크인 Nested-ReFT를 제안합니다. 표준 ReFT 프레임워크의 높은 계산 비용 문제를 해결하기 위해, Nested-ReFT는 대상 모델의 일부 레이어를 행동 모델로 사용하여 훈련 중 오프-정책 완성을 생성합니다. 동적 레이어 스킵을 통해 추론 비용을 감소시키고, 이론적 분석을 통해 무편향 기울기 추정을 보장하며, 경험적 분석을 통해 계산 효율성을 향상시킵니다. 또한, 오프-정책성을 최소화하기 위한 세 가지 편향 완화 방안을 제시하여 ReFT 성능을 유지합니다.

시사점, 한계점

시사점:
ReFT 훈련 시 계산 비용을 줄이는 새로운 프레임워크 제시.
오프-정책 학습을 활용하여 효율성 향상.
다양한 수학적 추론 벤치마크에서 계산 효율성 향상 입증.
편향 완화 기술을 통해 성능 유지.
한계점:
아직 구체적인 편향 완화 기술의 상세 내용 및 효과에 대한 정보 부족.
ReFT와 Nested-ReFT의 성능 차이에 대한 구체적인 분석 미흡 가능성.
다른 분야로의 확장성에 대한 추가적인 연구 필요.
👍