Daily Arxiv

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

Inference and Verbalization Functions During In-Context Learning

Created by
  • Haebom

저자

Junyi Tao, Xiaoyin Chen, Nelson F. Liu

개요

본 논문은 대규모 언어 모델(LLM)의 컨텍스트 내 학습(ICL)에서 무관한 레이블이 미치는 영향을 연구합니다. 기존 연구에서 무관한 레이블이 ICL 성능에 미치는 영향이 미미하다는 점을 발견한 것에 착안하여, LLM이 무관한 레이블을 사용하여 ICL을 수행하는 과정을 두 단계로 가정합니다. 첫 번째 단계는 작업을 해결하는 추론 함수이고, 두 번째 단계는 추론된 답을 레이블 공간에 매핑하는 언어화 함수입니다. 본 논문은 레이블 공간의 재매핑(예: "참"/"거짓"을 "고양이"/"개"로 변경)에도 추론 함수가 불변하다는 가정을 세우고, 제어된 계층 간 교환 개입 실험을 통해 이 가설을 실증적으로 검증합니다. 여러 데이터셋과 작업(자연어 추론, 감정 분석, 주제 분류)에서 가설을 확인하고, GEMMA-7B, MISTRAL-7B-V0.3, GEMMA-2-27B, LLAMA-3.1-70B 등 다양한 오픈소스 모델에서 두 함수를 특정 계층에 국지화할 수 있음을 제시합니다.

시사점, 한계점

시사점: LLM의 ICL 과정에 대한 새로운 이해를 제공하며, 추론 함수와 언어화 함수의 분리를 통해 모델의 작동 방식에 대한 통찰력을 제공합니다. 다양한 모델과 작업에서 이러한 분리가 확인되어 일반화 가능성을 시사합니다.
한계점: 실험은 특정 오픈소스 모델에 국한되어 있으며, 다른 모델이나 작업으로의 일반화 가능성에 대한 추가 연구가 필요합니다. 추론 함수와 언어화 함수의 정확한 경계와 상호작용에 대한 추가적인 분석이 필요합니다. 레이블 공간의 재매핑에 대한 불변성이 모든 경우에 적용되는지에 대한 추가적인 검증이 필요합니다.
👍