On the Strong Correlation Between Model Invariance and Generalization
Weijian Deng, Stephen Gould, Liang Zheng
Abstract
Generalization and invariance are two essential properties of any machine learning model. Generalization captures a model's ability to classify unseen data while invariance measures consistency of model predictions on transformations of the data. Existing research suggests a positive relationship: a model generalizing well should be invariant to certain visual factors. Building on this qualitative implication we make two contributions. First, we introduce effective invariance (EI), a simple and reasonable measure of model invariance which does not rely on image labels. Given predictions on a test image and its transformed version, EI measures how well the predictions agree and with what level of confidence. Second, using invariance scores computed by EI, we perform large-scale quantitative correlation studies between generalization and invariance, focusing on rotation and grayscale transformations. From a model-centric view, we observe generalization and invariance of different models exhibit a strong linear relationship, on both in-distribution and out-of-distribution datasets. From a dataset-centric view, we find a certain model's accuracy and invariance linearly correlated on different test sets. Apart from these major findings, other minor but interesting insights are also discussed. Preprint. Under review.
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 e080a2cd-ec56-4d68-999d-d8503b2aa7d0Cited by top-tier papers12
- GNNEvaluator: Evaluating GNN Performance On Unseen Graphs Without LabelsXin Zheng, Miao Zhang, Chunyang Chen, Soheila Molaei et al.NeurIPS 2023 · 30 citations
- Contrastive Tuning: A Little Help to Make Masked Autoencoders ForgetJohannes Lehner, Benedikt Alkin, Andreas Fürst, Elisabeth Rumetshofer et al.AAAI 2024 · 28 citations
- Confidence and Dispersity Speak: Characterizing Prediction Matrix for Unsupervised Accuracy EstimationWeijian Deng, Yumin Suh, Stephen Gould, Liang ZhengICML 2023 · 23 citations
- Energy-based Automated Model EvaluationRu Peng, Heming Zou, Haobo Wang, Yawen Zeng et al.ICLR 2024 · 18 citations
- Online GNN Evaluation Under Test-time Graph Distribution ShiftsXin Zheng, Dongjin Song, Qingsong Wen, Bo Du et al.ICLR 2024 · 16 citations
Builds on40
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- A ConvNet for the 2020sZhuang Liu, Hanzi Mao, Chao-Yuan Wu, Christoph Feichtenhofer et al.CVPR 2022 · 6,782 citations
- CutMix: Regularization Strategy to Train Strong Classifiers With Localizable FeaturesSangdoo Yun, Dongyoon Han, Sanghyuk Chun, Seong Joon Oh et al.ICCV 2019 · 5,843 citations
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and ConfidenceKihyuk Sohn, David Berthelot, Nicholas Carlini, Zizhao Zhang et al.NeurIPS 2020 · 5,129 citations
Related papers
- The Lie Derivative for Measuring Learned EquivarianceNate Gruver, Marc Anton Finzi, Micah Goldblum, Andrew Gordon WilsonICLR 2023 · 6 citations
- In What Ways Are Deep Neural Networks Invariant and How Should We Measure This?Henry Kvinge, Tegan Emerson, Grayson Jorgenson, Scott Vasquez et al.NeurIPS 2022 · 15 citations
- Provably Strict Generalisation Benefit for Equivariant ModelsBryn Elesedy, Sheheryar ZaidiICML 2021 · 100 citations
- Grounding inductive biases in natural images: invariance stems from variations in dataDiane Bouchacourt, Mark Ibrahim, Ari S. MorcosNeurIPS 2021 · 32 citations
- Background Invariance Testing According to Semantic ProximityZukang Liao, Min ChenICCV 2025
