Adaptive Risk Minimization: Learning to Adapt to Domain Shift
Not on the map. Why?
Abstract (arXiv)
A fundamental assumption of most machine learning algorithms is that the training and test data are drawn from the same underlying distribution. However, this assumption is violated in almost all practical applications: machine learning systems are regularly tested under distribution shift, due to changing temporal correlations, atypical end users, or other factors. In this work, we consider the problem setting of domain generalization, where the training data are structured into domains and there may be multiple test time shifts, corresponding to new domains or domain distributions. Most prior methods aim to learn a single robust model or invariant feature space that performs well on all domains. In contrast, we aim to learn models that adapt at test time to domain shift using unlabeled test points. Our primary contribution is to introduce the framework of adaptive risk minimization (ARM), in which models are directly optimized for effective adaptation to shift by learning to adapt on the training domains. Compared to prior methods for robustness, invariance, and adaptation, ARM methods provide performance gains of 1-4% test accuracy on a number of image classification problems exhibiting domain shift.
Key facts. Cited by 183 papers in the map (data through 14 Oct 2026). First posted on arXiv 6 Jul 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 7; 2022: 26; 2023: 57; 2024: 43; 2025: 25; 2026: 25.
183 papers.
Sort:
By area:
- 2021MEMO: Test Time Robustness via Adaptation and Augmentationcited by 377
- 2023A Comprehensive Survey on Test-Time Adaptation under Distribution Shiftscited by 359
- 2022Surgical Fine-Tuning Improves Adaptation to Distribution Shiftscited by 205
- 2024Efficient Test-Time Adaptation of Vision-Language Modelscited by 141
- 2022Domain Generalization by Mutual-Information Regularization with Pre-trained Modelscited by 133
- 2021The Norm Must Go On: Dynamic Unsupervised Domain Adaptation by Normalizationcited by 131
- 2022Sparse Mixture-of-Experts are Domain Generalizable Learnerscited by 85
- 2022Back to the Source: Diffusion-Driven Test-Time Adaptationcited by 71
- 2022Probable Domain Generalization via Quantile Risk Minimizationcited by 66
- 2023Advances and Challenges in Meta-Learning: A Technical Reviewcited by 65
- 2023On Pitfalls of Test-Time Adaptationcited by 64
- 2023Feed Two Birds with One Scone: Exploiting Wild Data for Both Out-of-Distribution Generalization and Detectioncited by 57
38 more
- 2022An Efficient Domain-Incremental Learning Approach to Drive in All Weather Conditionscited by 54
- 2022Meta-DMoE: Adapting to Domain Shift by Meta-Distillation from Mixture-of-Expertscited by 50
- 2023AdaNPC: Exploring Non-Parametric Classifier for Test-Time Adaptationcited by 48
- 2023Improved Test-Time Adaptation for Domain Generalizationcited by 47
- 2023MetaGCD: Learning to Continually Learn in Generalized Category Discoverycited by 42
- 2023Leveraging Vision-Language Models for Improving Domain Generalization in Image Classificationcited by 41
- 2022Self-Distilled Vision Transformer for Domain Generalizationcited by 35
- 2023Free Lunch for Domain Adversarial Training: Environment Label Smoothingcited by 33
- 2023Leveraging sparse and shared feature activations for disentangled representation learningcited by 32
- 2023Test-Time Domain Adaptation by Learning Domain-Aware Batch Normalizationcited by 31
- 2021Iterative Feature Matching: Toward Provable Domain Generalization with Logarithmic Environmentscited by 30
- 2021Meta-Calibration: Learning of Model Calibration Using Differentiable Expected Calibration Errorcited by 29
- 2023Domain Generalization in Computational Pathology: Survey and Guidelinescited by 29
- 2023Recurrence without Recurrence: Stable Video Landmark Detection with Deep Equilibrium Modelscited by 27
- 2023Decompose, Adjust, Compose: Effective Normalization by Playing with Frequency for Domain Generalizationcited by 26
- 2023Discovering environments with XRMcited by 26
- 2024FuRL: Visual-Language Models as Fuzzy Rewards for Reinforcement Learningcited by 26
- 2023Federated Domain Generalization: A Surveycited by 25
- 2023Learngene: Inheriting Condensed Knowledge from the Ancestry Model to Descendant Modelscited by 24
- 2024Rethinking Multi-domain Generalization with A General Learning Objectivecited by 23
- 2024Adapting to Distribution Shift by Visual Domain Prompt Generationcited by 23
- 2023MADG: Margin-based Adversarial Learning for Domain Generalizationcited by 22
- 2024DGMamba: Domain Generalization via Generalized State Space Modelcited by 21
- 2021Contrastive ACE: Domain Generalization Through Alignment of Causal Mechanismscited by 20
- 2022Causal Balancing for Domain Generalizationcited by 20
- 2023DART: Diversify-Aggregate-Repeat Training Improves Generalization of Neural Networkscited by 20
- 2023DomainAdaptor: A Novel Approach to Test-time Adaptationcited by 20
- 2023ERM++: An Improved Baseline for Domain Generalizationcited by 18
- 2023Benchmarking Test-Time Adaptation against Distribution Shifts in Image Classificationcited by 17
- 2023Rethinking Domain Generalization: Discriminability and Generalizabilitycited by 16
- 2024GM-DF: Generalized Multi-Scenario Deepfake Detectioncited by 16
- 2022Domain-Specific Risk Minimization for Out-of-Distribution Generalizationcited by 15
- 2023Energy-Based Test Sample Adaptation for Domain Generalizationcited by 14
- 2024Domain Generalisation via Imprecise Learningcited by 14
- 2025Learning to Adapt Frozen CLIP for Few-Shot Test-Time Domain Adaptationcited by 14
- 2023Multimodal Federated Learning in Healthcare: a Reviewcited by 13
- 2024HYPO: Hyperspherical Out-of-Distribution Generalizationcited by 13
- 2024Robust Fast Adaptation from Adversarially Explicit Task Distribution Generationcited by 13
50 most cited of 183 All 183 (JSON)
- 2026Domain Generalization under Sampling Pattern Shifts in Irregular Time Series27 Sep 2026
- 2026PAPT++: Risk-Aware Adversarial Tuning and Generation for Single Domain Generalization4 Sep 2026
- 2026PatchGen: Learning Soft Intra-Image Predictive Subsets for Visual Generalization13 Aug 2026
- 2026PIcsC: Partitioning-Induced Covariate Shift Correction28 Jul 2026
- 2026Projection Pursuit CPCANet for Domain Generalization24 Jul 2026
- 2026Online Variance Reduction for Domain Adaptation on Streaming Data22 Jul 2026
- 2026Variance-reduced Domain Adaptation using Paired Sampling22 Jul 2026
- 2026DA-MergeLoRA: Hypernetwork-Based LoRA Merging for Few-Shot Test-Time Domain Adaptation20 Jul 2026
- 2026Assessing Distribution Shift in Human Activity Recognition for Domain Generalization23 Jun 2026
- 2026Adversarial Domain Prompt Tuning and Generation for Single Domain Generalization19 Jun 2026
- 2026One-Step Generalization Ratio Guided Optimization for Domain Generalization15 Jun 2026
- 2026UME: A Unified Meta-Generalization Framework for Cross-Domain ETA31 May 2026
38 more
- 2026Beyond Fixed Budgets: Characterizing the Inelasticity and Limitations of Tree-of-Thought Reasoning Strategies19 May 2026
- 2026CPCANet: Deep Unfolding Common Principal Component Analysis for Domain Generalization6 May 2026
- 2026Stabilizing In-Context Multi-Source Domain Adaptation for Biomedical Images Through Controls22 Apr 2026
- 2026CrossFlowDG: Bridging the Modality Gap with Cross-modal Flow Matching for Domain Generalization18 Apr 2026
- 2026Rethinking Representations for Cross-Domain Infrared Small Target Detection: A Generalizable Perspective from the Frequency Domain2 Apr 2026
- 2026A Causal Framework for Mitigating Data Shifts in Healthcare13 Mar 2026
- 2026Adversarial Batch Representation Augmentation for Batch Correction in High-Content Cellular Screening5 Mar 2026
- 2026Reasoning-Driven Multimodal LLM for Domain Generalization27 Feb 2026
- 2026The Mean is the Mirage: Entropy-Adaptive Model Merging under Heterogeneous Domain Shifts in Medical Imaging24 Feb 2026
- 2026Distributional Deep Learning for Super-Resolution of 4D Flow MRI under Domain Shift16 Feb 2026
- 2026EdgeMask-DG*: Learning Domain-Invariant Graph Structures via Adversarial Edge Masking5 Feb 2026
- 2026Meta-learning to Address Data Shift in Time Series Classification13 Jan 2026
- 2026Fine-Grained Generalization via Structuralizing Concept and Feature Space into Commonality, Specificity and Confounding6 Jan 2026
- 2025Progressive Conditioned Scale-Shift Recalibration of Self-Attention for Online Test-time Adaptation14 Dec 2025
- 2025Variance Matters: Improving Domain Adaptation via Stratified Sampling4 Dec 2025
- 2025FedSDA: Federated Stain Distribution Alignment for Non-IID Histopathological Image Classification15 Nov 2025
- 2025Exploring Scale Shift in Crowd Localization under the Context of Domain Generalization22 Oct 2025
- 2025Domain Generalization Under Posterior Drift6 Oct 2025
- 2025Single Domain Generalization in Diabetic Retinopathy: A Neuro-Symbolic Learning Approach3 Sep 2025
- 2025Text-conditioned State Space Model For Domain-generalized Change Detection Visual Question Answering12 Aug 2025
- 2025Hierarchical Adaptive networks with Task vectors for Test-Time Adaptation11 Aug 2025
- 2025ETTA: Efficient Test-Time Adaptation for Vision-Language Models through Dynamic Embedding Updates7 Aug 2025
- 2025Domain Generalization via Pareto Optimal Gradient Matching16 Jul 2025
- 2025Prompt Disentanglement via Language Guidance and Representation Alignment for Domain Generalization3 Jul 2025
- 2025Multi-Source COVID-19 Detection via Variance Risk Extrapolation29 Jun 2025
- 2025Reproducible Evaluation of Camera Auto-Exposure Methods in the Field: Platform, Benchmark and Lessons Learned19 Jun 2025
- 2025Learning to Adapt Frozen CLIP for Few-Shot Test-Time Domain Adaptation18 Jun 2025
- 2025FEDTAIL: Federated Long-Tailed Domain Generalization with Sharpness-Guided Gradient Matching10 Jun 2025
- 2025Discovering Hierarchy-Grounded Domains with Adaptive Granularity for Clinical Domain Generalization8 Jun 2025
- 2025TeDA: Boosting Vision-Lanuage Models for Zero-Shot 3D Object Retrieval via Testing-time Distribution Alignment5 May 2025
- 2025Domain Generalization via Discrete Codebook Learning9 Apr 2025
- 2025Generative Classifier for Domain Generalization3 Apr 2025
- 2025DGSAM: Domain Generalization via Individual Sharpness-Aware Minimization30 Mar 2025
- 2025FIESTA: Fisher Information-based Efficient Selective Test-time Adaptation29 Mar 2025
- 2025OCRT: Boosting Foundation Models in the Open World with Object-Concept-Relation Triad24 Mar 2025
- 2025Domain Generalization for Improved Human Activity Recognition in Office Space Videos Using Adaptive Pre-processing16 Mar 2025
- 2025Robust Predictive Modeling Under Unseen Data Distribution Shifts: A Methodological Commentary5 Mar 2025
- 2025Generative Risk Minimization for Out-of-Distribution Generalization on Graphs11 Feb 2025
50 newest of 183 All 183 (JSON)
- 2024Efficient Test-Time Adaptation of Vision-Language Modelscited by 141
- 2023Leveraging Vision-Language Models for Improving Domain Generalization in Image Classificationcited by 41
- 2024FuRL: Visual-Language Models as Fuzzy Rewards for Reinforcement Learningcited by 26
- 2024Learning Invariant Causal Mechanism from Vision-Language Modelscited by 3
- 2025ETTA: Efficient Test-Time Adaptation for Vision-Language Models through Dynamic Embedding Updatescited by 3
- 2025TeDA: Boosting Vision-Lanuage Models for Zero-Shot 3D Object Retrieval via Testing-time Distribution Alignmentcited by 2
- 2024Language Plays a Pivotal Role in the Object-Attribute Compositional Generalization of CLIPcited by 1
- 2025ETTA: Efficient Test-Time Adaptation for Vision-Language Models through Dynamic Embedding Updates7 Aug 2025
- 2025TeDA: Boosting Vision-Lanuage Models for Zero-Shot 3D Object Retrieval via Testing-time Distribution Alignment5 May 2025
- 2024FuRL: Visual-Language Models as Fuzzy Rewards for Reinforcement Learning2 Jun 2024
- 2024Learning Invariant Causal Mechanism from Vision-Language Models24 May 2024
- 2024Language Plays a Pivotal Role in the Object-Attribute Compositional Generalization of CLIP27 Mar 2024
- 2024Efficient Test-Time Adaptation of Vision-Language Models27 Mar 2024
- 2023Leveraging Vision-Language Models for Improving Domain Generalization in Image Classification12 Oct 2023
- 2024FDS: Feedback-guided Domain Synthesis with Multi-Source Conditional Diffusion Models for Domain Generalizationcited by 5
- 2026Adversarial Domain Prompt Tuning and Generation for Single Domain Generalizationcited by 4
- 2026CrossFlowDG: Bridging the Modality Gap with Cross-modal Flow Matching for Domain Generalizationcited by 0
- 2026PAPT++: Risk-Aware Adversarial Tuning and Generation for Single Domain Generalizationcited by 0
- 2026PAPT++: Risk-Aware Adversarial Tuning and Generation for Single Domain Generalization4 Sep 2026
- 2026Adversarial Domain Prompt Tuning and Generation for Single Domain Generalization19 Jun 2026
- 2026CrossFlowDG: Bridging the Modality Gap with Cross-modal Flow Matching for Domain Generalization18 Apr 2026
- 2024FDS: Feedback-guided Domain Synthesis with Multi-Source Conditional Diffusion Models for Domain Generalization4 Jul 2024
- 2024Out-of-Distribution Learning with Human Feedbackcited by 6
- 2024Out-of-Distribution Learning with Human Feedback14 Aug 2024
- 2021MEMO: Test Time Robustness via Adaptation and Augmentationcited by 377
- 2023A Comprehensive Survey on Test-Time Adaptation under Distribution Shiftscited by 359
- 2022Surgical Fine-Tuning Improves Adaptation to Distribution Shiftscited by 205
- 2022Domain Generalization by Mutual-Information Regularization with Pre-trained Modelscited by 133
- 2021The Norm Must Go On: Dynamic Unsupervised Domain Adaptation by Normalizationcited by 131
- 2022Back to the Source: Diffusion-Driven Test-Time Adaptationcited by 71
- 2022Probable Domain Generalization via Quantile Risk Minimizationcited by 66
- 2023Advances and Challenges in Meta-Learning: A Technical Reviewcited by 65
- 2023On Pitfalls of Test-Time Adaptationcited by 64
- 2023Feed Two Birds with One Scone: Exploiting Wild Data for Both Out-of-Distribution Generalization and Detectioncited by 57
- 2022An Efficient Domain-Incremental Learning Approach to Drive in All Weather Conditionscited by 54
- 2023AdaNPC: Exploring Non-Parametric Classifier for Test-Time Adaptationcited by 48
12 most cited of 165 All 165 (JSON)
- 2026Domain Generalization under Sampling Pattern Shifts in Irregular Time Series27 Sep 2026
- 2026PatchGen: Learning Soft Intra-Image Predictive Subsets for Visual Generalization13 Aug 2026
- 2026PIcsC: Partitioning-Induced Covariate Shift Correction28 Jul 2026
- 2026Projection Pursuit CPCANet for Domain Generalization24 Jul 2026
- 2026Online Variance Reduction for Domain Adaptation on Streaming Data22 Jul 2026
- 2026Variance-reduced Domain Adaptation using Paired Sampling22 Jul 2026
- 2026DA-MergeLoRA: Hypernetwork-Based LoRA Merging for Few-Shot Test-Time Domain Adaptation20 Jul 2026
- 2026Assessing Distribution Shift in Human Activity Recognition for Domain Generalization23 Jun 2026
- 2026One-Step Generalization Ratio Guided Optimization for Domain Generalization15 Jun 2026
- 2026UME: A Unified Meta-Generalization Framework for Cross-Domain ETA31 May 2026
- 2026CPCANet: Deep Unfolding Common Principal Component Analysis for Domain Generalization6 May 2026
- 2026Stabilizing In-Context Multi-Source Domain Adaptation for Biomedical Images Through Controls22 Apr 2026
12 newest of 165 All 165 (JSON)