3 个月前

五要素:基于少量特征的可解释图像分类

五要素:基于少量特征的可解释图像分类

摘要

深度神经网络通常利用成千上万条难以理解的特征来识别单一类别,其决策过程远超人类可理解的范围。本文提出在深度神经网络中引入一种可解释的稀疏且低维度的最终决策层,该层具备可度量的可解释性,并在细粒度图像分类任务上进行了验证。我们认为,只有当模型所使用的特征具备可解释性,且单次决策仅依赖极少数特征时,人类才能真正理解机器学习模型的决策过程。为此,最终决策层必须具有稀疏性,同时为使特征解释成为可能,还应保持低维度。我们将具备这种特性的模型称为稀疏低维决策模型(Sparse Low-Dimensional Decision, SLDD-Model)。实验表明,相较于密集的高维决策层,SLDD-Model在局部和全局层面均更易于解释,同时仍能保持具有竞争力的分类准确率。此外,我们提出一种新的损失函数,能够有效提升模型的特征多样性与分类性能。在四个常用的基准数据集上,我们的SLDD-Model仅使用每类50个特征中的5个,便实现了与基准模型(使用2048个特征)相当的性能,准确率保持在97%至100%之间。

代码仓库

ThomasNorr/QPM
pytorch
GitHub 中提及
thomasnorr/q-senn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
interpretable-machine-learning-on-cub-200SLDD-Model
Top 1 Accuracy: 85.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
五要素:基于少量特征的可解释图像分类 | 论文 | HyperAI超神经