arXiv:2508.08635cs.LG2025-08

用关键词驱动方法微调小模型,高效完成专业文本分类。

Classifier Language Models: Unifying Sparse Finetuning and Adaptive Tokenization for Specialized Classification Tasks

  • 根据任务词元定位敏感参数,仅微调关键部分。
  • 在5个任务上优于全量微调、LoRA等方法,训练成本减半。
  • 适合资源有限的工业级专用分类场景。

语义文本分类需理解特定词元的上下文意义,而非表面模式或关键词,大语言模型(LLMs)对此天然适配。但工业中的语义分类任务(如客户意图识别、语义角色标注)高度专业化,需领域专家标注,且对推理吞吐要求高,限制了模型规模。因此,更适合通过微调小型语言模型(如mini-encoders)构建定制分类器。本文提出一种基于词元驱动的稀疏微调策略:利用微调数据集中任务相关的词元构造,识别并微调一小部分敏感参数,其余预训练权重保持不变。相比低秩适配(LoRA)等引入额外参数的方法,本方案不增加模型参数。案例研究显示,该方法能有效识别相关语义词元,在五个多样化语义分类任务中超越端到端微调、LoRA、层选择和前缀调优,训练成本减半,表现更稳定。

原文摘要 · Abstract (English)

Semantic text classification requires the understanding of the contextual significance of specific tokens rather than surface-level patterns or keywords (as in rule-based or statistical text classification), making large language models (LLMs) well-suited for this task. However, semantic classification applications in industry, like customer intent detection or semantic role labeling, tend to be highly specialized. They require annotation by domain experts in contrast to general-purpose corpora for pretraining. Further, they typically require high inference throughputs which limits the model size from latency and cost perspectives. Thus, for a range of specialized classification tasks, the preferred solution is to develop customized classifiers by finetuning smaller language models (e.g., mini-encoders, small language models). In this work, we develop a token-driven sparse finetuning strategy to adapt small language models to specialized classification tasks. We identify and finetune a small sensitive subset of model parameters by leveraging task-specific token constructs in the finetuning dataset, while leaving most of the pretrained weights unchanged. Unlike adapter approaches such as low rank adaptation (LoRA), we do not introduce additional parameters to the model. Our approach identifies highly relevant semantic tokens (case study in the Appendix) and outperforms end-to-end finetuning, LoRA, layer selection, and prefix tuning on five diverse semantic classification tasks. We achieve greater stability and half the training costs vs. end-to-end finetuning.

微调小模型分类

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