Daily Arxiv

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

DisCO: Reinforcing Large Reasoning Models with Discriminative Constrained Optimization

Created by
  • Haebom

저자

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

개요

본 논문은 대규모 추론 모델(LRM)을 위한 강화 학습 방법인 GRPO(Group Relative Policy Optimization)의 한계점을 분석하고, 새로운 강화 학습 프레임워크인 DisCO(Discriminative Constrained Optimization)를 제시합니다. GRPO는 질문 난이도 편향이라는 고질적인 문제를 가지고 있는데, DisCO는 차별적 학습 원리를 기반으로 이러한 문제를 해결합니다. DisCO는 그룹 상대적 목적 함수 대신 점수 함수로 정의된 차별적 목적 함수를 사용하고, 클리핑 기반 대리 목적 함수 대신 비클리핑 RL 대리 목적 함수를 사용하며, KL divergence 제약 조건을 강화하는 간단하면서도 효과적인 제약 최적화 방식을 채택합니다. 실험 결과, DisCO는 수학적 추론 능력 향상에 있어 GRPO 및 DAPO(GRPO의 개선된 변형)보다 6~7%의 성능 향상을 보였습니다.

시사점, 한계점

시사점:
GRPO의 질문 난이도 편향 문제를 효과적으로 해결하는 새로운 강화 학습 프레임워크 DisCO를 제시했습니다.
DisCO는 GRPO 및 DAPO보다 우수한 성능을 보이며, 대규모 추론 모델의 성능 향상에 기여할 수 있습니다.
비클리핑 RL 대리 목적 함수와 제약 최적화 방식을 통해 GRPO의 엔트로피 불안정성 문제를 해결했습니다.
차별적 학습 기법을 활용하여 데이터 불균형 문제를 해결할 수 있습니다.
한계점:
본 논문의 실험은 특정 크기(1.5B)의 모델과 수학적 추론 과제에 국한되어 있습니다. 다른 크기의 모델이나 다른 종류의 과제에 대한 일반화 가능성은 추가 연구가 필요합니다.
DisCO의 성능 향상이 특정 데이터셋이나 모델 아키텍처에 의존할 가능성이 있습니다. 다양한 데이터셋과 모델 아키텍처에 대한 추가 실험이 필요합니다.
👍