Daily Arxiv

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

GRPO is Secretly a Process Reward Model

Created by
  • Haebom

저자

Michael Sullivan

개요

본 논문은 GRPO RL 알고리즘이 토큰 시퀀스 간의 특정 가정 하에서 비자명한 과정 보상 모델(PRM)을 유도한다는 것을 이론적으로 증명한다. 실제 조건에서 이러한 가정이 충족됨을 실증적으로 보이고, GRPO 목적 함수의 결함을 파악하여 탐험과 활용을 방해하는 비균등하게 분포된 과정 단계를 지적한다. 이를 해결하기 위해 $\lambda$-GRPO를 제안하고, $\lambda$-GRPO로 훈련된 LLM이 표준 GRPO로 훈련된 LLM보다 높은 검증 정확도와 다운스트림 추론 작업 성능을 달성하며, 최고 성능에 더 빠르게 도달함을 보여준다. 연구 결과는 비용이 많이 드는 명시적으로 정의된 PRM의 이점에 의문을 제기하며, 바닐라 GRPO 알고리즘 내의 숨겨진 PRM 구조를 활용하여 모델 성능을 향상시킬 수 있음을 보여준다.

시사점, 한계점

GRPO 알고리즘 내의 숨겨진 PRM 구조를 활용하여 모델 성능 향상 가능성을 제시한다.
$\lambda$-GRPO를 통해 GRPO 알고리즘의 성능을 개선하고, 더 빠른 학습 속도를 달성했다.
비용이 많이 드는 명시적 PRM의 필요성에 대한 의문을 제기한다.
논문에서 제시된 가정은 특정 조건에 국한될 수 있으며, 일반적인 경우에도 적용될 수 있는지 추가 연구가 필요하다.
$\lambda$-GRPO의 개선 효과가 다른 종류의 모델이나 작업에도 적용될 수 있는지 확인해야 한다.
👍