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

风格链接:理解深度学习模型中的学习特征

Linking in Style: Understanding learned features in deep learning models

Frankfurt Institute for Advanced Studies (FIAS)(法兰克福高等研究院);Goethe University Frankfurt(法兰克福大学)

Maren H. Wehrheim, Pamela Osuna-Vargas, Matthias Kaschube

AI总结:

提出一种自动方法,通过链接网络将预训练分类器的特征映射到StyleGAN-XL的潜在空间,实现CNN学习特征的可视化与系统量化,为理解抽象表示提供客观视角。

摘要:

卷积神经网络(CNN)学习抽象特征以执行目标分类,但由于结果难以解释或计算成本高昂,理解这些特征仍然具有挑战性。我们提出了一种自动方法来可视化和系统分析CNN中学习到的特征。具体而言,我们引入了一个链接网络,将预训练分类器的倒数第二层映射到生成模型(StyleGAN-XL)的潜在空间,从而实现分类器表示的可解释、人类友好的可视化。我们的研究结果表明,这两个空间中存在一致的语义顺序,使得它们之间能够进行直接的线性映射。训练链接网络在计算上是低成本的,并且与GAN和分类器的训练相解耦。我们引入了一个自动流程,利用这种基于GAN的可视化,通过分析图像域中分类器的激活变化来量化学习到的表示。这种量化使我们能够同时系统地研究数千个单元的学习表示,并提取和可视化对特定语义概念具有选择性的单元。此外,我们展示了如何使用我们的方法通过反事实示例来量化和解释分类器的决策边界。总体而言,我们的方法为CNN中学习到的抽象表示提供了系统和客观的视角。 https://github.com/kaschube-lab/LinkingInStyle.git

英文摘要:

Convolutional neural networks (CNNs) learn abstract features to perform object classification, but understanding these features remains challenging due to difficult-to-interpret results or high computational costs. We propose an automatic method to visualize and systematically analyze learned features in CNNs. Specifically, we introduce a linking network that maps the penultimate layer of a pre-trained classifier to the latent space of a generative model (StyleGAN-XL), thereby enabling an interpretable, human-friendly visualization of the classifier's representations. Our findings indicate a congruent semantic order in both spaces, enabling a direct linear mapping between them. Training the linking network is computationally inexpensive and decoupled from training both the GAN and the classifier. We introduce an automatic pipeline that utilizes such GAN-based visualizations to quantify learned representations by analyzing activation changes in the classifier in the image domain. This quantification allows us to systematically study the learned representations in several thousand units simultaneously and to extract and visualize units selective for specific semantic concepts. Further, we illustrate how our method can be used to quantify and interpret the classifier's decision boundary using counterfactual examples. Overall, our method offers systematic and objective perspectives on learned abstract representations in CNNs. https://github.com/kaschube-lab/LinkingInStyle.git

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

↑