Beyond the Static-World: Lifelong Learning for All-in-One Medical Image Restoration
Shihao Shan, Hongying Liu, Fanhua Shang, Liang Wan, Jingjing Deng
Abstract
All-in-one Medical Image Restoration (MedIR) models offer a promising path towards generalized medical imaging intelligence but face two critical spatiotemporal challenges: 1) Spatial modality interference, where conflicting gradients from diverse modalities (e.g., MRI, CT, PET) degrade performance; and 2) a temporal static-world assumption that ignores the continual data streams in realworld clinical settings, leading to catastrophic forgetting. To address this dual challenge, we propose a novel lifelong learning framework (called Resilient On-the-fly Medical Enhancement, ROME) governed by a "disentangleoptimize-consolidate" paradigm. ROME first resolves the foundational modality conflict via the Modality-Invariant Disentanglement via Adversarial Balancing (MIDAB) module. It establishes a strategic "adversarial balance" between a "content preservation force" and a "modality erasure force" to optimize a disentangled, unified feature manifold. Building on this stable foundation, the Adaptive Feature Consolidation (AFC) module combats forgetting. AFC dynamically locates an optimal feature consolidation point via a prediction network, enforced by a novel diversity loss to ensure robust continuous learning. Experiments demonstrate that ROME not only achieves SOTA performance in static settings but also exhibits superior resilience in rigorous domain-incremental benchmarks, reducing the average catastrophic performance degradation by over 10%.
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 7671a5c1-1ece-48a8-829c-802bc3e3da47Builds on13
- Restormer: Efficient Transformer for High-Resolution Image RestorationSyed Waqas Zamir, Aditya Arora, Salman Khan, Munawar Hayat et al.CVPR 2022 · 3,348 citations
- Conflict-Averse Gradient Descent for Multi-task learningBo Liu, Xingchao Liu, Xiaojie Jin, Peter Stone et al.NeurIPS 2021 · 686 citations
- All-In-One Image Restoration for Unknown CorruptionBoyun Li, Xiao Liu, Peng Hu, Zhongqin Wu et al.CVPR 2022 · 338 citations
- IDM: An Intermediate Domain Module for Domain Adaptive Person Re-IDYongxing Dai, Jun Liu, Yifan Sun, Zekun Tong et al.ICCV 2021 · 145 citations
- CPPO: Continual Learning for Reinforcement Learning with Human FeedbackHan Zhang, Yu Lei, Lin Gui, Min Yang et al.ICLR 2024 · 43 citations
Related papers
- Multi-modal Medical Diagnosis via Large-small Model CollaborationWanyi Chen, Zihua Zhao, Jiangchao Yao, Ya Zhang et al.CVPR 2025
- CoSMIC: Continual Self-Supervised Learning for Multi-Domain Medical Imaging Via Conditional Mutual Information MaximizationYihang Liu, Ying Wen, Longzhen Yang, Lianghua He et al.ICCV 2025 · 2 citations
- MedCRP-CL: Continual Medical Image Segmentation via Bayesian Nonparametric Semantic Modality DiscoveryZiyuan GaoICML 2026
- Continual Self-Supervised Learning: Towards Universal Multi-Modal Medical Data Representation LearningYiwen Ye, Yutong Xie, Jianpeng Zhang, Ziyang Chen et al.CVPR 2024 · 30 citations
- Effective and Robust Multimodal Medical Image AnalysisJoy Dhar, Nayyar Zaidi, Maryam HaghighatKDD 2026 · 1 citation
