用反事实解释找模型弱点,提升CNN分类准确率1-2%。
Leveraging counterfactual concepts for debugging and improving CNN model performance
- 通过反事实推理识别影响决策的关键卷积滤波器。
- 设计新损失函数,强化相关滤波器激活,抑制无关滤波器。
- 可定位误判与模型偏差,适合模型调试与性能优化者。
反事实解释方法因能提供贴近人类思维的直观解释,近年来在基于CNN的图像分类器中受到广泛关注。然而,如何利用可解释性方法改进模型性能仍关注不足。本文提出一种利用反事实概念提升CNN图像分类性能的新方法。该方法通过反事实推理识别决策过程中关键的卷积滤波器,并设计新的训练策略与损失函数,以增强特定类别相关滤波器的激活,抑制无关滤波器的激活。此过程有效缩小局部预测的激活模式与对应类别全局激活模式之间的偏差。结合反事实解释,可验证未见样本的预测并识别误分类。所提方法揭示了模型学习中的潜在缺陷与偏差,支持针对性改进。在公开数据集上的实验表明,模型性能提升1-2%,验证了方法的有效性。
原文摘要 · Abstract (English)
Counterfactual explanation methods have recently received significant attention for explaining CNN-based image classifiers due to their ability to provide easily understandable explanations that align more closely with human reasoning. However, limited attention has been given to utilizing explainability methods to improve model performance. In this paper, we propose to leverage counterfactual concepts aiming to enhance the performance of CNN models in image classification tasks. Our proposed approach utilizes counterfactual reasoning to identify crucial filters used in the decision-making process. Following this, we perform model retraining through the design of a novel methodology and loss functions that encourage the activation of class-relevant important filters and discourage the activation of irrelevant filters for each class. This process effectively minimizes the deviation of activation patterns of local predictions and the global activation patterns of their respective inferred classes. By incorporating counterfactual explanations, we validate unseen model predictions and identify misclassifications. The proposed methodology provides insights into potential weaknesses and biases in the model's learning process, enabling targeted improvements and enhanced performance. Experimental results on publicly available datasets have demonstrated an improvement of 1-2\%, validating the effectiveness of the approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。