arXiv:2601.22531cs.LGcs.AI2026-01中稿 · ICML

让小模型从大模型的解释中学习,提升可解释性预测效果

Learn from A Rationalist: Distilling Intermediate Interpretable Rationales

  • 用知识蒸馏让小模型学习大模型的推理过程和预测结果
  • 在IMDB、CIFAR等数据集上,小模型性能显著提升
  • 适合想提升小模型可解释性和准确率的研究者

由于深度神经网络在高风险领域广泛应用,其可解释性受到更多关注。理由提取(RE)通过选择-预测架构,使两个神经网络协同学习特征选择与预测。但仅依赖最终任务监督时,搜索所有特征组合极为困难,尤其当基础模型能力有限时。为此,我们提出REKD(基于知识蒸馏的理由提取),让一个学生模型不仅优化自身,还学习教师模型(即理性主义者)提供的理由与预测。该方法结构上模拟人类从可解释知识中学习的过程,且对神经网络类型无要求,可适配任意黑盒模型。我们在多个BERT和视觉变换器(ViT)变体上进行实验,结果表明,在语言与视觉分类数据集(如IMDB电影评论、CIFAR-10和CIFAR-100)上,使用REKD的学生模型预测性能显著提升。

原文摘要 · Abstract (English)

Because of the pervasive use of deep neural networks (DNNs), especially in high-stakes domains, the interpretability of DNNs has received increased attention. The general idea of rationale extraction (RE) is to provide an interpretable-by-design framework for DNNs via a select-predict architecture where two neural networks learn jointly to perform feature selection and prediction, respectively. Given only the remote supervision from the final task prediction, the process of learning to select subsets of features (or rationales) requires searching in the space of all possible feature combinations, which is computationally challenging and even harder when the base neural networks are not sufficiently capable. To improve the predictive performance of RE models that are based on less capable or smaller neural networks (i.e., the students), we propose REKD (Rationale Extraction with Knowledge Distillation) where a student RE model learns from the rationales and predictions of a teacher (i.e., a rationalist) in addition to the student's own RE optimization. This structural adjustment to RE aligns well with how humans could learn effectively from interpretable and verifiable knowledge. Because of the neural-model agnostic nature of the method, any black-box neural network could be integrated as a backbone model. To demonstrate the viability of REKD, we conduct experiments with multiple variants of BERT and vision transformer (ViT) models. Our experiments across language and vision classification datasets (i.e., IMDB movie reviews, CIFAR 10 and CIFAR 100) show that REKD significantly improves the predictive performance of the student RE models.

可解释性知识蒸馏模型压缩

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