Daily Arxiv

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

Advancing Single and Multi-task Text Classification through Large Language Model Fine-tuning

Created by
  • Haebom

저자

Hang Zhao, Qile P. Chen, Yijing Barry Zhang, Gang Yang

개요

본 논문은 text classification task에서 encoder-only 모델(예: RoBERTa)과 LLM(예: Llama3)의 성능을 체계적으로 비교 분석한 연구입니다. 다양한 크기와 구조의 모델들을 사용하여, 사전 훈련된 모델과 fine-tuning된 모델 모두를 평가했습니다. 20 Newsgroups와 MASSIVE 데이터셋을 사용하여, Llama3-70B 모델을 RoBERTa-large 모델과 비교하고, intent detection과 slot-filling을 포함한 다중 task 학습의 가능성을 탐색했습니다. 결과적으로, fully fine-tuned된 Llama3-70B 모델이 다양한 task와 데이터셋에서 RoBERTa-large 및 다른 decoder LLM들을 능가하는 성능을 보였으며, 통합된 다중 task fine-tuning LLM이 두 가지 task에서 dual-model 설정과 동등한 성능을 달성했습니다. 이는 latency 감소와 동등한 성능을 위해 두 개 이상의 fully fine-tuned decoder LLM을 결합하는 방법을 제시합니다.

시사점, 한계점

시사점:
Fully fine-tuned Llama3-70B 모델이 text classification task에서 우수한 성능을 보임을 확인.
다중 task fine-tuning을 통해 dual-model 설정과 동등한 성능을 달성 가능함을 입증.
두 개 이상의 fully fine-tuned decoder LLM을 결합하여 latency를 감소시키는 효과적인 방법 제시.
한계점:
사용된 데이터셋의 제한으로 일반화 가능성에 대한 추가 연구 필요.
다양한 LLM과 encoder-only 모델에 대한 포괄적인 비교가 아닌, 특정 모델들에 대한 비교 연구임.
다른 text classification task나 데이터셋에 대한 추가 실험 필요.
👍