用物理先验指导扩散模型,提升弥散MRI的高角分辨率重建质量
Physics-Guided Diffusion Transformer with Spherical Harmonic Posterior Sampling for High-Fidelity Angular Super-Resolution in Diffusion MRI
- 融合量子空间几何与球谐函数的生成式建模方法
- 在真实数据上实现更精细的角细节恢复与更高保真度
- 适合需要高精度弥散MRI的神经科学研究者
弥散MRI角超分辨率(ASR)旨在不延长扫描时间的前提下,从有限的低角分辨率(LAR)数据中重建高角分辨率(HAR)信号。现有方法因对q空间几何建模不足及物理约束利用不够,难以恢复细微角细节或保持高保真度。本文提出物理引导的扩散变换器(PGDiT),在训练阶段通过基于b向量调制与随机角遮蔽的q空间几何感知模块(QGAM),实现方向感知表征学习;在推理阶段采用两阶段球谐函数引导后验采样(SHPS),确保与实际采集数据对齐,并结合热扩散基球谐正则化,保证物理合理性。该粗到精优化策略有效缓解纯数据驱动生成模型常见的过度平滑与伪影问题。在通用ASR任务及两种下游应用(扩散张量成像DTI与神经元取向分散与密度成像NODDI)上,实验表明PGDiT在细节恢复与数据保真度方面均优于现有深度学习模型,为神经科学与临床研究提供高保真HAR dMRI重建新范式。
原文摘要 · Abstract (English)
Diffusion MRI (dMRI) angular super-resolution (ASR) aims to reconstruct high-angular-resolution (HAR) signals from limited low-angular-resolution (LAR) data without prolonging scan time. However, existing methods are limited in recovering fine-grained angular details or preserving high fidelity due to inadequate modeling of q-space geometry and insufficient incorporation of physical constraints. In this paper, we introduce a Physics-Guided Diffusion Transformer (PGDiT) designed to explore physical priors throughout both training and inference stages. During training, a Q-space Geometry-Aware Module (QGAM) with b-vector modulation and random angular masking facilitates direction-aware representation learning, enabling the network to generate directionally consistent reconstructions with fine angular details from sparse and noisy data. In inference, a two-stage Spherical Harmonics-Guided Posterior Sampling (SHPS) enforces alignment with the acquired data, followed by heat-diffusion-based SH regularization to ensure physically plausible reconstructions. This coarse-to-fine refinement strategy mitigates oversmoothing and artifacts commonly observed in purely data-driven or generative models. Extensive experiments on general ASR tasks and two downstream applications, Diffusion Tensor Imaging (DTI) and Neurite Orientation Dispersion and Density Imaging (NODDI), demonstrate that PGDiT outperforms existing deep learning models in detail recovery and data fidelity. Our approach presents a novel generative ASR framework that offers high-fidelity HAR dMRI reconstructions, with potential applications in neuroscience and clinical research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。