Daily Arxiv

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

DisCO: Reinforcing Large Reasoning Models with Discriminative Constrained Optimization

Created by
  • Haebom
Category
Empty

저자

Gang Li, Ming Lin, Tomer Galanti, Zhengzhong Tu, Tianbao Yang

개요

본 논문은 대규모 추론 모델(LRMs)을 위한 강화 학습 방법인 Group Relative Policy Optimization (GRPO)의 한계를 분석하고, 이를 극복하기 위한 Discriminative Constrained Optimization (DisCO) 프레임워크를 제안합니다. 특히, GRPO가 질문 수준 난이도 편향을 가지고 있으며, DisCO는 차별적 학습 원리에 기반하여 이 문제를 해결합니다. DisCO는 그룹 상대 목표 대신 차별적 목표를 사용하고, 클리핑 기반 대리 목표 대신 비클리핑 RL 대리 목표를 사용하며, KL 발산 제약을 적용하기 위해 제약 최적화 방식을 활용합니다. 실험 결과, DisCO는 GRPO 및 DAPO와 같은 기존 방법보다 성능이 향상되었습니다.

시사점, 한계점

시사점:
DisCO는 GRPO의 질문 수준 난이도 편향 문제를 해결하여 모델 성능을 향상시켰습니다.
비클리핑 목표 함수와 제약 최적화를 통해 훈련의 안정성을 확보했습니다.
차별적 학습 기법을 활용하여 데이터 불균형 문제를 해결할 수 있는 가능성을 제시했습니다.
수학적 추론 능력을 향상시키는 데 기여했습니다.
한계점:
1.5B 모델에 대한 실험 결과만 제시되어, 다른 모델 크기에 대한 일반화 가능성은 추가적인 연구가 필요합니다.
DisCO의 효과가 특정 수학적 추론 task에 국한될 수 있으며, 다른 종류의 task에 대한 적용 가능성 검토가 필요합니다.
구체적인 구현 세부 사항과 하이퍼파라미터 설정에 대한 정보가 부족합니다.
👍