Separation and Concentration in Deep Networks
John Zarka, Florentin Guth, Stéphane Mallat
Abstract
Numerical experiments demonstrate that deep neural network classifiers progressively separate class distributions around their mean, achieving linear separability on the training set, and increasing the Fisher discriminant ratio. We explain this mechanism with two types of operators. We prove that a rectifier without biases applied to sign-invariant tight frames can separate class means and increase Fisher ratios. On the opposite, a soft-thresholding on tight frames can reduce within-class variabilities while preserving class means. Variance reduction bounds are proved for Gaussian mixture models. For image classification, we show that separation of class means can be achieved with rectified wavelet tight frames that are not learned. It defines a scattering transform. Learning convolutional tight frames along scattering channels and applying a soft-thresholding reduces within-class variabilities. The resulting scattering network reaches the classification accuracy of ResNet-18 on CIFAR-10 and ImageNet, with fewer layers and no learned biases.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f1e27c8a-6553-4149-8315-ab09e0e6a718Cited by top-tier papers14
- A Geometric Analysis of Neural Collapse with Unconstrained FeaturesZhihui Zhu, Tianyu Ding, Jinxin Zhou, Xiao Li et al.NeurIPS 2021 · 303 citations
- Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?Yibo Yang, Shixiang Chen, Xiangtai Li, Liang Xie et al.NeurIPS 2022 · 144 citations
- Extended Unconstrained Features Model for Exploring Deep Neural CollapseTom Tirer, Joan BrunaICML 2022 · 118 citations
- Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features ModelPeter Súkeník, Marco Mondelli, Christoph H. LampertNeurIPS 2023 · 51 citations
- Improving Self-Supervised Learning by Characterizing Idealized RepresentationsYann Dubois, Stefano Ermon, Tatsunori B. Hashimoto, Percy LiangNeurIPS 2022 · 50 citations
Builds on2
- Robust And Interpretable Blind Image Denoising Via Bias-Free Convolutional Neural NetworksSreyas Mohan, Zahra Kadkhodaie, Eero P. Simoncelli, Carlos Fernandez-GrandaICLR 2020 · 154 citations
- Deep Network Classification by Scattering and Homotopy Dictionary LearningJohn Zarka, Louis Thiry, Tomás Angles, Stéphane MallatICLR 2020 · 43 citations
Related papers
- Phase Collapse in Neural NetworksFlorentin Guth, John Zarka, Stéphane MallatICLR 2022 · 8 citations
- Unlocking Better Closed-Set Alignment Based on Neural Collapse for Open-Set RecognitionChaohua Li, Enhao Zhang, Chuanxing Geng, Songcan ChenAAAI 2025 · 1 citation
- Neural Collapse for Cross-entropy Class-Imbalanced Learning with Unconstrained ReLU Features ModelHien Dang, Tho Tran Huu, Tan Minh Nguyen, Nhat HoICML 2024 · 19 citations
- Neural Collapse in Deep Linear Networks: From Balanced to Imbalanced DataHien Dang, Tho Tran Huu, Stanley J. Osher, Hung Tran-The et al.ICML 2023 · 44 citations
- Neural Collapse in Multi-label Learning with Pick-all-label LossPengyu Li, Xiao Li, Yutong Wang, Qing QuICML 2024 · 15 citations
