Neural Representations Reveal Distinct Modes of Class Fitting in Residual Convolutional Networks
Michal Jamroz, Marcin Kurdziel
Abstract
We leverage probabilistic models of neural representations to investigate how residual networks fit classes. To this end, we estimate class-conditional density models for representations learned by deep ResNets. We then use these models to characterize distributions of representations across learned classes. Surprisingly, we find that classes in the investigated models are not fitted in an uniform way. On the contrary: we uncover two groups of classes that are fitted with markedly different distributions of representations. These distinct modes of class-fitting are evident only in the deeper layers of the investigated models, indicating that they are not related to low-level image features. We show that the uncovered structure in neural representations correlate with memorization of training examples and adversarial robustness. Finally, we compare class-conditional distributions of neural representations between memorized and typical examples. This allows us to uncover where in the network structure class labels arise for memorized and standard inputs.
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 741ce57e-b145-4b04-ad70-ed8359b8ac25Builds on5
- MLP-Mixer: An all-MLP Architecture for VisionIlya O. Tolstikhin, Neil Houlsby, Alexander Kolesnikov, Lucas Beyer et al.NeurIPS 2021 · 3,862 citations
- Sharpness-aware Minimization for Efficiently Improving GeneralizationPierre Foret, Ariel Kleiner, Hossein Mobahi, Behnam NeyshaburICLR 2021 · 1,861 citations
- What Neural Networks Memorize and Why: Discovering the Long Tail via Influence EstimationVitaly Feldman, Chiyuan ZhangNeurIPS 2020 · 674 citations
- Does learning require memorization? a short tale about a long tailVitaly FeldmanSTOC 2020 · 28 citations
- A Bayesian Nonparametrics View into Deep RepresentationsMichal Jamroz, Marcin Kurdziel, Mateusz OpalaNeurIPS 2020 · 1 citation
Related papers
- On the Functional Similarity of Robust and Non-Robust Neural RepresentationsAndrás Balogh, Márk JelasityICML 2023 · 4 citations
- Identifying and Understanding Cross-Class Features in Adversarial TrainingZeming Wei, Steven Y. Guo, Yisen WangICML 2025
- Understanding Invariance via Feedforward Inversion of Discriminatively Trained ClassifiersPiotr Teterwak, Chiyuan Zhang, Dilip Krishnan, Michael C. MozerICML 2021 · 11 citations
- On the Clean Generalization and Robust Overfitting in Adversarial Training from Two Theoretical Views: Representation Complexity and Training DynamicsBinghui Li, Yuanzhi LiICML 2025
- Understanding Robust Learning through the Lens of Representation SimilaritiesChristian Cianfarani, Arjun Nitin Bhagoji, Vikash Sehwag, Ben Y. Zhao et al.NeurIPS 2022 · 20 citations
