[공지사항]을 빙자한 안부와 근황 
Show more

Daily Arxiv

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

Steering Out-of-Distribution Generalization with Concept Ablation Fine-Tuning

Created by
  • Haebom

저자

Helena Casademunt, Caden Juang, Adam Karvonen, Samuel Marks, Senthooran Rajamanoharan, Neel Nanda

개요

본 논문은 대규모 언어 모델(LLM)의 미세 조정 과정에서 발생하는 의도치 않은 분포 외 일반화 문제를 해결하기 위한 새로운 기법인 개념 제거 미세 조정(CAFT)을 제안합니다. 기존의 접근 방식은 훈련 데이터를 수정하는 데 의존하지만, CAFT는 해석 가능성 도구를 활용하여 훈련 데이터를 수정하지 않고도 LLM의 일반화 방식을 제어합니다. 원하지 않는 개념에 해당하는 LLM의 잠재 공간 내 방향 벡터들을 선형 투영을 통해 제거함으로써, 의도하지 않은 일반화로부터 모델을 멀리하게 합니다. 본 연구는 CAFT를 세 가지 미세 조정 작업에 적용하여, 좁은 작업에 미세 조정된 LLM이 일반적인 질문에 대해 심각하게 잘못 정렬된 응답을 생성하는 출현 오정렬 현상을 포함한 문제를 해결했습니다. 훈련 데이터 변경 없이도 CAFT는 잘못 정렬된 응답을 10배 감소시키면서 훈련 분포에 대한 성능 저하 없이 효과를 보였습니다.

시사점, 한계점

시사점:
훈련 데이터 수정 없이 LLM의 일반화를 제어하는 새로운 방법 제시
해석 가능성 도구를 활용하여 LLM의 잠재 공간 조작을 통해 의도치 않은 일반화 방지
출현 오정렬과 같은 문제 해결에 효과적임을 실험적으로 증명
미세 조정 과정에서 발생하는 다양한 문제에 적용 가능성 제시
한계점:
제안된 방법의 효과는 특정 작업 및 데이터셋에 국한될 가능성 존재
원하지 않는 개념에 대한 정확한 방향 벡터 식별의 어려움
고차원 잠재 공간에서의 선형 투영의 한계
다양한 LLM 아키텍처 및 크기에 대한 일반화 가능성 검증 필요
👍