The Mean is the Mirage: Entropy-Adaptive Model Merging under Heterogeneous Domain Shifts in Medical Imaging
Not on the map. Why?
Abstract (arXiv)
Model merging under unseen test-time distribution shifts often renders naive strategies, such as mean averaging unreliable. This challenge is especially acute in medical imaging, where models are fine-tuned locally at clinics on private data, producing domain-specific models that differ by scanner, protocol, and population. When deployed at an unseen clinical site, test cases arrive in unlabeled, non-i.i.d. batches, and the model must adapt immediately without labels. In this work, we introduce an entropy-adaptive, fully online model-merging method that yields a batch-specific merged model via only forward passes, effectively leveraging target information. We further demonstrate why mean merging is prone to failure and misaligned under heterogeneous domain shifts. Next, we mitigate encoder classifier mismatch by decoupling the encoder and classification head, merging with separate merging coefficients. We extensively evaluate our method with state-of-the-art baselines using two backbones across nine medical and natural-domain generalization image classification datasets, showing consistent gains across standard evaluation and challenging scenarios. These performance gains are achieved while retaining single-model inference at test-time, thereby demonstrating the effectiveness of our method.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 78 of its 99 references that are in the map. First posted on arXiv 24 Feb 2026.
Builds on
78 of 99 references.
References · 78
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2019Advances and Open Problems in Federated Learningcited by 3,021
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
38 more
- 2020Do We Really Need to Access the Source Data? Source Hypothesis Transfer for Unsupervised Domain Adaptationcited by 844
- 2018Recognition in Terra Incognitacited by 655
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2022Continual Test-Time Domain Adaptationcited by 520
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2023Towards Stable Test-Time Adaptation in Dynamic Wild Worldcited by 409
- 2023A Comprehensive Survey on Test-Time Adaptation under Distribution Shiftscited by 353
- 2017Snapshot Ensembles: Train 1, get M for freecited by 349
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2020Evaluating Prediction-Time Batch Normalization for Robustness under Covariate Shiftcited by 235
- 2022Surgical Fine-Tuning Improves Adaptation to Distribution Shiftscited by 205
- 2022Parameter-free Online Test-time Adaptationcited by 184
- 2020Adaptive Risk Minimization: Learning to Adapt to Domain Shiftcited by 183
- 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
- 2022NOTE: Robust Continual Test-time Adaptation Against Temporal Correlationcited by 150
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2021Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariancescited by 125
- 2020Domain Generalization for Medical Imaging Classification with Linear-Dependency Regularizationcited by 122
- 2024EMR-Merging: Tuning-Free High-Performance Model Mergingcited by 108
- 2022Test-Time Adaptation via Conjugate Pseudo-labelscited by 98
- 2022Model Ratatouille: Recycling Diverse Models for Out-of-Distribution Generalizationcited by 97
- 2024Representation Surgery for Multi-Task Model Mergingcited by 95
- 2023Deep Model Fusion: A Surveycited by 89
- 2024Model Stock: All we need is just a few fine-tuned modelscited by 87
- 2024Merging Multi-Task Models via Weight-Ensembling Mixture of Expertscited by 85
- 2023AdapterSoup: Weight Averaging to Improve Generalization of Pretrained Language Modelscited by 84
- 2022When Do Flat Minima Optimizers Work?cited by 78
- 2023Model Merging by Uncertainty-Based Gradient Matchingcited by 77
- 2024DELLA-Merging: Reducing Interference in Model Merging through Magnitude-Based Samplingcited by 77
- 2021Loss Surface Simplexes for Mode Connecting Volumes and Fast Ensemblingcited by 75
- 2024Test-Time Model Adaptation with Only Forward Passescited by 74
- 2022Test-Time Adaptation via Self-Training with Nearest Neighbor Informationcited by 64
- 2023On Pitfalls of Test-Time Adaptationcited by 63
- 2024Beyond Model Adaptation at Test Time: A Surveycited by 55
50 most cited of 78 All 78 (JSON)
Built on by
Not cited yet.