从神经坍缩视角解析DEQ模型表示,揭示其在不平衡数据下的优势
Understanding Representation of Deep Equilibrium Models from Neural Collapse Perspective
- 用神经坍缩理论分析DEQ的特征表示机制
- 在不平衡数据下,特征收敛到等角紧框架顶点
- 适合研究隐式网络理论性质的研究者参考
深度均衡模型(DEQ)作为典型的隐式神经网络,具有内存效率高、性能优越的特点。然而,关于DEQ表示能力的理论分析仍较有限。本文借助神经坍缩($/mathcal{NC}$)这一工具,系统分析了DEQ在平衡与非平衡条件下的表示特性。$/mathcal{NC}$ 揭示了类别特征与分类器权重的几何结构,在传统显式网络中已被广泛研究,但在隐式网络中尚无深入探讨。理论上证明:在平衡条件下,DEQ 中存在 $/mathcal{NC}$ 现象;在非平衡设置下,尽管出现少数类坍缩,但DEQ仍优于显式网络——其提取特征可收敛至单纯形等角紧框架的顶点,并在弱条件下具备自对偶性,凸显其处理不平衡数据的优势。实验验证了上述理论结论在平衡与非平衡场景中的有效性。
原文摘要 · Abstract (English)
Deep Equilibrium Model (DEQ), which serves as a typical implicit neural network, emphasizes their memory efficiency and competitive performance compared to explicit neural networks. However, there has been relatively limited theoretical analysis on the representation of DEQ. In this paper, we utilize the Neural Collapse ($\mathcal{NC}$) as a tool to systematically analyze the representation of DEQ under both balanced and imbalanced conditions. $\mathcal{NC}$ is an interesting phenomenon in the neural network training process that characterizes the geometry of class features and classifier weights. While extensively studied in traditional explicit neural networks, the $\mathcal{NC}$ phenomenon has not received substantial attention in the context of implicit neural networks. We theoretically show that $\mathcal{NC}$ exists in DEQ under balanced conditions. Moreover, in imbalanced settings, despite the presence of minority collapse, DEQ demonstrated advantages over explicit neural networks. These advantages include the convergence of extracted features to the vertices of a simplex equiangular tight frame and self-duality properties under mild conditions, highlighting DEQ's superiority in handling imbalanced datasets. Finally, we validate our theoretical analyses through experiments in both balanced and imbalanced scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。