深度网络隐式偏好低秩结构,影响分类器性能
The Implicit Bias of Depth: From Neural Collapse to Softmax Codes

- 无正则化深度模型中,梯度下降与深度共同诱导低秩偏置
- 深度使神经坍缩的吸引域缩小,促进软最大码解法
- 适合研究深度学习优化机制与隐式偏差的学者
神经坍缩(NC)描述了训练后分类器特征与权重中出现的结构化几何形态。近期理论表明,在深层架构中NC可能次优,归因于L2正则化带来的显式低秩偏置。本文研究无约束特征模型(UFM),即具有正交输入的深层线性网络,在无正则化条件下训练,以分离梯度下降和深度对NC的影响。结果表明,深度本身会引入隐式低秩偏置:低秩矩阵在连续乘法中传播范数更高效,从而促进低秩替代解。这些替代解对应于软最大码——此前在宽度瓶颈网络中发现的最大间距解。通过谱初始化分析训练动态,我们识别出奇异值早期的排斥现象驱动低秩结构出现,并刻画了深度如何缩小NC的吸引域。最后,我们发现某些效应方向相反:对于随机初始化网络,增加宽度会偏向更高秩解。本研究首次提供了无正则化多类交叉熵训练下深层UFM的渐近与动态隐式偏差表征。
原文摘要 · Abstract (English)
Neural collapse (NC) describes the structured geometry that emerges in the features and weights of trained classifiers. Recent theory suggests NC can be suboptimal in deep architectures, attributing this to an explicit low-rank bias from L2 regularization. We study the deep unconstrained feature model (UFM)-equivalent to a deep linear network with orthogonal inputs-trained without regularization, to isolate how gradient descent and depth alone shape NC. We show that depth induces an implicit low-rank bias: low-rank matrices propagate norm more efficiently through successive multiplications, promoting low-rank alternatives to NC. These alternatives, we argue, correspond to softmax codes: max-margin solutions previously found in width-bottlenecked networks. Analyzing training dynamics under spectral initialization, we identify an early-time repulsion among singular values that drives low-rank emergence, and characterize how depth shrinks NC's basin of attraction. Finally, we show that some effects act in the opposite direction: for randomly initialized networks, increasing width biases training toward higher-rank solutions. Our results provide the first asymptotic and dynamic characterization of implicit bias in deep UFMs trained with unregularized multiclass cross-entropy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。