Daily Arxiv

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

Unearthing Gems from Stones: Policy Optimization with Negative Sample Augmentation for LLM Reasoning

Created by
  • Haebom

저자

Zhaohui Yang, Shilei Jiang, Chen Hu, Linjing Li, Shihong Deng, Daxin Jiang

개요

본 논문은 장문의 Chain of Thought(CoT) 모델에서 계산 비용을 최소화하기 위해 기존 학습 데이터의 활용도를 극대화하는 방법을 제시합니다. 기존 방법들이 부정적인 응답을 무시하거나 모든 토큰에 동일한 페널티를 적용하는 한계를 지적하며, 부정적인 응답에도 자기 반성 및 오류 수정 단계와 같은 유용한 정보가 포함되어 있음을 분석합니다. 이를 해결하기 위해, 부정적 샘플 증강을 포함하는 행동 제약 정책 경사(BCPG-NSA)라는 새로운 오프라인 강화 학습 프레임워크를 제안합니다. BCPG-NSA는 샘플 분할, LLM 및 PRM 판정자를 결합한 합의 기반 단계 정확도 평가, 부정적 샘플 내 긍정적 단계를 효과적으로 추출하도록 설계된 NSA를 이용한 정책 최적화의 세 단계로 구성됩니다. 실험 결과, BCPG-NSA는 여러 어려운 수학/코딩 추론 벤치마크에서 기존 방법보다 성능이 우수하며, 샘플 효율성이 향상되고 반복 횟수가 증가해도 강건성과 확장성이 유지됨을 보여줍니다.

시사점, 한계점

시사점:
장문 CoT 모델의 샘플 효율성을 향상시키는 새로운 방법 제시.
부정적 응답에서 유용한 학습 신호를 추출하는 효과적인 전략 제시.
LLM과 PRM을 결합한 합의 기반 평가 방식의 효용성 입증.
여러 어려운 벤치마크에서 기존 방법 대비 성능 향상 확인.
반복 횟수 증가에도 강건하고 확장 가능한 모델 제시.
한계점:
제안된 방법의 일반화 성능에 대한 추가적인 연구 필요.
다양한 유형의 문제에 대한 적용성 검증 필요.
LLM과 PRM 판정자의 신뢰도 및 한계에 대한 고찰 필요.
계산 비용 및 시간 복잡도에 대한 자세한 분석 필요.
👍