arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2511.17068cs.CVcs.AI

ReBrain: 通过检索增强扩散模型从稀疏CT切片重建脑部MRI

ReBrain: Brain MRI Reconstruction from Sparse CT Slice via Retrieval-Augmented Diffusion

Tongji University(同济大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Junming Liu, Yifei Sun, Weihua Cheng, Yujin Kang, Yirong Chen, Ding Wang, Guosun Zeng

AI总结:

ReBrain通过检索增强扩散模型从稀疏CT切片重建脑部MRI,利用BBDM和ControlNet实现结构连续性,提升稀疏条件下的跨模态重建性能。

摘要:

磁共振成像(MRI)在脑部疾病诊断中起着至关重要的作用,但某些患者由于物理或临床限制,无法总是进行MRI检查。最近的研究尝试通过CT扫描合成MRI,但低剂量协议往往导致CT体积高度稀疏,通过平面分辨率差,使得准确重建完整的脑MRI体积特别具有挑战性。为此,我们提出了ReBrain,一种用于脑MRI重建的检索增强扩散框架。给定任何具有有限切片的3D CT扫描,我们首先使用布朗桥扩散模型(BBDM)在2D维度上合成MRI切片。同时,我们通过微调的检索模型从全面的先前数据库中检索出在结构和病理学上相似的CT切片。这些检索到的切片被用作参考,并通过ControlNet分支引导中间MRI切片的生成,以确保结构连续性。我们进一步考虑了数据库缺乏合适参考时的罕见检索失败情况,并应用球面线性插值来提供补充指导。在SynthRAD2023和BraTS上的大量实验表明,ReBrain在稀疏条件下实现了跨模态重建的最先进性能。

英文摘要:

Magnetic Resonance Imaging (MRI) plays a crucial role in brain disease diagnosis, but it is not always feasible for certain patients due to physical or clinical constraints. Recent studies attempt to synthesize MRI from Computed Tomography (CT) scans; however, low-dose protocols often result in highly sparse CT volumes with poor through-plane resolution, making accurate reconstruction of the full brain MRI volume particularly challenging. To address this, we propose ReBrain, a retrieval-augmented diffusion framework for brain MRI reconstruction. Given any 3D CT scan with limited slices, we first employ a Brownian Bridge Diffusion Model (BBDM) to synthesize MRI slices along the 2D dimension. Simultaneously, we retrieve structurally and pathologically similar CT slices from a comprehensive prior database via a fine-tuned retrieval model. These retrieved slices are used as references, incorporated through a ControlNet branch to guide the generation of intermediate MRI slices and ensure structural continuity. We further account for rare retrieval failures when the database lacks suitable references and apply spherical linear interpolation to provide supplementary guidance. Extensive experiments on SynthRAD2023 and BraTS demonstrate that ReBrain achieves state-of-the-art performance in cross-modal reconstruction under sparse conditions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑