Daily Arxiv

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

An Efficient and Precise Training Data Construction Framework for Process-supervised Reward Model in Mathematical Reasoning

Created by
  • Haebom

저자

Wei Sun, Qianlong Du, Fuwei Cui, Jiajun Zhang

개요

본 논문은 대규모 언어 모델(LLM)의 수학적 추론 능력 향상을 위한 새로운 프레임워크인 EpicPRM을 제안합니다. 기존의 프로세스 감독 보상 모델(PRM) 학습을 위한 데이터 생성 방법은 수동 주석이나 단계별 몬테카를로 추정 등으로 비용이 많이 들거나 품질이 낮다는 한계를 가지고 있습니다. EpicPRM은 각 중간 추론 단계의 기여도를 정량화하여 주석을 달고, 적응형 이진 탐색 알고리즘을 사용하여 주석의 정확성과 효율성을 높입니다. 이를 통해 5만 개의 주석이 달린 중간 단계로 구성된 고품질 프로세스 감독 학습 데이터셋 Epic50k를 효율적으로 구축합니다. Epic50k로 학습된 PRM은 기존 공개 데이터셋을 사용한 것보다 훨씬 우수한 성능을 보입니다. Epic50k는 GitHub에서 공개됩니다.

시사점, 한계점

시사점:
LLM의 수학적 추론 능력 향상에 효과적인 새로운 프레임워크(EpicPRM) 제시
고품질의 프로세스 감독 학습 데이터셋(Epic50k) 구축 및 공개
기존 방법보다 효율적이고 정확한 주석 생성 방법 제시
Epic50k를 사용한 PRM이 기존 방법보다 우수한 성능을 보임
한계점:
EpicPRM의 성능이 다른 유형의 추론 문제 또는 다른 LLM 아키텍처에 대해서도 일반화되는지 추가적인 연구 필요
Epic50k 데이터셋의 크기가 더 큰 데이터셋에 비해 상대적으로 작을 수 있음
적응형 이진 탐색 알고리즘의 최적 매개변수 설정에 대한 추가적인 연구 필요
👍