Daily Arxiv

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

Weight Ensembling Improves Reasoning in Language Models

Created by
  • Haebom

저자

Xingyu Dang, Christina Baek, Kaiyue Wen, Zico Kolter, Aditi Raghunathan

개요

본 논문은 추론 모델 학습 중 발생하는 다양성 붕괴 현상에 대한 연구를 다룹니다. 지도 학습 미세조정(SFT) 과정에서 Pass@1 지표는 향상되지만 Pass@k 지표는 급격히 저하되는 현상을 발견했습니다. 이에 초기 체크포인트와 최신 SFT 체크포인트 가중치를 보간하는 간단한 방법인 WiSE-FT를 제시합니다. WiSE-FT는 Pass@k 지표를 거의 완벽하게 회복시키는 동시에 Pass@1 지표도 향상시키며, 강화 학습을 통해 더욱 개선될 수 있습니다. 또한, WiSE-FT는 온도 조절과 같은 다양성 유도 디코딩 전략만으로는 달성할 수 없는 상호 보완적인 성능 향상을 제공합니다. Pass@k의 편향-분산 트레이드오프를 공식화하여 WiSE-FT가 편향과 분산을 동시에 줄이는 반면, 온도 조절은 편향과 분산 간의 트레이드오프를 야기한다는 것을 보였습니다.

시사점, 한계점

시사점:
지도 학습 미세조정 과정에서 발생하는 추론 모델의 다양성 붕괴 문제를 해결하는 새로운 방법인 WiSE-FT를 제시합니다.
WiSE-FT는 Pass@k 지표를 향상시키는 동시에 Pass@1 지표도 개선합니다.
WiSE-FT는 강화 학습과 결합하여 더 나은 성능을 달성할 수 있습니다.
WiSE-FT는 기존의 다양성 유도 디코딩 전략과 상호 보완적인 효과를 제공합니다.
Pass@k의 편향-분산 트레이드오프에 대한 새로운 분석을 제공합니다.
한계점:
WiSE-FT의 효과가 모든 추론 모델과 데이터셋에 적용될 수 있는지는 추가적인 연구가 필요합니다.
WiSE-FT의 작동 원리를 더욱 명확히 밝힐 필요가 있습니다.
특정 설정에서만 효과적일 수 있는 일반화 가능성에 대한 추가적인 검증이 필요합니다.
👍