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

面向卷积神经网络的信息论可视化分析框架

An Information-theoretic Visual Analysis Framework for Convolutional Neural Networks

The Ohio State University(俄亥俄州立大学)

Jingyi Shen, Han-Wei Shen

AI总结:

本文提出信息论可视化分析框架CNNSlicer,通过信息熵计算与交互式可视化揭示CNN内部信息变化,并在MNIST和CIFAR-10上验证其有效性。

摘要:

尽管卷积神经网络(CNN)在计算机视觉和自然语言处理领域取得了巨大成功,但其背后的工作机制仍处于广泛的讨论和研究之中。在对神经网络进行理论解释的强烈需求驱动下,一些研究者利用信息论来洞察这一黑盒模型。然而,据我们所知,在可视化研究社区中,运用信息论对神经网络进行定量分析和定性可视化尚未得到充分研究。本文结合信息熵与可视化技术,以揭示CNN的工作方式。具体而言,我们首先引入一个数据模型来组织可从CNN模型中提取的数据。然后,我们提出两种在不同情况下计算熵的方法。为了从信息论视角对CNN的基本构建模块(例如卷积层、池化层、归一化层)提供基础性理解,我们开发了一个可视化分析系统CNNSlicer。CNNSlicer允许用户交互式地探索模型内部信息量的变化。通过在广泛使用的基准数据集(MNIST和CIFAR-10)上的案例研究,我们证明了该系统在打开CNN黑盒方面的有效性。

英文摘要:

Despite the great success of Convolutional Neural Networks (CNNs) in Computer Vision and Natural Language Processing, the working mechanism behind CNNs is still under extensive discussions and research. Driven by a strong demand for the theoretical explanation of neural networks, some researchers utilize information theory to provide insight into the black box model. However, to the best of our knowledge, employing information theory to quantitatively analyze and qualitatively visualize neural networks has not been extensively studied in the visualization community. In this paper, we combine information entropies and visualization techniques to shed light on how CNN works. Specifically, we first introduce a data model to organize the data that can be extracted from CNN models. Then we propose two ways to calculate entropy under different circumstances. To provide a fundamental understanding of the basic building blocks of CNNs (e.g., convolutional layers, pooling layers, normalization layers) from an information-theoretic perspective, we develop a visual analysis system, CNNSlicer. CNNSlicer allows users to interactively explore the amount of information changes inside the model. With case studies on the widely used benchmark datasets (MNIST and CIFAR-10), we demonstrate the effectiveness of our system in opening the blackbox of CNNs.

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

↑