arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.08067cs.CVcs.AI

使用条件生成对抗网络解释CNN预测

Interpreting CNN Predictions using Conditional Generative Adversarial Networks

Amrita Vishwa Vidyapeetham(阿米塔大学);Universitat Politècnica de Catalunya (UPC-BarcelonaTECH)(加泰罗尼亚理工大学(巴塞罗那理工大学));Institut de Recerca Sant Joan de Déu (IRSJD)(圣胡安·德·德乌研究所)

R T Akash Guna, Raul Benitez, O K Sikha

AI总结:

本研究提出用条件GAN生成CNN视觉解释的方法,通过累积平均中间解释图构建适配表示,设计两种馈入与训练策略,经定性定量评估验证其跨架构跨数据集的通用性,发现CNN首尾层对解释同等重要,为深度学习赋能模型解释提供了新路径。

摘要:

我们提出了一种训练条件生成对抗网络(GAN,Generative Adversarial Network)以生成卷积神经网络(CNN,Convolutional Neural Network)视觉解释的新方法。为理解CNN,训练GAN时会输入关于CNN在做预测时如何处理图像的信息。提供这类信息存在两大主要挑战:如何以可输入GAN的形式表示该信息,以及如何将该表示有效馈入GAN。为解决这些问题,我们通过对中间解释图进行累积平均,开发了一种适配CNN架构的表示方法。我们还提出了两种将表示馈入GAN并选择有效训练策略的备选方案。我们的方法能学习CNN的通用特性,且不受数据集和CNN架构的限制。本研究包含定性和定量评估,并将所提出的GAN与当前最优方法进行了对比。我们在对所提GAN进行解释时发现,CNN的初始层和最终层对解释CNN同等关键。我们认为,训练GAN来解释CNN将为利用快速发展的深度学习技术提升解释效果打开大门。实验所用代码已公开于https://github.com/Akash-guna/Explain-CNN-With-GANS。

英文摘要:

We propose a novel method that trains a conditional Generative Adversarial Network (GAN) to generate visual interpretations of a Convolutional Neural Network (CNN). To comprehend a CNN, the GAN is trained with information on how the CNN processes an image when making predictions. Supplying that information has two main challenges: how to represent this information in a form that is feedable to the GANs and how to effectively feed the representation to the GAN. To address these issues, we developed a suitable representation of CNN architectures by cumulatively averaging intermediate interpretation maps. We also propose two alternative approaches to feed the representations to the GAN and to choose an effective training strategy. Our approach learned the general aspects of CNNs and was agnostic to datasets and CNN architectures. The study includes both qualitative and quantitative evaluations and compares the proposed GANs with state-of-the-art approaches. We found that the initial layers of CNNs and final layers are equally crucial for interpreting CNNs upon interpreting the proposed GAN. We believe training a GAN to interpret CNNs would open doors for improved interpretations by leveraging fast-paced deep learning advancements. The code used for experimentation is publicly available at https://github.com/Akash-guna/Explain-CNN-With-GANS

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

↑