arXiv:2511.11574cs.LG2025-11

用主动学习减少大模型标注成本,高效训练小模型分类器。

LLM on a Budget: Active Knowledge Distillation for Efficient Classification of Large Text Corpora

  • 结合不确定性和随机采样,只选最需标注的数据点。
  • 相比随机采样,样本量减少80%,准确率更高。
  • 适合预算有限、需快速部署的文本分类场景。

大型语言模型(LLMs)在分类任务中表现优异,但其高昂的计算与财务成本限制了在动态环境中的大规模应用。知识蒸馏通过让大模型教师指导小模型学生,提供了可行的解决方案。然而,传统蒸馏过程对大规模数据集仍成本高昂,因需大量样本由教师标注并消耗大量token。本文提出M-RARU(多类别随机接受/拒绝不确定性采样)算法,结合不确定性与随机接受-拒绝机制,仅选择最具信息量的样本供教师标注。该方法显著减少API调用次数和数据处理时间。在五个不同学生模型(SVM、LDA、RF、GBDT、DistilBERT)上,于多个基准数据集上的实验表明,相较于随机采样,本方法可将样本需求降低高达80%,同时提升分类准确率,大幅降低财务开销与训练时长。

原文摘要 · Abstract (English)

Large Language Models (LLMs) are highly accurate in classification tasks, however, substantial computational and financial costs hinder their large-scale deployment in dynamic environments. Knowledge Distillation (KD) where a LLM "teacher" trains a smaller and more efficient "student" model, offers a promising solution to this problem. However, the distillation process itself often remains costly for large datasets, since it requires the teacher to label a vast number of samples while incurring significant token consumption. To alleviate this challenge, in this work we explore the active learning (AL) as a way to create efficient student models at a fraction of the cost while preserving the LLM's performance. In particular, we introduce M-RARU (Multi-class Randomized Accept/Reject Uncertainty Sampling), a novel AL algorithm that significantly reduces training costs. M-RARU employs an innovative strategy combining uncertainty with a randomized accept-reject mechanism to select only the most informative data points for the LLM teacher. This focused approach significantly minimizes required API calls and data processing time. We evaluate M-RARU against random sampling across five diverse student models (SVM, LDA, RF, GBDT, and DistilBERT) on multiple benchmark datasets. Experiments demonstrate that our proposed method achieves up to 80% reduction in sample requirements as compared to random sampling, substantially improving classification accuracy while reducing financial costs and overall training time.

知识蒸馏主动学习大模型压缩文本分类

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