arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.04441cs.CVcs.AI
收藏

基于知识注入的图像分类网络增强方法

Image classification network enhancement methods based on knowledge injection

Xidian University(西安电子科技大学)

Yishuang Tian, Ning Wang, Liang Zhang

AI总结:

针对现有深度神经网络缺乏可解释性且未利用人类知识的问题,本文提出基于人类认知模型构建的多层级深度学习算法与知识注入数据集,有效提升了模型在不同规模隐藏层上的可解释性与分类性能。

摘要:

当前的深度神经网络算法仍停留在像Image-Label对这样的端到端训练监督方法上,这使得传统算法难以解释结果产生的原因,且预测逻辑难以理解和分析。当前算法未利用现有的人类知识信息,导致模型不符合人类认知模型,不便于人类使用。为解决上述问题,本发明提供了一种基于人类知识的深度神经网络训练方法,该方法利用人类认知模型构建深度神经网络训练模型,并使用现有的人类知识信息构建深度神经网络训练模型。本文提出了一种多层级深度学习算法,由多层级深度神经网络架构和多层级深度学习框架组成。实验结果表明,所提出的算法能有效解释神经网络的隐藏信息。我们研究的目标是通过分析知识注入对分类任务的影响,提升深度神经网络(DNNs)的可解释性。我们构建了一个知识注入数据集,其中包含匹配的知识数据和图像分类数据。该知识注入数据集是本文实验的基准数据集。我们的模型展示了在不同规模下隐藏层在可解释性和分类任务性能上的提升。

英文摘要:

The current deep neural network algorithm still stays in the end-to-end training supervision method like Image-Label pairs, which makes traditional algorithm is difficult to explain the reason for the results, and the prediction logic is difficult to understand and analyze. The current algorithm does not use the existing human knowledge information, which makes the model not in line with the human cognition model and makes the model not suitable for human use. In order to solve the above problems, the present invention provides a deep neural network training method based on the human knowledge, which uses the human cognition model to construct the deep neural network training model, and uses the existing human knowledge information to construct the deep neural network training model. This paper proposes a multi-level hierarchical deep learning algorithm, which is composed of multi-level hierarchical deep neural network architecture and multi-level hierarchical deep learning framework. The experimental results show that the proposed algorithm can effectively explain the hidden information of the neural network. The goal of our study is to improve the interpretability of deep neural networks (DNNs) by providing an analysis of the impact of knowledge injection on the classification task. We constructed a knowledge injection dataset with matching knowledge data and image classification data. The knowledge injection dataset is the benchmark dataset for the experiments in the paper. Our model expresses the improvement in interpretability and classification task performance of hidden layers at different scales.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑