Daily Arxiv

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

Robust Reinforcement Learning from Human Feedback for Large Language Models Fine-Tuning

Created by
  • Haebom

저자

Kai Ye, Hongyi Zhou, Jin Zhu, Francesco Quinzan, Chengchung Shi

개요

본 논문은 인간 피드백으로부터 강화 학습(RLHF)을 사용하여 대규모 언어 모델(LLM)의 출력을 인간의 선호도와 일치시키는 문제를 다룬다. 기존 RLHF 알고리즘들은 대부분 Bradley-Terry 모델을 사용하여 보상 함수를 학습하지만, 이 모델은 인간의 선호도에 대한 가정이 현실 세계 판단의 복잡성과 변동성을 반영하지 못할 수 있다는 한계를 지닌다. 본 논문에서는 이러한 보상 모델의 오류에도 불구하고 기존 접근 방식의 성능을 향상시키는 강건한 알고리즘을 제안한다. 이 알고리즘은 이론적으로 보상 및 정책 추정기의 분산을 줄여 개선된 후회 상한선을 제공한다. LLM 벤치마크 데이터 세트에 대한 실험적 평가 결과, 제안된 알고리즘은 기존 방법보다 일관되게 우수한 성능을 보이며, Anthropic Helpful and Harmless 데이터 세트에서 77~81%의 응답이 기준선보다 선호되는 것으로 나타났다.

시사점, 한계점

시사점: 기존 RLHF 알고리즘의 한계를 극복하는 강건한 알고리즘을 제시하여 LLM의 인간 선호도와의 정렬 성능을 향상시켰다. Anthropic Helpful and Harmless 데이터셋에서 기존 방법 대비 77-81%의 우수한 성능을 보였다. 보상 및 정책 추정기의 분산 감소를 통한 이론적 근거를 제시하였다.
한계점: 제안된 알고리즘의 성능 향상이 특정 데이터셋에 국한될 가능성이 있다. 실제 적용에 필요한 계산 비용 및 복잡도에 대한 추가적인 분석이 필요하다. 다양한 유형의 LLM 및 인간 선호도 데이터에 대한 추가적인 실험이 필요하다.
👍