arXiv:2510.21631cs.LGcs.AI2025-10NeurIPS被引 2

用反事实解释提升小样本知识蒸馏效果,少数据也能高效复制大模型能力。

Few-Shot Knowledge Distillation of LLMs With Counterfactual Explanations

  • 通过反事实解释生成关键样本,精准捕捉教师模型决策边界。
  • 仅需8-512样本即超越传统蒸馏方法,且使用样本量减半仍更优。
  • 适合资源受限场景下快速部署小型高效语言模型。

知识蒸馏是将复杂教师模型的能力迁移至轻量学生模型的有效方法,尤其适用于任务感知场景。然而,现有任务感知蒸馏方法通常需要大量数据,在实际应用中可能难以获取或成本高昂。本文提出一种新策略CoD(Counterfactual-explanation-infused Distillation),通过系统注入反事实解释(CFEs)实现少样本任务感知知识蒸馏。CFEs指能以最小扰动改变教师模型输出的输入。CoD利用这些样本在极少数据下精确映射教师模型的决策边界。我们从统计与几何角度提供理论支持,证明CFEs可提升参数估计精度,因其提供了靠近决策边界的高信息量样本。同时揭示了CFEs作为知识探针的作用机制。实验在多个数据集与LLMs上验证,CoD在极低样本量(8-512)下表现优于标准蒸馏方法。值得注意的是,仅需基线一半样本量(配对CFEs)即可取得更好性能。

原文摘要 · Abstract (English)

Knowledge distillation is a promising approach to transfer capabilities from complex teacher models to smaller, resource-efficient student models that can be deployed easily, particularly in task-aware scenarios. However, existing methods of task-aware distillation typically require substantial quantities of data which may be unavailable or expensive to obtain in many practical scenarios. In this paper, we address this challenge by introducing a novel strategy called Counterfactual-explanation-infused Distillation CoD for few-shot task-aware knowledge distillation by systematically infusing counterfactual explanations. Counterfactual explanations (CFEs) refer to inputs that can flip the output prediction of the teacher model with minimum perturbation. Our strategy CoD leverages these CFEs to precisely map the teacher's decision boundary with significantly fewer samples. We provide theoretical guarantees for motivating the role of CFEs in distillation, from both statistical and geometric perspectives. We mathematically show that CFEs can improve parameter estimation by providing more informative examples near the teacher's decision boundary. We also derive geometric insights on how CFEs effectively act as knowledge probes, helping the students mimic the teacher's decision boundaries more effectively than standard data. We perform experiments across various datasets and LLMs to show that CoD outperforms standard distillation approaches in few-shot regimes (as low as 8-512 samples). Notably, CoD only uses half of the original samples used by the baselines, paired with their corresponding CFEs and still improves performance.

知识蒸馏少样本学习反事实解释大模型

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