Learning useful representations for shifting tasks and distributions
Not on the map. Why?
Abstract (arXiv)
Does the dominant approach to learn representations (as a side effect of optimizing an expected cost for a single training distribution) remain a good approach when we are dealing with multiple distributions? Our thesis is that such scenarios are better served by representations that are richer than those obtained with a single optimization episode. We support this thesis with simple theoretical arguments and with experiments utilizing an apparently na"ive ensembling technique: concatenating the representations obtained from multiple training episodes using the same data, model, algorithm, and hyper-parameters, but different random seeds. These independently trained networks perform similarly. Yet, in a number of scenarios involving new distributions, the concatenated representation performs substantially better than an equivalently sized network trained with a single training run. This proves that the representations constructed by multiple training episodes are in fact different. Although their concatenation carries little additional information about the training task under the training distribution, it becomes substantially more informative when tasks or distributions change. Meanwhile, a single training episode is unlikely to yield such a redundant representation because the optimization process has no reason to accumulate features that do not incrementally improve the training performance.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 49 of its 81 references that are in the map. First posted on arXiv 14 Dec 2022.
Builds on
49 of 81 references.
References · 49
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2013Intriguing properties of neural networkscited by 5,114
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2016Matching Networks for One Shot Learningcited by 2,246
37 more
- 2020Shortcut Learning in Deep Neural Networkscited by 1,928
- 2019Invariant Risk Minimizationcited by 1,832
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
- 2017Learning to Compare: Relation Network for Few-Shot Learningcited by 1,139
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2019A Closer Look at Few-shot Classificationcited by 679
- 2021How to train your ViT? Data, Augmentation, and Regularization in Vision Transformerscited by 562
- 2017Opening the Black Box of Deep Neural Networks via Informationcited by 494
- 2018Meta-Learning with Latent Embedding Optimizationcited by 476
- 2020Self-Challenging Improves Cross-Domain Generalizationcited by 426
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
- 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 374
- 2019Meta-Dataset: A Dataset of Datasets for Learning to Learn from Few Examplescited by 312
- 2021Ensemble deep learning: A reviewcited by 283
- 2020Gradient Starvation: A Learning Proclivity in Neural Networkscited by 262
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2021On Calibration and Out-of-domain Generalizationcited by 134
- 2019Implicit regularization for deep neural networks driven by an Ornstein-Uhlenbeck like processcited by 123
- 2021Evading the Simplicity Bias: Training a Diverse Set of Models Discovers Solutions with Superior OOD Generalizationcited by 83
- 2022Vision Models Are More Robust And Fair When Pretrained On Uncurated Images Without Supervisioncited by 80
- 2022SGD with Large Step Sizes Learns Sparse Featurescited by 66
- 2021No One Representation to Rule Them All: Overlapping Features of Training Methodscited by 57
- 2021Universal Representation Learning from Multiple Domains for Few-shot Classificationcited by 56
- 2017Universal representations:The missing link between faces, text, planktons, and cat breedscited by 45
- 2022Rich Feature Construction for the Optimization-Generalization Dilemmacited by 34
- 2021Few-shot Image Classification: Just Use a Library of Pre-trained Feature Extractors and a Simple Classifiercited by 32
- 2022Universal Representations: A Unified Look at Multiple Task and Domain Learningcited by 23
- 2021Conditional Meta-Learning of Linear Representationscited by 12
Built on by
Citing papers per year: 2022: 1; 2023: 3; 2024: 6; 2025: 4.
14 papers.
Sort:
- 2022Model Ratatouille: Recycling Diverse Models for Out-of-Distribution Generalizationcited by 97
- 2023Understanding and Improving Feature Learning for Out-of-Distribution Generalizationcited by 35
- 2023Towards Stable Backdoor Purification through Feature Shift Tuningcited by 30
- 2023Spurious Feature Diversification Improves Out-of-distribution Generalizationcited by 26
- 2024A Second-Order Perspective on Model Compositionality and Incremental Learningcited by 18
- 2024Dual Consolidation for Pre-Trained Model-Based Domain-Incremental Learningcited by 16
- 2024Empowering Graph Invariance Learning with Deep Spurious Infomaxcited by 14
- 2024Feature contamination: Neural networks learn uncorrelated features and fail to generalizecited by 11
- 2024Is Multiple Object Tracking a Matter of Specialization?cited by 5
- 2024Fine-tuning with Very Large Dropoutcited by 3
- 2025Leveraging Registers in Vision Transformers for Robust Adaptationcited by 3
- 2025These Are Not All the Features You Are Looking For: A Fundamental Bottleneck in Supervised Pretrainingcited by 0
- 2025High-Rate Mixout: Revisiting Mixout for Robust Domain Generalization8 Oct 2025
- 2025Can multimodal representation learning by alignment preserve modality-specific information?22 Sep 2025
- 2025These Are Not All the Features You Are Looking For: A Fundamental Bottleneck in Supervised Pretraining23 Jun 2025
- 2025Leveraging Registers in Vision Transformers for Robust Adaptation8 Jan 2025
- 2024Is Multiple Object Tracking a Matter of Specialization?1 Nov 2024
- 2024Dual Consolidation for Pre-Trained Model-Based Domain-Incremental Learning1 Oct 2024
- 2024Empowering Graph Invariance Learning with Deep Spurious Infomax13 Jul 2024
- 2024Feature contamination: Neural networks learn uncorrelated features and fail to generalize5 Jun 2024
- 2024A Second-Order Perspective on Model Compositionality and Incremental Learning25 May 2024
- 2024Fine-tuning with Very Large Dropout1 Mar 2024
- 2023Towards Stable Backdoor Purification through Feature Shift Tuning3 Oct 2023
- 2023Spurious Feature Diversification Improves Out-of-distribution Generalization29 Sep 2023
- 2022Model Ratatouille: Recycling Diverse Models for Out-of-Distribution Generalizationcited by 97
- 2023Understanding and Improving Feature Learning for Out-of-Distribution Generalizationcited by 35
- 2023Towards Stable Backdoor Purification through Feature Shift Tuningcited by 30
- 2023Spurious Feature Diversification Improves Out-of-distribution Generalizationcited by 26
- 2024A Second-Order Perspective on Model Compositionality and Incremental Learningcited by 18
- 2024Dual Consolidation for Pre-Trained Model-Based Domain-Incremental Learningcited by 16
- 2024Empowering Graph Invariance Learning with Deep Spurious Infomaxcited by 14
- 2024Feature contamination: Neural networks learn uncorrelated features and fail to generalizecited by 11
- 2024Is Multiple Object Tracking a Matter of Specialization?cited by 5
- 2024Fine-tuning with Very Large Dropoutcited by 3
- 2025Leveraging Registers in Vision Transformers for Robust Adaptationcited by 3
- 2025These Are Not All the Features You Are Looking For: A Fundamental Bottleneck in Supervised Pretrainingcited by 0
12 most cited of 14 All 14 (JSON)
- 2025High-Rate Mixout: Revisiting Mixout for Robust Domain Generalization8 Oct 2025
- 2025Can multimodal representation learning by alignment preserve modality-specific information?22 Sep 2025
- 2025These Are Not All the Features You Are Looking For: A Fundamental Bottleneck in Supervised Pretraining23 Jun 2025
- 2025Leveraging Registers in Vision Transformers for Robust Adaptation8 Jan 2025
- 2024Is Multiple Object Tracking a Matter of Specialization?1 Nov 2024
- 2024Dual Consolidation for Pre-Trained Model-Based Domain-Incremental Learning1 Oct 2024
- 2024Empowering Graph Invariance Learning with Deep Spurious Infomax13 Jul 2024
- 2024Feature contamination: Neural networks learn uncorrelated features and fail to generalize5 Jun 2024
- 2024A Second-Order Perspective on Model Compositionality and Incremental Learning25 May 2024
- 2024Fine-tuning with Very Large Dropout1 Mar 2024
- 2023Towards Stable Backdoor Purification through Feature Shift Tuning3 Oct 2023
- 2023Spurious Feature Diversification Improves Out-of-distribution Generalization29 Sep 2023
12 newest of 14 All 14 (JSON)