mlsanityData to 14 Oct 2026

MARR: Decoupling Policy, Execution, and Calibration for All-in-One Medical Image Restoration

arXiv 2609.06645 ↗First posted 6 Sep 2026cs.CVcited by 0

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
  4. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  5. 2021SwinIR: Image Restoration Using Swin Transformercited by 1,887
  6. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
  7. 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
  8. 2024VMamba: Visual State Space Modelcited by 969
  9. 2022Simple Baselines for Image Restorationcited by 708
  10. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 352
  11. 2024MambaIRv2: Attentive State Space Restorationcited by 94
  12. 2024Restore-RWKV: Efficient and Effective Medical Image Restoration with RWKVcited by 36
10 more
  1. 2024All-In-One Medical Image Restoration via Task-Adaptive Routingcited by 30
  2. 20223D Segmentation Guided Style-based Generative Adversarial Networks for PET Synthesiscited by 13
  3. 2024DenoMamba: A fused state-space model for low-dose CT denoisingcited by 11
  4. 2023DRMC: A Generalist Model with Dynamic Routing for Multi-Center PET Image Synthesiscited by 8
  5. 2026Learning Domain-Aware Task Prompt Representations for Multi-Domain All-in-One Image Restorationcited by 7
  6. 2024Region Attention Transformer for Medical Image Restorationcited by 5
  7. 2025All-in-One Medical Image Restoration with Latent Diffusion-Enhanced Vector-Quantized Codebook Priorcited by 5
  8. 2026QualiTeacher: Quality-Conditioned Pseudo-Labeling for Real-World Image Restorationcited by 5
  9. 2025TAT: Task-Adaptive Transformer for All-in-One Medical Image Restorationcited by 3
  10. 2026MAP-Diff: Multi-Anchor Guided Diffusion for Progressive 3D Whole-Body Low-Dose PET Denoisingcited by 1

Built on by

Not cited yet.