Daily Arxiv

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

Not All Thoughts are Generated Equal: Efficient LLM Reasoning via Multi-Turn Reinforcement Learning

Created by
  • Haebom

저자

Yansong Ning, Wei Li, Jun Fang, Naiqiang Tan, Hao Liu

개요

본 논문은 대규모 언어 모델(LLM)의 추론 효율을 향상시키기 위해 긴 사고 과정(CoT)을 압축하는 새로운 전략을 제시합니다. 기존 연구들이 긴 CoT 내 모든 사고를 동일하게 압축하는 것과 달리, 본 연구는 각 사고의 중요도를 분석하여 효과적이고 간결한 추론을 가능하게 합니다. 자동 긴 CoT 청킹과 몬테카를로 롤아웃을 통해 각 사고의 효과성과 효율성을 분석하고, 이를 바탕으로 효과성과 효율성을 종합적으로 측정하는 이론적 경계가 있는 지표를 제안합니다. 이후, 긴 사고를 효과적으로 생성하는 LLM과 나머지 사고를 효율적으로 생성하는 LLM이 공동으로 문제를 해결하는 효율적인 추론 프레임워크인 Long$\otimes$Short를 제안합니다. 각각의 LLM을 위한 소량의 콜드 스타트 데이터를 합성하여 미세 조정하고, 모델 자체 진화와 두 LLM 간의 협업에 중점을 둔 시너지 지향적 다회차 강화 학습을 제안합니다. 실험 결과, 제안된 방법은 MATH500, AIME24/25, AMC23 및 GPQA Diamond 벤치마크에서 토큰 길이를 80% 이상 줄이면서 DeepSeek-R1-Distill-Qwen-7B 및 DeepSeek-R1-Distill-Llama-8B와 비교할 만한 성능을 달성함을 보여줍니다.

시사점, 한계점

시사점:
긴 CoT의 효율적인 압축을 위한 새로운 프레임워크인 Long$\otimes$Short 제안.
각 사고의 중요도에 따른 차별적 압축을 통해 추론 효율 향상.
소량의 데이터를 활용한 LLM 미세 조정 및 시너지 지향적 강화 학습을 통한 성능 개선.
MATH500, AIME24/25, AMC23, GPQA Diamond 벤치마크에서 기존 모델과 비교하여 우수한 성능 및 토큰 길이 감소 효과 검증.
한계점:
제안된 지표의 이론적 경계 및 일반화 가능성에 대한 추가 연구 필요.
콜드 스타트 데이터 생성 과정의 효율성 및 일반화 성능 개선 필요.
다양한 LLM 및 벤치마크에 대한 추가 실험 필요.
Long$\otimes$Short 프레임워크의 확장성 및 적용 가능성에 대한 추가 연구 필요.
👍