Daily Arxiv

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

Direct Reasoning Optimization: LLMs Can Reward And Refine Their Own Reasoning for Open-Ended Tasks

Created by
  • Haebom

저자

Yifei Xu, Tusher Chakraborty, Srinagesh Sharma, Leonardo Nunes, Emre K{\i}c{\i}man, Songwu Lu, Ranveer Chandra

개요

본 논문은 대규모 언어 모델(LLM)의 개방형 장문 추론 과제에 대한 강화 학습 프레임워크인 직접 추론 최적화(DRO)를 제안합니다. 기존의 검증 가능한 보상 신호 부재 문제를 해결하기 위해, 모델의 이전 사고 과정 추론의 영향을 반영하는 주요 토큰을 선택적으로 식별하고 강조하는 새로운 보상 신호인 추론 반영 보상(R3)을 사용합니다. R3는 최적화되는 동일한 모델을 내부적으로 사용하여 계산되므로 완전히 자체 포함된 학습 설정이 가능합니다. 또한, 개방형 추론 과제에 대한 R3 기반의 동적 데이터 필터링 전략을 도입하여 비용을 절감하면서 성능을 향상시킵니다. ParaRev(장문 단락 수정 과제)와 FinQA(수학 중심 QA 벤치마크) 두 가지 데이터셋에서 DRO가 강력한 기준 모델보다 성능이 우수함을 보여줍니다.

시사점, 한계점

시사점:
개방형 장문 추론 과제에 대한 새로운 강화 학습 프레임워크인 DRO 제시
R3 보상 신호를 통해 모델의 추론 과정과 결과의 일관성을 미세한 수준에서 포착
자체 포함된 학습 설정으로 효율적인 학습 가능
동적 데이터 필터링 전략을 통해 비용 절감 및 성능 향상
개방형 및 구조화된 영역 모두에서 광범위하게 적용 가능성을 보임
한계점:
R3 보상 신호의 설계 및 성능에 대한 추가적인 분석 필요
다양한 유형의 개방형 추론 과제에 대한 일반화 성능 평가 필요
다른 강화 학습 기법과의 비교 분석 필요
R3 기반 동적 데이터 필터링 전략의 최적화 파라미터에 대한 추가 연구 필요
👍