arXiv:2409.18999cs.CLcs.LG2024-09被引 5

用GPT生成金融数据,让小模型达到大模型效果

Enhancing TinyBERT for Financial Sentiment Analysis Using GPT-Augmented FinBERT Distillation

  • 用GPT-4 Omni生成金融领域合成数据增强训练
  • 小型模型TinyFinBERT性能接近原版FinBERT
  • 适合资源有限但需高效金融情感分析的场景

在快速发展的金融情感分析领域,模型的效率与准确性至关重要。尽管基于Transformer的BERT和大型语言模型(如GPT-4)显著提升了NLP任务表现,但其计算开销大,在边缘计算环境中难部署,且大模型的高资源需求限制了实际应用。本研究利用GPT-4 Omni的生成能力,创建领域特定的合成训练数据,解决数据稀缺问题,并提升小模型性能,使其媲美大模型。研究聚焦于增强专用于金融情感分析的FinBERT模型,通过双阶段知识蒸馏策略构建轻量级模型TinyFinBERT。使用GPT-4 Omni生成的新样本与数据转换方法显著提升FinBERT精度,使其作为教师模型。随后,该优化后的FinBERT将知识传递给TinyFinBERT,同时结合GPT-4 Omni与GPT-3.5 Turbo生成的数据进行蒸馏,采用软标签与中间层特征蒸馏。训练与评估基于PhraseBank和FiQA 2018 Task1数据集,结果表明TinyFinBERT在性能上可比肩FinBERT,体积更小、效率更高。

原文摘要 · Abstract (English)

In the rapidly evolving field of financial sentiment analysis, the efficiency and accuracy of predictive models are critical due to their significant impact on financial markets. Transformer based models like BERT and large language models (LLMs) like GPT-4, have advanced NLP tasks considerably. Despite their advantages, BERT-based models face challenges with computational intensity in edge computing environments, and the substantial size and compute requirements of LLMs limit their practical deployment. This study proposes leveraging the generative capabilities of LLMs, such as GPT-4 Omni, to create synthetic, domain-specific training data. This approach addresses the challenge of data scarcity and enhances the performance of smaller models by making them competitive with their larger counterparts. The research specifically aims to enhance FinBERT, a BERT model fine-tuned for financial sentiment analysis, and develop TinyFinBERT, a compact transformer model, through a structured, two-tiered knowledge distillation strategy. Using data augmented by GPT-4 Omni, which involves generating new training examples and transforming existing data, we significantly improved the accuracy of FinBERT, preparing it to serve as a teacher model. This enhanced FinBERT then distilled knowledge to TinyFinBERT, employing both GPT-4 Omni and GPT-3.5 Turbo augmented data. The distillation strategy incorporated both logit and intermediate layer distillation. The training and evaluation of TinyFinBERT utilized the PhraseBank dataset and the FiQA 2018 Task1 dataset, achieving performance comparable to FinBERT while being substantially smaller and more efficient. This research demonstrates how LLMs can effectively contribute to the advancement of financial sentiment analysis by enhancing the capabilities of smaller, more efficient models through innovative data augmentation and distillation techniques.

金融分析知识蒸馏GPT增强小模型

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