用扩散模型实现超声波速重建,精度高且能估测不确定性。
DiffSOS: Acoustic Conditional Diffusion Model for Speed-of-Sound Reconstruction in Ultrasound Computed Tomography
- 基于声学控制网络的条件扩散模型,物理约束强。
- 多尺度结构相似性达0.957,优于现有方法。
- 支持快速生成并输出像素级置信度,适合临床应用。
从声学波形准确重建声速(SoS)是超声计算机断层成像(USCT)的核心,可实现定量速度映射,揭示常规影像中难以察觉的解剖细节与病理变化。然而,现有算法受限:传统全波形反演(FWI)计算量大,当前深度学习方法常产生过度平滑结果,缺乏细节。本文提出DiffSOS,一种直接将声学波形映射为声速图的条件扩散模型。框架采用专用声学ControlNet,严格约束去噪过程符合物理波测量;通过融合噪声预测、空间重构与噪声频谱内容的混合损失函数,保证结构一致性;采用随机扩散隐式模型(DDIM)采样,仅10步即可近实时重建。关键优势在于利用生成模型的随机性,实现像素级不确定性估计,提供确定性方法缺失的可靠性度量。在OpenPros USCT基准测试中,DiffSOS显著超越现有先进网络,平均多尺度结构相似性达0.957。该方法生成高保真声速图,并提供可信度度量,有助于更安全高效的临床解读。
原文摘要 · Abstract (English)
Accurate Speed-of-Sound (SoS) reconstruction from acoustic waveforms is a cornerstone of ultrasound computed tomography (USCT), enabling quantitative velocity mapping that reveals subtle anatomical details and pathological variations often invisible in conventional imaging. However, practical utility is hindered by the limitations of existing algorithms; traditional Full Waveform Inversion (FWI) is computationally intensive, while current deep learning approaches tend to produce oversmoothed results lacking fine details. We propose DiffSOS, a conditional diffusion model that directly maps acoustic waveforms to SoS maps. Our framework employs a specialized acoustic ControlNet to strictly ground the denoising process in physical wave measurements. To ensure structural consistency, we optimize a hybrid loss function that integrates noise prediction, spatial reconstruction, and noise frequency content. To accelerate inference, we employ stochastic Denoising Diffusion Implicit Model (DDIM) sampling, achieving near real-time reconstruction with only 10 steps. Crucially, we exploit the stochastic generative nature of our framework to estimate pixel-wise uncertainty, providing a measure of reliability that is often absent in deterministic approaches. Evaluated on the OpenPros USCT benchmark, DiffSOS significantly outperforms state-of-the-art networks, achieving an average Multi-scale Structural Similarity of 0.957. Our approach provides high-fidelity SoS maps with a principled measure of confidence, facilitating safer and faster clinical interpretation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。