HiFi-MambaV2: Hierarchical Shared-Routed MoE for High-Fidelity MRI Reconstruction
On the map: Mixture-of-experts → Medical and biomedical MoE
Abstract (arXiv)
Reconstructing high-fidelity MR images from undersampled k-space data requires recovering high-frequency details while maintaining anatomical coherence. We present HiFi-MambaV2, a hierarchical shared-routed Mixture-of-Experts (MoE) Mamba architecture that couples frequency decomposition with content-adaptive computation. The model comprises two core components: (i) a separable frequency-consistent Laplacian pyramid (SF-Lap) that delivers alias-resistant, stable low- and high-frequency streams; and (ii) a hierarchical shared-routed MoE that performs per-pixel top-1 sparse dispatch to shared experts and local routers, enabling effective specialization with stable cross-depth behavior. A lightweight global context path is fused into an unrolled, data-consistency-regularized backbone to reinforce long-range reasoning and preserve anatomical coherence. Evaluated on fastMRI, CC359, ACDC, M4Raw, and Prostate158, HiFi-MambaV2 consistently outperforms CNN-, Transformer-, and prior Mamba-based baselines in PSNR, SSIM, and NMSE across single- and multi-coil settings and multiple acceleration factors, consistently surpassing consistent improvements in high-frequency detail and overall structural fidelity. These results demonstrate that HiFi-MambaV2 enables reliable and robust MRI reconstruction.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Medical and biomedical MoE, one of 12 directions in Mixture-of-experts. Builds on 20 of its 35 references that are in the map. First posted on arXiv 23 Nov 2025.
Builds on
20 of 35 references.
References · 20
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2024VMamba: Visual State Space Modelcited by 955
- 2018fastMRI: An Open Dataset and Benchmarks for Accelerated MRIcited by 464
- 2016Deep Convolutional Neural Network for Inverse Problems in Imagingcited by 431
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 391
- 2017Learning a Variational Network for Reconstruction of Accelerated MRI Datacited by 340
- 2017Convolutional Recurrent Neural Networks for Dynamic MR Image Reconstructioncited by 105
- 2017Deep learning for undersampled MRI reconstructioncited by 91
8 more
- 2021Unsupervised MRI Reconstruction via Zero-Shot Learned Adversarial Transformerscited by 76
- 2017A Deep Cascade of Convolutional Neural Networks for MR Image Reconstructioncited by 74
- 2020DuDoRNet: Learning a Dual-Domain Recurrent Network for Fast MRI Reconstruction with Deep T1 Priorcited by 72
- 2022Swin Transformer for Fast MRIcited by 56
- 2022ReconFormer: Accelerated MRI Reconstruction Using Recurrent Transformercited by 52
- 2015Loss Functions for Neural Networks for Image Processingcited by 41
- 2024Boosting ViT-based MRI Reconstruction from the Perspectives of Frequency Modulation, Spatial Purification, and Scale Diversificationcited by 6
- 2025HiFi-Mamba: Dual-Stream W-Laplacian Enhanced Mamba for High-Fidelity MRI Reconstructioncited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026MoRE: A Mixture-of-Experts-Based Task-Adaptive End-to-End Network for Multimodal MRI Reconstruction
- 2026SO-Mamba: State-Ownership Mamba for Unrolled MRI Reconstruction
- 2026MoE-dqINR: A Unified Mixture-of-Experts Implicit Neural Representation Framework for Scan-Specific Dynamic and Quantitative MRI Reconstruction
- 2025DH-Mamba: Exploring Dual-domain Hierarchical State Space Models for MRI Reconstruction
- 2025Mamba-Based Modality Disentanglement Network for Multi-Contrast MRI Reconstruction
- 2025Branch Learning in MRI: More Data, More Models, More Training
- 2024MMR-Mamba: Multi-Modal MRI Reconstruction with Mamba and Spatial-Frequency Information Fusion
- 2026Vision-Language Model-Guided Deep Unrolling Enables Personalized, Fast MRI