arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.01331cs.CVcs.AIcs.LG

基于生成模型重构无监督概念可解释网络的风格

Restyling Unsupervised Concept Based Interpretable Networks with Generative Models

Sorbonne Université(索邦大学);Technical University of Munich(慕尼黑工业大学);Helmholtz Munich(亥姆霍兹慕尼黑中心);Munich Center for Machine Learning (MCML)(慕尼黑机器学习中心);LTCI, Télécom Paris, Institut Polytechnique de Paris(巴黎理工学院 巴黎高等电信学院 LTCI)

Jayneel Parekh, Quentin Bouniot, Pavlo Mozharovskyi, Alasdair Newson, Florence d'Alché-Buc

AI总结:

针对无监督概念可解释网络的概念可视化难题,提出将概念特征映射到预训练生成模型潜空间的方法,实现高质量可视化与交互式解释,提升训练效率,经多基准验证在准确率、保真度等方面表现优异。

摘要:

近年来,开发具备内在可解释性的预测模型愈发受到重视。这类模型中的一个子类——可解释网络依赖于学习高层概念——因概念表示与人类沟通方式相近而备受重视。然而,对学习到的无监督概念字典进行可视化和理解存在重大局限,尤其是针对大规模图像。本文提出一种新方法,核心是将概念特征映射到预训练生成模型的潜空间中。生成模型的使用可实现高质量可视化,并提供了一种直观的交互式流程,通过填补概念激活并可视化生成的修改效果,更好地解释学习到的概念。此外,利用预训练生成模型还有提升系统训练效率的额外优势。我们从可解释预测网络的准确率、重建保真度,以及学习概念的忠实度和一致性等方面,定量验证了方法的有效性。实验在多个大规模图像识别基准数据集上开展。项目页面见:https://jayneelparekh.github.io/VisCoIN_project_page/

英文摘要:

Developing inherently interpretable models for prediction has gained prominence in recent years. A subclass of these models, wherein the interpretable network relies on learning high-level concepts, are valued because of closeness of concept representations to human communication. However, the visualization and understanding of the learnt unsupervised dictionary of concepts encounters major limitations, especially for large-scale images. We propose here a novel method that relies on mapping the concept features to the latent space of a pretrained generative model. The use of a generative model enables high quality visualization, and lays out an intuitive and interactive procedure for better interpretation of the learnt concepts by imputing concept activations and visualizing generated modifications. Furthermore, leveraging pretrained generative models has the additional advantage of making the training of the system more efficient. We quantitatively ascertain the efficacy of our method in terms of accuracy of the interpretable prediction network, fidelity of reconstruction, as well as faithfulness and consistency of learnt concepts. The experiments are conducted on multiple image recognition benchmarks for large-scale images. Project page available at https://jayneelparekh.github.io/VisCoIN_project_page/

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

↑