Daily Arxiv

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

Squeeze the Soaked Sponge: Efficient Off-policy Reinforcement Finetuning for Large Language Model

Created by
  • Haebom

저자

Jing Liang, Hongyao Tang, Yi Ma, Jinyi Liu, Yan Zheng, Shuyue Hu, Lei Bai, Jianye Hao

개요

본 논문은 대규모 언어 모델(LLM)의 추론 능력 향상을 위한 강화 학습(RL)의 한계점을 해결하고자, 기존의 온-폴리시 강화 학습 기법들의 비효율적인 컴퓨팅 비용 문제를 해결하는 오프-폴리시 강화 학습 기법인 ReMix를 제안한다. ReMix는 PPO 및 GRPO와 같은 온-폴리시 RFT 방법을 활용하여 오프-폴리시 데이터를 활용할 수 있도록 설계되었으며, Mix-policy proximal policy gradient, KL-Convex policy constraint, Policy reincarnation의 세 가지 주요 구성 요소로 이루어져 있다. 실험 결과, ReMix는 다양한 수학 추론 벤치마크에서 최첨단 성능을 달성하면서 기존 방법 대비 30배에서 450배까지 훈련 비용을 절감하는 것을 보여준다. 또한, 오프-폴리시 불일치의 Whipping Effect로 인한 짧은 응답 선호 현상, 심각한 오프-폴리시 상황에서의 자기 반성 행동 붕괴 모드 등의 통찰력 있는 분석 결과도 제시한다.

시사점, 한계점

시사점:
기존 온-폴리시 강화 학습 기반의 LLM 추론 능력 향상 방법의 비효율적인 컴퓨팅 비용 문제를 효과적으로 해결하는 새로운 오프-폴리시 기법 ReMix 제시.
다양한 수학 추론 벤치마크에서 최첨단(SOTA) 성능 달성 및 훈련 비용(rollout 데이터 양)의 극적인 감소(30배~450배).
오프-폴리시 강화 학습의 효율성과 안정성을 높이는 핵심 기술(Mix-policy proximal policy gradient, KL-Convex policy constraint, Policy reincarnation) 제시.
오프-폴리시 학습 과정에서 발생하는 현상에 대한 심층적인 분석 및 통찰 제공 (Whipping Effect, 자기 반성 행동 붕괴 등).
한계점:
ReMix의 성능 향상이 특정 수학 추론 벤치마크에 국한될 가능성. 다른 종류의 작업에 대한 일반화 성능 평가 필요.
오프-폴리시 데이터 활용에 따른 편향 및 안정성 문제에 대한 추가적인 연구 필요.
제시된 분석 결과의 일반성 및 범용성에 대한 추가적인 검증 필요.
👍