用流模型加速扩散模型,实现快速高分辨率脑部代谢成像。
A Flow-based Truncated Denoising Diffusion Model for Super-resolution Magnetic Resonance Spectroscopic Imaging
- 用流模型估计截断扩散步骤,大幅减少采样迭代次数。
- 相比基线扩散模型提速超9倍,生成图像质量更优。
- 支持不确定性评估和清晰度调节,适合临床医生使用。
磁共振波谱成像(MRSI)是一种无创代谢研究技术,对神经疾病、癌症和糖尿病具有重要意义。高空间分辨率有助于病变表征,但受限于低代谢物浓度导致的时间与灵敏度问题,实际采集多为低分辨率。因此亟需一种后处理方法,从快速、高灵敏度的低分辨率数据重建高分辨率图像。现有深度学习超分辨率方法虽有进展,但生成精度与质量仍有限。近期扩散模型在多种任务中表现优异,但采样需大量迭代,耗时长。本文提出基于流模型的截断去噪扩散模型(FTDDM),通过截断扩散链并利用归一化流网络估计截断步数,显著缩短生成时间。模型条件于缩放因子,支持多尺度超分辨率。我们构建了25名高级别胶质瘤患者的1H-MRSI数据集用于训练与评估。结果表明,FTDDM优于现有生成模型,采样速度比基线扩散模型提升超过9倍。放射科医生评价确认其临床优势,且支持不确定性估计与锐度调节,拓展了临床应用潜力。
原文摘要 · Abstract (English)
Magnetic Resonance Spectroscopic Imaging (MRSI) is a non-invasive imaging technique for studying metabolism and has become a crucial tool for understanding neurological diseases, cancers and diabetes. High spatial resolution MRSI is needed to characterize lesions, but in practice MRSI is acquired at low resolution due to time and sensitivity restrictions caused by the low metabolite concentrations. Therefore, there is an imperative need for a post-processing approach to generate high-resolution MRSI from low-resolution data that can be acquired fast and with high sensitivity. Deep learning-based super-resolution methods provided promising results for improving the spatial resolution of MRSI, but they still have limited capability to generate accurate and high-quality images. Recently, diffusion models have demonstrated superior learning capability than other generative models in various tasks, but sampling from diffusion models requires iterating through a large number of diffusion steps, which is time-consuming. This work introduces a Flow-based Truncated Denoising Diffusion Model (FTDDM) for super-resolution MRSI, which shortens the diffusion process by truncating the diffusion chain, and the truncated steps are estimated using a normalizing flow-based network. The network is conditioned on upscaling factors to enable multi-scale super-resolution. To train and evaluate the deep learning models, we developed a 1H-MRSI dataset acquired from 25 high-grade glioma patients. We demonstrate that FTDDM outperforms existing generative models while speeding up the sampling process by over 9-fold compared to the baseline diffusion model. Neuroradiologists' evaluations confirmed the clinical advantages of our method, which also supports uncertainty estimation and sharpness adjustment, extending its potential clinical applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。