MARR: Decoupling Policy, Execution, and Calibration for All-in-One Medical Image Restoration
Not on the map. Why?
Abstract (arXiv)
All-in-one medical image restoration seeks to recover heterogeneous clinical images with a single model, but PET, CT, and MRI differ substantially in degradation statistics, anatomical contrast, and output-space bias. A fully shared network can entangle modality-specific residual errors, whereas separate modality-specific networks sacrifice the practical advantages of unified deployment. We therefore recast all-in-one restoration as a question of where limited adaptation should be placed: policy selection, feature execution, or output calibration. We propose MARR, a compact restoration framework that constrains multi-modality adaptation into degradation-aware policy routing, modality-private residual execution, and image-domain residual correction without requiring degradation labels or separate modality-specific models. The policy branch forms a routing prompt from input statistics, latent content, and modality identity, and uses it only as a control signal. Prompt-gated modality-private adapters then perform lightweight residual refinement at intermediate decoder stages, while zero-initialized modality-specific output heads calibrate the final image-domain residual without perturbing the initial shared prediction. On an all-in-one PET, CT, and MRI restoration benchmark, MARR outperforms thirteen methods re-trained under the same protocol, achieving PSNR values of 37.34 dB, 33.85 dB, and 32.09 dB on PET, CT, and MRI, respectively, and the best modality-average PSNR of 34.43 dB. The code is publicly available at https://github.com/CHB-learner/MARR.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 34 references that are in the map. First posted on arXiv 6 Sep 2026.
Builds on
22 of 34 references.
References · 22
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2021SwinIR: Image Restoration Using Swin Transformercited by 1,887
- 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
- 2024VMamba: Visual State Space Modelcited by 969
- 2022Simple Baselines for Image Restorationcited by 708
- 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 352
- 2024MambaIRv2: Attentive State Space Restorationcited by 94
- 2024Restore-RWKV: Efficient and Effective Medical Image Restoration with RWKVcited by 36
10 more
- 2024All-In-One Medical Image Restoration via Task-Adaptive Routingcited by 30
- 20223D Segmentation Guided Style-based Generative Adversarial Networks for PET Synthesiscited by 13
- 2024DenoMamba: A fused state-space model for low-dose CT denoisingcited by 11
- 2023DRMC: A Generalist Model with Dynamic Routing for Multi-Center PET Image Synthesiscited by 8
- 2026Learning Domain-Aware Task Prompt Representations for Multi-Domain All-in-One Image Restorationcited by 7
- 2024Region Attention Transformer for Medical Image Restorationcited by 5
- 2025All-in-One Medical Image Restoration with Latent Diffusion-Enhanced Vector-Quantized Codebook Priorcited by 5
- 2026QualiTeacher: Quality-Conditioned Pseudo-Labeling for Real-World Image Restorationcited by 5
- 2025TAT: Task-Adaptive Transformer for All-in-One Medical Image Restorationcited by 3
- 2026MAP-Diff: Multi-Anchor Guided Diffusion for Progressive 3D Whole-Body Low-Dose PET Denoisingcited by 1
Built on by
Not cited yet.