arXiv:2605.22031cs.CV2026-05被引 1

提出SO-Mamba,让MRI重建中不同阶段证据各归其位,提升图像质量。

SO-Mamba: State-Ownership Mamba for Unrolled MRI Reconstruction

论文配图:SO-Mamba: State-Ownership Mamba for Unrolled MRI Reconstruction
图 1 · 摘自论文原文
  • 设计状态归属路由机制,区分持久内容与更新证据
  • 在5个公开数据集上优于CNN、Transformer和Mamba基线
  • 适合需要高保真重建的医学影像研究者

加速MRI重建需在大空间范围内恢复缺失细节并保持解剖一致性。状态空间模型如Mamba具备高效长程建模能力,适合作为未展开重建中的学习正则项。但在数据一致性耦合的未展开求解器中,不同阶段处理不同重构迭代结果,其中驻留载体应保持跨阶段内容连贯性,而阶段依赖的非驻留证据应关联当前更新。若统一处理这些角色,会将持久驻留证据与更新相关非驻留证据混入同一循环内容路径。为此,我们提出SO-Mamba,一种状态归属的Mamba正则器,将每阶段重构证据分配至循环驻留、状态接口访问及非状态输出修正三类。SO-Mamba通过状态归属路由器(SOR)构建循环内容的驻留载体,并将非驻留证据路由至仿射调制的B/C状态接口及输出修正出口。驻留载体提供Mamba内容路径,而非驻留证据流通过调整状态接口和输出出口实现作用,不进入循环内容路径。我们进一步引入双层外带泄漏诊断,通过测量选择性扫描状态轨迹的外带能量与扫描后读出的表达,分离隐藏状态存储与读出表达。在涵盖多种解剖结构、采样模式与线圈配置的五个公开MRI重建基准上,SO-Mamba持续优于基于CNN、Transformer和Mamba的基线,且计算效率具有竞争力。

原文摘要 · Abstract (English)

Accelerated MRI reconstruction requires recovering missing details while preserving anatomically coherent structures across large spatial regions. State-space models such as Mamba provide efficient long-range modeling, making them attractive learned regularizers for unrolled reconstruction. However, in a data-consistency-coupled unrolled solver, different stages operate on different reconstruction iterates, where the resident carrier should preserve coherent reconstruction content across stages while stage-dependent non-resident evidence is tied to the current update. Treating these roles uniformly can place persistent resident-carrier evidence and update-dependent non-resident evidence into the same recurrent content route. We therefore propose SO-Mamba, a state-ownership Mamba regularizer that assigns reconstruction evidence within each Mamba stage to recurrent residency, state-interface access, and non-state output correction. SO-Mamba implements this ownership rule with a State-Ownership Router (SOR), which constructs a resident carrier for recurrent content and routes non-resident evidence to affine modulation of the B/C state interfaces and an output correction outlet. The resident carrier supplies the Mamba content route, while the non-resident evidence stream adapts the state interfaces and contributes through the output outlet without entering the recurrent content route. We further introduce a two-level outer-band leakage diagnostic that separates hidden-state storage from readout expression by measuring outer-band energy in the selective-scan state trajectory and the post-scan Mamba readout. Experiments on five public MRI reconstruction benchmarks spanning diverse anatomies, sampling patterns, and coil configurations show that SO-Mamba consistently improves over CNN-, Transformer-, and Mamba-based baselines with competitive computational efficiency.

MRI重建状态空间医学影像Mamba

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。