提出双阶段自适应方法,提升MRI重建在不同设备下的效率与精度。
D2SA: Dual-Stage Distribution and Slice Adaptation for Efficient Test-Time Adaptation in MRI Reconstruction
- 分两阶段:先患者级分布适配,再单切片精细化调整
- 在五种分布偏移下性能提升,收敛速度加快30%以上
- 适合临床多设备场景的快速重建,尤其适用于数据少的情况
磁共振成像(MRI)设备与采集协议的差异导致分布偏移,影响重建模型在未见数据上的表现。测试时自适应(TTA)是解决此问题的有前景方案。然而,现有单次训练的TTA方法效率低,且分布建模不优;自监督学习在数据稀缺时易出现过平滑。为此,本文提出基于MRI隐式神经表示(MR-INR)的双阶段分布与切片自适应(D2SA)方法,包含两个阶段:第一阶段通过MR-INR分支学习跨切片共享表征,并利用均值与方差调整建模患者特异性偏移;第二阶段通过可学习的各向异性扩散模块,在冻结卷积层基础上对单切片输出进行精调,避免过平滑并降低计算开销。在五个不同MRI分布偏移数据集上的实验表明,该方法可有效融合多种自监督学习框架,在多样条件下显著提升性能并加速收敛。
原文摘要 · Abstract (English)
Variations in Magnetic resonance imaging (MRI) scanners and acquisition protocols cause distribution shifts that degrade reconstruction performance on unseen data. Test-time adaptation (TTA) offers a promising solution to address this discrepancies. However, previous single-shot TTA approaches are inefficient due to repeated training and suboptimal distributional models. Self-supervised learning methods may risk over-smoothing in scarce data scenarios. To address these challenges, we propose a novel Dual-Stage Distribution and Slice Adaptation (D2SA) via MRI implicit neural representation (MR-INR) to improve MRI reconstruction performance and efficiency, which features two stages. In the first stage, an MR-INR branch performs patient-wise distribution adaptation by learning shared representations across slices and modelling patient-specific shifts with mean and variance adjustments. In the second stage, single-slice adaptation refines the output from frozen convolutional layers with a learnable anisotropic diffusion module, preventing over-smoothing and reducing computation. Experiments across five MRI distribution shifts demonstrate that our method can integrate well with various self-supervised learning (SSL) framework, improving performance and accelerating convergence under diverse conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。