用分层专家路由Mamba模型,提升磁共振成像的高保真重建质量。
HiFi-MambaV2: Hierarchical Shared-Routed MoE for High-Fidelity MRI Reconstruction
- 分频一致拉普拉斯金字塔+分层共享路由专家,自适应处理不同频率内容。
- 在多个数据集上优于CNN、Transformer和先前Mamba模型,高频率细节更清晰。
- 适合医学影像重建研究者,尤其关注超分辨率与结构保真度的场景。
从欠采样k空间数据中重建高保真磁共振图像,需同时恢复高频细节并保持解剖一致性。我们提出HiFi-MambaV2,一种分层共享路由的混合专家(MoE)Mamba架构,将频率分解与内容自适应计算结合。模型包含两个核心组件:(i) 可分离的频率一致拉普拉斯金字塔(SF-Lap),提供抗伪影、稳定的低频与高频流;(ii) 分层共享路由MoE,采用逐像素top-1稀疏调度至共享专家与局部路由器,实现有效专业化且跨深度行为稳定。轻量级全局上下文路径融合于无展开、数据一致性正则化的主干网络中,增强长程推理能力并保持解剖一致性。在fastMRI、CC359、ACDC、M4Raw和Prostate158数据集上评估显示,无论单线圈或多线圈设置及多种加速因子下,其在PSNR、SSIM和NMSE指标上均持续优于基于CNN、Transformer及先前Mamba的基线模型,在高频细节与整体结构保真度上均有显著提升。结果表明,HiFi-MambaV2实现了可靠且稳健的MRI重建。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。