arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2008.08186cs.LGcs.CVstat.ML
收藏

深度学习训练末期神经崩溃的发生率

Prevalence of Neural Collapse during the terminal phase of deep learning training

Stanford University(斯坦福大学);Cornell University(康奈尔大学)

Vardan Papyan, X. Y. Han, David L. Donoho

摘要:

训练分类深度网络的现代实践涉及训练的终端阶段(TPT),该阶段开始于训练误差首次消失的时期;在TPT期间,训练误差实际上保持为零,而训练损失被推向零。针对三种原型深网架构和七个规范分类数据集的TPT直接测量暴露了一种普遍存在的归纳偏差,我们称之为神经崩溃,涉及四种深度相互关联的现象:(NC 1)最后一层训练激活的跨示例类内变异性崩溃为零,因为单个激活本身崩溃为它们的类均值;(NC 3)直到重新缩放,最后一层分类器塌缩到类均值,或者换句话说,塌缩到单纯形ETF,即,塌缩到自对偶配置;(NC 4)对于给定的激活,分类器的决策简化为简单地选择具有最接近的训练类均值的任何类,即,最接近类中心(NCC)决策规则。由TPT引起的对称和非常简单的几何结构赋予重要的益处,包括更好的泛化性能、更好的鲁棒性和更好的可解释性。

英文摘要:

Modern practice for training classification deepnets involves a Terminal Phase of Training (TPT), which begins at the epoch where training error first vanishes; During TPT, the training error stays effectively zero while training loss is pushed towards zero. Direct measurements of TPT, for three prototypical deepnet architectures and across seven canonical classification datasets, expose a pervasive inductive bias we call Neural Collapse, involving four deeply interconnected phenomena: (NC1) Cross-example within-class variability of last-layer training activations collapses to zero, as the individual activations themselves collapse to their class-means; (NC2) The class-means collapse to the vertices of a Simplex Equiangular Tight Frame (ETF); (NC3) Up to rescaling, the last-layer classifiers collapse to the class-means, or in other words to the Simplex ETF, i.e. to a self-dual configuration; (NC4) For a given activation, the classifier's decision collapses to simply choosing whichever class has the closest train class-mean, i.e. the Nearest Class Center (NCC) decision rule. The symmetric and very simple geometry induced by the TPT confers important benefits, including better generalization performance, better robustness, and better interpretability.

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

↑