Progressive Feedforward Collapse conjecture for ResNet training

Let Hl\boldsymbol{H}^{l} denote the feature matrix at layer ll, with class means hkl\boldsymbol{h}_k^l and global mean hGl\boldsymbol{h}_G^l. For each layer, measure feature collapse toward class means, collapse of centered class means toward the simplex equiangular tight frame (ETF), and nearest class center (NCC) accuracy using the three progressive feedforward collapse metrics PFC1\mathcal{PFC}_1, PFC2\mathcal{PFC}_2, and PFC3\mathcal{PFC}_3. Here, PFC2\mathcal{PFC}_2 measures the distance between the normalized Gram matrix of the centered class means and the simplex ETF, while PFC3\mathcal{PFC}_3 is the NCC accuracy.

Progressive Feedforward Collapse conjecture. At the terminal phase of ResNet training, neural collapse emerges in the last-layer features. Before the effective depth, there exists an order in the degree of collapse of each layer, measured by the three PFC metrics: the features of each layer progressively collapse to their class means; the centered class means of each layer's features progressively collapse to the simplex ETF; and the NCC accuracy of each layer progressively collapses to 11. During training, the metrics at each layer start at random initialization and gradually converge to the final order.

This conjecture proposes a layerwise progression of neural-collapse geometry through ResNet forward propagation, extending last-layer neural collapse to intermediate layers. The supplied text does not provide evidence that the conjecture has been proved or refuted.

Sources & referencesView supporting material

Primary source

Sicong Wang, Kuo Gai and Shihua Zhang, “Progressive Feedforward Collapse of ResNet Training”, arXiv:2405.00985 (2024).

Progress summary

Never refreshed

Nothing recorded yet. Refresh searches the literature and the public web for attempts on this problem, and writes the first summary here.

Solutions 0

No solutions have been posted yet.