mlsanityData to 7 Oct 2026

The Mean is the Mirage: Entropy-Adaptive Model Merging under Heterogeneous Domain Shifts in Medical Imaging

arXiv 2602.21372 ↗First posted 24 Feb 2026cs.LG cs.AIcited by 0

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  4. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
  5. 2019Advances and Open Problems in Federated Learningcited by 3,021
  6. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  7. 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
  8. 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
  9. 2022Editing Models with Task Arithmeticcited by 1,076
  10. 2020In Search of Lost Domain Generalizationcited by 1,042
  11. 2017Deeper, Broader and Artier Domain Generalizationcited by 983
  12. 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
38 more
  1. 2020Do We Really Need to Access the Source Data? Source Hypothesis Transfer for Unsupervised Domain Adaptationcited by 844
  2. 2018Recognition in Terra Incognitacited by 655
  3. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
  4. 2021Merging Models with Fisher-Weighted Averagingcited by 613
  5. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
  6. 2022Continual Test-Time Domain Adaptationcited by 520
  7. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
  8. 2023Towards Stable Test-Time Adaptation in Dynamic Wild Worldcited by 409
  9. 2023A Comprehensive Survey on Test-Time Adaptation under Distribution Shiftscited by 353
  10. 2017Snapshot Ensembles: Train 1, get M for freecited by 349
  11. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
  12. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
  13. 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
  14. 2020Evaluating Prediction-Time Batch Normalization for Robustness under Covariate Shiftcited by 235
  15. 2022Surgical Fine-Tuning Improves Adaptation to Distribution Shiftscited by 205
  16. 2022Parameter-free Online Test-time Adaptationcited by 184
  17. 2020Adaptive Risk Minimization: Learning to Adapt to Domain Shiftcited by 183
  18. 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
  19. 2022NOTE: Robust Continual Test-time Adaptation Against Temporal Correlationcited by 150
  20. 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
  21. 2021Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariancescited by 125
  22. 2020Domain Generalization for Medical Imaging Classification with Linear-Dependency Regularizationcited by 122
  23. 2024EMR-Merging: Tuning-Free High-Performance Model Mergingcited by 108
  24. 2022Test-Time Adaptation via Conjugate Pseudo-labelscited by 98
  25. 2022Model Ratatouille: Recycling Diverse Models for Out-of-Distribution Generalizationcited by 97
  26. 2024Representation Surgery for Multi-Task Model Mergingcited by 95
  27. 2023Deep Model Fusion: A Surveycited by 89
  28. 2024Model Stock: All we need is just a few fine-tuned modelscited by 87
  29. 2024Merging Multi-Task Models via Weight-Ensembling Mixture of Expertscited by 85
  30. 2023AdapterSoup: Weight Averaging to Improve Generalization of Pretrained Language Modelscited by 84
  31. 2022When Do Flat Minima Optimizers Work?cited by 78
  32. 2023Model Merging by Uncertainty-Based Gradient Matchingcited by 77
  33. 2024DELLA-Merging: Reducing Interference in Model Merging through Magnitude-Based Samplingcited by 77
  34. 2021Loss Surface Simplexes for Mode Connecting Volumes and Fast Ensemblingcited by 75
  35. 2024Test-Time Model Adaptation with Only Forward Passescited by 74
  36. 2022Test-Time Adaptation via Self-Training with Nearest Neighbor Informationcited by 64
  37. 2023On Pitfalls of Test-Time Adaptationcited by 63
  38. 2024Beyond Model Adaptation at Test Time: A Surveycited by 55

50 most cited of 78 All 78 (JSON)

Built on by

Not cited yet.