Daily Arxiv

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

CPPO: Accelerating the Training of Group Relative Policy Optimization-Based Reasoning Models

Created by
  • Haebom
Category
Empty

저자

Zhihang Lin, Mingbao Lin, Yuan Xie, Rongrong Ji

개요

본 논문은 Group Relative Policy Optimization (GRPO) 기반 추론 모델의 훈련 속도를 높이기 위해 Completion Pruning Policy Optimization (CPPO)를 제안합니다. GRPO는 효과적이지만 각 질문에 대해 여러 완료를 샘플링해야 하므로 높은 훈련 비용이 발생합니다. CPPO는 낮은 절대 이점을 가진 완료를 제거하여 gradient 계산 및 업데이트에 필요한 완료 수를 줄이고, 동적 완료 할당 전략을 도입하여 GPU 활용도를 극대화합니다. CPPO는 GSM8K에서 최대 $7.98\times$, Math에서 $3.48\times$ 속도 향상을 달성하면서 정확도를 유지하거나 향상시킵니다.

시사점, 한계점

시사점:
CPPO는 GRPO 기반 추론 모델의 훈련 속도를 크게 향상시킵니다.
CPPO는 정확도를 유지하거나 향상시키면서 훈련 비용을 절감합니다.
동적 완료 할당 전략을 통해 GPU 활용도를 개선합니다.
한계점:
논문에서 구체적인 한계점은 명시되지 않았습니다.
👍