Daily Arxiv

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

Amplify Adjacent Token Differences: Enhancing Long Chain-of-Thought Reasoning with Shift-FFN

Created by
  • Haebom
Category
Empty

저자

Yao Xu, Mingyu Xu, Fangyu Lei, Wangtao Sun, Xiangrong Zeng, Bingning Wang, Guang Liu, Shizhu He, Jun Zhao, Kang Liu

개요

본 논문은 Long Chain-of-Thought (Long-CoT) 추론을 사용하는 OpenAI-o1 및 DeepSeek-R1과 같은 모델들이 복잡한 추론 과제에서 뛰어난 성능을 보이는 데 주목합니다. 하지만 Long-CoT 데이터로 전체 매개변수 또는 낮은 계수의 LoRA를 사용하여 LLM을 미세 조정하면 모델이 이전 추론 단계를 반복하는 순환적 추론(Cyclical Reasoning)으로 이어질 수 있다는 것을 발견했습니다. 인접 토큰 간 표현의 차이가 작을수록 순환적 추론 경향이 높아지는 것을 분석을 통해 밝혔습니다. 이 문제를 해결하기 위해, 본 논문에서는 이전 토큰으로 현재 토큰의 표현을 수정한 후 FFN에 입력하는 새로운 방법인 Shift Feedforward Networks (Shift-FFN)을 제안합니다. 이 구조는 인접 토큰 간 표현 차이를 동적으로 증폭합니다. 다양한 수학적 추론 과제에 대한 광범위한 실험을 통해, Shift-FFN과 결합된 LoRA가 전체 미세 조정 및 표준 LoRA에 비해 다양한 데이터 크기에 걸쳐 더 높은 정확도와 더 낮은 순환적 추론 비율을 달성함을 보여줍니다.

시사점, 한계점

시사점: Shift-FFN이 Long-CoT 학습에서 발생하는 순환적 추론 문제를 효과적으로 완화하며, LoRA와 결합하여 LLM의 성능 향상과 훈련 효율성을 높일 수 있음을 제시합니다. 다양한 데이터 크기에서 일관된 성능 향상을 보였습니다.
한계점: 제시된 방법의 효과는 주로 수학적 추론 과제에 국한되어 있으며, 다른 유형의 과제에 대한 일반화 가능성은 추가 연구가 필요합니다. 또한, https://anonymous.4open.science/r/Shift-FFN 에 공개된 데이터 및 코드에 대한 접근성 및 검증이 필요합니다. 순환적 추론의 정의 및 측정 방식에 대한 명확한 설명이 부족할 수 있습니다.
👍