用物理重整化群解释神经网络如何提取数据主特征
Interpreting FCDNNs via RG on Exponential Family

- 将神经网络训练等同于连续场的重整化群计算
- 训练后网络特征参数等于输入数据的重整化群不动点
- 为深度网络在真实数据上的优异表现提供理论解释
本文通过构建统计物理中的重整化群(RG)方法与深度神经网络(DNN)训练过程之间的对应关系,建立深度学习可解释性理论。此前已在一维伊辛模型上验证该框架;本文将其推广至连续输入数据情形,选取指数族分布作为代表性数据分布。证明当全连接DNN参数达到最优时,其特征层输出的特征参数等于输入数据在连续场下经RG变换后的不动点。这表明DNN训练过程等价于对这类数据的RG计算,因此网络能像RG一样自动提取关键特征。该等价性进一步验证了所建框架的合理性,为DNN在真实数据上的卓越性能提供了理论支持。
原文摘要 · Abstract (English)
We consider establishing the interpretability theory of deep learning through constructing a corresponding relationship between the renormalization group (RG) method in statistical physics and the training process of deep neural networks (DNNs). We have proved the constructed relationship using the one-dimensional Ising model as the input data. In this paper we generalize our results to the case of continuous input data, which is a necessary preparation for applying the corresponding framework to real-world data. To be representative, we consider a class of data distribution in the exponential family. We prove that when the parameters of fully connected (FC) DNNs achieve their optimal value after training, the characteristic parameters of the feature layer output of DNNs are equal to the fixed points of the characteristic parameters of input data under RG method for continuous fields. This conclusion shows that the training process of DNNs is equivalent to RG calculation on this kind of data and therefore the network can extract main features from the input data just like RG. Also, the equivalence further validates the correspondence framework we have established, providing an explanation for the outstanding performance of DNNs on real-world data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。