arXiv:2410.08407cs.LGcs.CY2024-10被引 6

知识蒸馏会改变模型对某些类别的偏见,可能提升公平性

What is Left After Distillation? How Knowledge Transfer Impacts Fairness and Bias

  • 通过调整蒸馏温度,改变学生模型的类别偏差分布
  • 在多个数据集上,高温蒸馏使学生模型公平性超越教师模型
  • 适合关注模型公平性与敏感场景应用的研究者

知识蒸馏是常用的深度神经网络压缩方法,通常保持整体泛化性能。然而,我们在平衡图像分类数据集(如CIFAR-100、Tiny ImageNet和ImageNet)上发现,多达41%的类别在比较教师模型与蒸馏学生模型或非蒸馏学生模型的类别准确率时,表现出统计显著的类别偏差变化。在使用CelebA、Trifeature和HateXplain数据集评估公平性时,结果表明提高蒸馏温度可增强学生模型的公平性,且在高温下其公平性甚至超过教师模型。此外,个体公平性分析也显示,更高温度能改善学生模型对相似样本的预测一致性。本研究揭示了蒸馏对特定类别的不均衡影响及其在公平性中的潜在作用,提醒在敏感应用中使用蒸馏模型需谨慎。

原文摘要 · Abstract (English)

Knowledge Distillation is a commonly used Deep Neural Network (DNN) compression method, which often maintains overall generalization performance. However, we show that even for balanced image classification datasets, such as CIFAR-100, Tiny ImageNet and ImageNet, as many as 41% of the classes are statistically significantly affected by distillation when comparing class-wise accuracy (i.e. class bias) between a teacher/distilled student or distilled student/non-distilled student model. Changes in class bias are not necessarily an undesirable outcome when considered outside of the context of a model's usage. Using two common fairness metrics, Demographic Parity Difference (DPD) and Equalized Odds Difference (EOD) on models trained with the CelebA, Trifeature, and HateXplain datasets, our results suggest that increasing the distillation temperature improves the distilled student model's fairness, and the distilled student fairness can even surpass the fairness of the teacher model at high temperatures. Additionally, we examine individual fairness, ensuring similar instances receive similar predictions. Our results confirm that higher temperatures also improve the distilled student model's individual fairness. This study highlights the uneven effects of distillation on certain classes and its potentially significant role in fairness, emphasizing that caution is warranted when using distilled models for sensitive application domains.

知识蒸馏模型公平性类别偏差

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