arXiv:2505.16078cs.CL2025-05ACL被引 5

小模型在工业文本分类中表现如何?实测给出高效部署方案。

Small Language Models in the Real World: Insights from Industrial Text Classification

  • 对比提示工程与微调,验证小模型在真实场景的可行性。
  • 小模型在邮件、法律文档等任务上达到80%以上准确率。
  • 重点关注显存效率,适合本地部署的轻量级应用。

随着ChatGPT的兴起,基于Transformer的解码器模型如Llama在文本分类任务中表现出色,但其推理效率低,依赖高质量提示,且对GPU资源消耗大,限制了实际部署。本文针对工业场景中的实际需求,系统评估了提示工程与监督微调在小型语言模型上的表现,涵盖邮件分类、法律文档归类及超长学术文本分类等典型任务。重点分析小模型在性能与显存(VRAM)利用率方面的平衡,揭示其在本地化部署中的潜力,为工业界提供可落地的小模型应用参考。

原文摘要 · Abstract (English)

With the emergence of ChatGPT, Transformer models have significantly advanced text classification and related tasks. Decoder-only models such as Llama exhibit strong performance and flexibility, yet they suffer from inefficiency on inference due to token-by-token generation, and their effectiveness in text classification tasks heavily depends on prompt quality. Moreover, their substantial GPU resource requirements often limit widespread adoption. Thus, the question of whether smaller language models are capable of effectively handling text classification tasks emerges as a topic of significant interest. However, the selection of appropriate models and methodologies remains largely underexplored. In this paper, we conduct a comprehensive evaluation of prompt engineering and supervised fine-tuning methods for transformer-based text classification. Specifically, we focus on practical industrial scenarios, including email classification, legal document categorization, and the classification of extremely long academic texts. We examine the strengths and limitations of smaller models, with particular attention to both their performance and their efficiency in Video Random-Access Memory (VRAM) utilization, thereby providing valuable insights for the local deployment and application of compact models in industrial settings.

小模型文本分类工业应用显存优化

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