arXiv:2410.01109cs.AIcs.CE2024-10被引 16

金融领域微调大模型时,多任务混合反而比单一任务更有效。

Mixing It Up: The Cocktail Effect of Multi-Task Fine-Tuning on LLM Performance -- A Case Study in Finance

  • 用多个相关任务一起微调,提升模型表现
  • 小模型Phi-3-Mini在金融任务上超越GPT-4-o
  • 适合需要高效精准微调的金融类应用

大型语言模型(LLMs)在金融等特定领域的应用迅速扩展。通常通过评估其在多个下游任务中的表现来衡量领域专用模型的性能。本文对领域内微调方法进行了深入分析,发现针对目标任务单独微调并非最佳策略。相反,将多个相关任务组合进行多任务微调,可显著提升模型性能。我们通过超过200次大规模实验验证了这一方法的有效性,结果显示,如Phi-3-Mini这样的小型模型在金融基准测试中表现优于更大的GPT-4-o模型。研究还探索了通用指令数据作为正则化手段的作用,表明其有助于减少性能下降;同时引入数学数据后,数值推理能力提升并有效迁移到金融任务中。然而,仅针对下游任务微调虽能提高特定任务表现,但并未带来领域知识或复杂推理能力的普遍增强。

原文摘要 · Abstract (English)

The application of large language models (LLMs) in domain-specific contexts, including finance, has expanded rapidly. Domain-specific LLMs are typically evaluated based on their performance in various downstream tasks relevant to the domain. In this work, we present a detailed analysis of fine-tuning LLMs for such tasks. Somewhat counterintuitively, we find that in domain-specific cases, fine-tuning exclusively on the target task is not always the most effective strategy. Instead, multi-task finetuning - where models are trained on a cocktail of related tasks - can significantly enhance performance. We demonstrate how this approach enables a small model, such as Phi-3-Mini, to achieve state-of-the-art results, even surpassing the much larger GPT-4-o model on financial benchmarks. Our study involves a large-scale experiment, conducting over 200 training experiments using several widely adopted LLMs as baselines, and empirically confirms the benefits of multi-task fine-tuning. Additionally, we explore the use of general instruction data as a form of regularization, suggesting that it helps minimize performance degradation. We also investigate the inclusion of mathematical data, finding improvements in numerical reasoning that transfer effectively to financial tasks. Finally, we note that while fine-tuning for downstream tasks leads to targeted improvements in task performance, it does not necessarily result in broader gains in domain knowledge or complex domain reasoning abilities.

多任务微调金融AI小模型优化

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