Daily Arxiv

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

Uncertainty-Based Methods for Automated Process Reward Data Construction and Output Aggregation in Mathematical Reasoning

Created by
  • Haebom

저자

Jiuzhou Han, Wray Buntine, Ehsan Shareghi

개요

본 논문은 대규모 언어 모델의 다단계 수학적 추론 과정에서 발생하는 오류를 해결하기 위해, 각 중간 단계를 감독하고 평가하는 Process-level Reward Models (PRMs)에 초점을 맞추고 있다. 기존 PRMs 학습을 위한 고품질 보상 데이터 생성의 어려움을 해결하고자, 불확실성 기반의 자동화된 PRMs 보상 데이터 생성 및 주석 프레임워크를 제안한다. 또한, 다수결 투표와 PRMs의 한계를 지적하고, 두 방법의 장점을 결합한 Hybrid Majority Reward Vote와 Weighted Reward Frequency Vote라는 두 가지 불확실성 인식 출력 집계 방법을 제시한다. ProcessBench, MATH, GSMPlus 데이터셋을 이용한 실험을 통해 제안된 프레임워크와 출력 집계 방법의 효과성과 효율성을 검증한다. 소스 코드와 데이터는 공개적으로 제공될 예정이다.

시사점, 한계점

시사점:
불확실성 기반의 자동화된 PRM 보상 데이터 생성 프레임워크를 통해 고품질 데이터 생성의 효율성을 높였다.
다수결 투표와 PRMs의 한계를 극복하는 새로운 출력 집계 방법(Hybrid Majority Reward Vote, Weighted Reward Frequency Vote)을 제시하였다.
다양한 PRMs에서 수학적 추론 능력 향상을 실험적으로 입증하였다.
제안된 방법의 소스 코드와 데이터를 공개하여 재현성과 확장성을 확보하였다.
한계점:
제안된 프레임워크의 성능은 사용된 데이터셋에 의존적일 수 있다.
새로운 출력 집계 방법의 일반화 성능에 대한 추가적인 연구가 필요하다.
특정 유형의 수학 문제에 대한 편향이 존재할 가능성이 있다.
👍