arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02697cs.LGstat.ML
收藏

针对高维高斯混合,深度平衡模型几乎等价于不那么深的显式模型

Deep Equilibrium Models are Almost Equivalent to Not-so-deep Explicit Models for High-dimensional Gaussian Mixtures

School of Electronic Information and Communications, Huazhong University of Science and Technology(华中科技大学电子信息与通信学院);Ezhou Industrial Technology Research Institute, Huazhong University of Science and Technology(华中科技大学鄂州工业技术研究院);Department of CSE, Shanghai Jiao Tong University(上海交通大学计算机科学与工程系);China-Austria Belt and Road Joint Laboratory on AI and AM, Hangzhou Dianzi University(杭州电子科技大学中奥人工智能与增材制造一带一路联合实验室);Center for Applied Statistics and School of Statistics, Renmin University of China(中国人民大学应用统计中心与统计学院)

Zenan Ling, Longbo Li, Zhanbo Feng, Yixuan Zhang, Feng Zhou, Robert C. Qiu, Zhenyu Liao

AI总结:

该研究借助随机矩阵理论分析高维高斯混合输入下隐式深度平衡模型(DEQ)的共轭核与神经切线核谱特性,证明其仅由含四个非线性方程的系统决定,且可通过精心设计的浅层显式网络实现等价核,该结论经实证也适用于真实数据集。

摘要:

深度平衡模型(Deep Equilibrium Models, DEQs)作为一类典型的隐式神经网络,已在诸多任务中展现出显著成效。然而,目前学界对隐式DEQ与显式神经网络模型之间的关联和差异仍缺乏理论层面的认知。本文借助随机矩阵理论(random matrix theory, RMT)的最新进展,针对输入数据采样自高维高斯混合的场景,对隐式DEQ的共轭核(conjugate kernel, CK)矩阵和神经切线核(neural tangent kernel, NTK)矩阵的特征谱展开了深入分析。我们证明,在此设定下,这些隐式CK和NTK的谱特性依赖于DEQ的激活函数与初始权重方差,但这种依赖仅通过一个由四个非线性方程构成的系统体现。作为该理论结果的直接推论,我们证明可以精心设计一个浅层显式网络,使其产生与给定DEQ完全相同的CK或NTK。尽管该理论是针对高斯混合数据推导得出的,但实验结果表明,所提出的理论与设计原则同样适用于常用的真实世界数据集。

英文摘要:

Deep equilibrium models (DEQs), as a typical implicit neural network, have demonstrated remarkable success on various tasks. There is, however, a lack of theoretical understanding of the connections and differences between implicit DEQs and explicit neural network models. In this paper, leveraging recent advances in random matrix theory (RMT), we perform an in-depth analysis on the eigenspectra of the conjugate kernel (CK) and neural tangent kernel (NTK) matrices for implicit DEQs, when the input data are drawn from a high-dimensional Gaussian mixture. We prove, in this setting, that the spectral behavior of these Implicit-CKs and NTKs depend on the DEQ activation function and initial weight variances, but only via a system of four nonlinear equations. As a direct consequence of this theoretical result, we demonstrate that a shallow explicit network can be carefully designed to produce the same CK or NTK as a given DEQ. Despite derived here for Gaussian mixture data, empirical results show the proposed theory and design principle also apply to popular real-world datasets.

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

↑