用微调大模型提升单任务与多任务文本分类效果
Advancing Single and Multi-task Text Classification through Large Language Model Fine-tuning
- 对比微调后的Llama3-70B与RoBERTa,发现前者表现更优
- 多任务联合训练的微调大模型性能媲美双模型独立部署
- 适合追求高精度且希望降低推理延迟的研究者
编码器模型(如BERT、RoBERTa)和大语言模型(如Llama3)广泛应用于文本分类任务。然而,关于微调情况下编码器模型与大语言模型在文本分类中的系统性比较仍不足。本研究选取多种不同规模和架构的模型,涵盖预训练与微调方法,在20 Newsgroups(20NG)和MASSIVE数据集上评估其性能。结果表明,全量微调的Llama3-70B在各类分类任务和数据集上均优于RoBERTa-large及其他解码器型大模型。此外,通过将意图识别与槽位填充等多任务整合进单一模型,联合微调的大语言模型在两个数据集上的表现与双模型独立设置相当。研究提供了编码器模型与大语言模型在文本分类上的全面基准,并验证了通过融合多个微调解码器模型实现低延迟、高性能的可行性。
原文摘要 · Abstract (English)
Both encoder-only models (e.g., BERT, RoBERTa) and large language models (LLMs, e.g., Llama3) have been widely used for text classification tasks. However, there is a lack of systematic studies comparing the performance of encoder-based models and LLMs in text classification, particularly when fine-tuning is involved. This study employed a diverse range of models and methods, varying in size and architecture, and including both fine-tuned and pre-trained approaches. We first assessed the performances of these LLMs on the 20 Newsgroups (20NG) and MASSIVE datasets, comparing them to encoder-only RoBERTa models. Additionally, we explored the multi-task capabilities of both model types by combining multiple classification tasks, including intent detection and slot-filling, into a single model using data from both datasets. Our results indicate that fully fine-tuned Llama3-70B models outperform RoBERTa-large and other decoder LLMs across various classification tasks and datasets. Moreover, the consolidated multi-task fine-tuned LLMs matched the performance of dual-model setups in both tasks across both datasets. Overall, our study provides a comprehensive benchmark of encoder-only and LLM models on text classification tasks and demonstrates a method to combine two or more fully fine-tuned decoder LLMs for reduced latency and equivalent performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。