Daily Arxiv

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

MachineLearningLM: Scaling Many-shot In-context Learning via Continued Pretraining

Created by
  • Haebom

저자

Haoyu Dong, Pengkun Zhang, Mingzhe Lu, Yanzhen Shen, Guolin Ke

개요

본 논문은 대규모 언어 모델(LLM)의 컨텍스트 학습(ICL) 능력을 향상시키는 MachineLearningLM이라는 새로운 프레임워크를 제시합니다. MachineLearningLM은 수백만 개의 구조적 인과 모델(SCM)에서 생성된 다양한 기계 학습(ML) 작업을 사용하여 사전 훈련됩니다. 특히, 랜덤 포레스트 기반의 의사결정 전략을 LLM에 주입하여 수치 모델링의 강건성을 높이고, 토큰 효율적인 프롬프트를 통해 컨텍스트 창당 더 많은 예시를 처리하여 처리량을 향상시킵니다. 실험 결과, MachineLearningLM은 다양한 분야의 분포 외 표 데이터 분류 작업에서 기존 강력한 LLM 기준 모델보다 평균 15% 향상된 성능을 보였으며, 컨텍스트 내 예시의 수가 증가함에 따라 정확도가 단조 증가하는 현저한 다중 샷 스케일링 법칙을 보였습니다. 또한, 일반적인 채팅 기능, 지식 및 추론 능력도 유지했습니다.

시사점, 한계점

시사점:
LLM의 컨텍스트 학습 능력을 효과적으로 향상시키는 새로운 프레임워크를 제시.
다양한 분야의 ML 작업에서 기존 모델보다 우수한 성능을 달성.
컨텍스트 내 예시의 수 증가에 따라 성능이 향상되는 다중 샷 스케일링 법칙 확인.
일반적인 채팅 기능과 지식, 추론 능력을 유지하면서 ML 능력을 향상시킴.
토큰 효율적인 프롬프트를 통해 처리량을 크게 향상.
한계점:
현재는 특정 규모의 LLM (Qwen-2.5-7B-Instruct)과 LoRA를 사용한 실험 결과만 제시. 다른 모델이나 설정에서의 일반화 가능성에 대한 추가 연구 필요.
구조적 인과 모델(SCM)의 생성 및 선택 과정에 대한 자세한 설명 부족.
다양한 ML 작업 유형에 대한 일반화 성능에 대한 추가적인 검증 필요.
👍