Daily Arxiv

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

Training Large Language Models To Reason In Parallel With Global Forking Tokens

Created by
  • Haebom

저자

Sheng Jia, Xiao Wang, Shiva Prasad Kasiviswanathan

개요

LLM의 병렬 추론 성능 향상을 위해 다양한 추론 경로를 생성하는 것이 중요하지만, 복잡한 문제에서는 정확성과 다양성 사이의 trade-off가 발생합니다. 본 논문에서는 병렬 추론을 set-of-next-token-prediction 문제로 간주하고, 전역 분기 토큰과 고유한 추론 경로 간의 자기 지도 이분 매칭을 사용하여 Supervised Fine-Tuning (SFT)에 set-based global loss를 통합하는 Set Supervised Fine-Tuning (SSFT) 방법을 제안합니다. SSFT는 고유한 추론 모드를 유지하고, emergent global forking tokens을 생성하며, 여러 추론 벤치마크에서 SFT보다 일관되게 높은 성능을 보입니다.

시사점, 한계점

시사점:
병렬 추론을 위한 LLM의 성능 향상을 위한 새로운 fine-tuning 기법 제시.
다양성과 정확성 간의 trade-off 문제를 해결하기 위한 set-based global loss 도입.
SSFT 기법을 통해 SFT 대비 향상된 성능 입증.
한계점:
구체적인 한계점은 논문 요약에서 명시되지 않음. (논문 원문을 참고해야 함)
👍