Daily Arxiv

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

ETTRL: Balancing Exploration and Exploitation in LLM Test-Time Reinforcement Learning Via Entropy Mechanism

Created by
  • Haebom

저자

Jia Liu, ChangYi He, YingQiao Lin, MingMin Yang, FeiYang Shen, ShaoGuo Liu

개요

본 논문은 대규모 언어 모델(LLM)의 복잡한 추론 능력 향상을 위해 테스트 시간 강화 학습(TTRL)을 활용하는 방법을 제시합니다. 기존 TTRL의 높은 추론 비용과 과신 문제를 해결하기 위해, 엔트로피 기반 메커니즘을 도입하여 탐험-활용 균형을 개선하는 두 가지 전략, 즉 엔트로피 분기 트리 다수결 롤아웃(ETMR)과 엔트로피 기반 이점 재구성(EAR)을 제안합니다. Llama3.1-8B 모델에 적용한 결과, AIME 2024 벤치마크에서 Pass at 1 지표를 68% 향상시키면서 롤아웃 토큰 예산은 60%만 사용하는 효율성을 보였습니다. 이는 추론 효율성, 다양성, 추정 강건성 간의 균형을 효과적으로 최적화하는 방법임을 보여줍니다.

시사점, 한계점

시사점:
TTRL의 효율성과 성능을 향상시키는 새로운 엔트로피 기반 메커니즘 제시.
AIME 2024 벤치마크에서 상당한 성능 향상 달성 (Pass at 1 지표 68% 향상).
추론 비용 감소 (롤아웃 토큰 예산 60% 절감).
엔트로피 기반 전략을 통해 탐험-활용 균형 개선 및 과신 문제 완화.
개방형 도메인 추론 작업을 위한 비지도 강화 학습 발전에 기여.
한계점:
제안된 방법의 일반화 성능에 대한 추가적인 실험 필요.
다른 LLM 및 벤치마크에 대한 적용성 검증 필요.
엔트로피 기반 메커니즘의 최적 매개변수 설정에 대한 연구 필요.
AIME 2024 벤치마크에 특화된 성능 향상일 가능성. 다른 벤치마크에서도 동일한 효과를 보이는지 검증 필요.
👍