An Information Theoretic Interpretation to Deep Neural Networks

• Main Authors: Xu, Xiangxiang (Author), Huang, Shao-Lun (Author), Zheng, Lizhong (Author), Wornell, Gregory W. (Author)
• Format: Article
• Language: English
• Published: Multidisciplinary Digital Publishing Institute, 2022-01-20T19:32:08Z.
• Subjects: Article
• Online Access: Get fulltext

 With the unprecedented performance achieved by deep learning, it is commonly believed that deep neural networks (DNNs) attempt to extract informative features for learning tasks. To formalize this intuition, we apply the local information geometric analysis and establish an information-theoretic framework for feature selection, which demonstrates the information-theoretic optimality of DNN features. Moreover, we conduct a quantitative analysis to characterize the impact of network structure on the feature extraction process of DNNs. Our investigation naturally leads to a performance metric for evaluating the effectiveness of extracted features, called the H-score, which illustrates the connection between the practical training process of DNNs and the information-theoretic framework. Finally, we validate our theoretical results by experimental designs on synthesized data and the ImageNet dataset.