arXiv:2411.02476cs.CLcs.AI2024-11被引 12

用指令微调和模型融合提升小模型在金融文本分类中的表现

A Comparative Analysis of Instruction Fine-Tuning LLMs for Financial Text Classification

  • 对Mistral-7B等小模型进行指令微调,提升金融任务性能
  • 融合多任务微调模型后,零样本测试准确率显著提高
  • 适合金融领域研究者与需要轻量级模型的从业者

大语言模型在自然语言处理任务中表现出色,但在金融文本这类专业领域仍面临挑战。本研究针对Mistral-7B、Llama3-8B和Phi3-mini等小规模模型,通过指令微调提升其在四类金融文本分类任务中的表现。实验发现,基础模型微调后性能下降明显,而指令微调模型保持更稳健的表现。为缓解这一问题,采用模型融合技术,将单任务领域微调模型与基础模型结合,显著提升了在三类未见复杂金融任务(论点分类、交易完整性分类、因果分类)上的零样本能力,部分数据集甚至超过原始模型精度。结果表明,指令微调与模型融合是适配LLM于金融文本分类的有效策略。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated impressive capabilities across diverse Natural Language Processing (NLP) tasks, including language understanding, reasoning, and generation. However, general-domain LLMs often struggle with financial tasks due to the technical and specialized nature of financial texts. This study investigates the efficacy of instruction fine-tuning smaller-scale LLMs, including Mistral-7B, Llama3-8B, and Phi3-mini, to enhance their performance in financial text classification tasks. We fine-tuned both instruction-tuned and base models across four financial classification tasks, achieving significant improvements in task-specific performance. Furthermore, we evaluated the zero-shot capabilities of these fine-tuned models on three unseen complex financial tasks, including argument classification, deal completeness classification, and causal classification. Our results indicate while base model fine-tuning led to greater degradation, instruction-tuned models maintained more robust performance. To address this degradation, we employed model merging techniques, integrating single-task domain-specific fine-tuned models with the base model. Using this merging method resulted in significant enhancements in zero-shot performance, even exceeding the original model's accuracy on certain datasets. Our findings underscore the effectiveness of instruction fine-tuning and model merging for adapting LLMs to specialized financial text classification tasks.

金融文本指令微调模型融合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。