Daily Arxiv

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

ThinkTuning: Instilling Cognitive Reflections without Distillation

Created by
  • Haebom

저자

Aswin RRV, Jacob Dineen, Divij Handa, Md Nayem Uddin, Mihir Parmar, Chitta Baral, Ben Zhou

개요

본 논문은 강화 학습(RL)만으로는 사고 능력을 갖춘 대규모 언어 모델(LLM)을 만들 수 없다는 기존 연구 결과를 바탕으로, 사고 능력이 부족한 모델을 학습시키는 새로운 방법인 ThinkTuning을 제안합니다. ThinkTuning은 교사 모델의 지도를 받는 학생 모델의 rollout을 강화하는 GRPO 기반의 상호작용적 학습 접근 방식입니다. 교사 모델은 문제를 제시하고 학생 모델의 답변에 대한 수정 피드백을 제공하며, 이를 통해 학생 모델의 추론 능력을 향상시킵니다. 실험 결과, ThinkTuning은 다양한 벤치마크에서 zero-shot baseline에 비해 평균 3.85% 향상을 보였으며, MATH-500, AIME, GPQA-Diamond에서는 각각 2.08%, 2.23%, 3.99% 향상을 보였습니다. 소스 코드는 깃허브에 공개되어 있습니다.

시사점, 한계점

시사점:
교사-학생 모델 상호작용 기반의 상호작용적 학습 방식을 통해 LLM의 추론 능력 향상 가능성을 제시.
GRPO와 교사 모델의 피드백을 결합하여 효과적인 사고 능력 학습 방법을 제안.
제한된 사고 능력을 가진 모델에서도 사고 능력을 향상시킬 수 있음을 실험적으로 증명.
다양한 벤치마크에서 성능 향상을 보여 실용적인 효과를 확인.
한계점:
교사 모델의 질에 따라 학습 성능이 크게 좌우될 가능성 존재.
제안된 방법의 효과가 특정 유형의 문제 또는 모델에 국한될 가능성 존재.
더욱 다양하고 복잡한 문제에 대한 성능 평가가 필요.
계산 비용 및 학습 시간에 대한 고려 필요.
👍