Daily Arxiv

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

CARFT: Boosting LLM Reasoning via Contrastive Learning with Annotated Chain-of-Thought-based Reinforced Fine-Tuning

Created by
  • Haebom

저자

Wenqiao Zhu, Ji Liu, Rongjuncheng Zhang, Haipang Wu, Yulun Zhang

개요

본 논문은 대규모 언어 모델(LLM)의 추론 능력 향상을 위한 새로운 강화 학습 기반 미세 조정 방법인 Contrastive learning with annotated CoT-based Reinforced Fine-Tuning (\TheName{})을 제안합니다. 기존 RL 기반 방법의 한계점인 불안정한 추론 경로 샘플링과 주석이 달린 사고 과정(CoT)의 무시, 그리고 기존 SFT 접근 방식의 과도한 CoT 강조 문제를 해결하기 위해, 각 CoT에 대한 표현을 학습하고 새로운 대조 신호를 설계하여 미세 조정 과정을 안내합니다. \TheName{}는 주석이 달린 CoT를 완전히 활용하는 동시에 비지도 학습 신호를 통합하여 미세 조정 과정을 안정화시킵니다. 실험 결과, 세 가지 기준 방법, 두 가지 기본 모델, 두 가지 데이터 세트를 사용한 실험을 통해 \TheName{}의 강건성, 성능(최대 10.15% 향상), 효율성(최대 30.62% 향상) 측면에서 상당한 이점을 보여줍니다.

시사점, 한계점

시사점:
LLM의 추론 성능 향상을 위한 효과적인 새로운 방법 제시.
기존 RL 기반 방법의 한계점인 불안정성 및 모델 붕괴 문제 해결.
주석된 CoT를 효과적으로 활용하여 성능 향상 및 효율성 증대.
대조 학습을 통한 안정적이고 효율적인 미세 조정 과정 구현.
한계점:
제안된 방법의 일반화 성능에 대한 추가적인 연구 필요.
다양한 LLM 및 데이터 세트에 대한 추가적인 실험 필요.
\TheName{}의 성능 향상이 특정 데이터셋이나 모델에 국한될 가능성 존재.
알고리즘의 복잡성으로 인한 계산 비용 증가 가능성.
👍