统一模型解决不同扫描协议下的低剂量CT图像重建难题
One CT Unified Model Training Framework to Rule All Scanning Protocols
- 通过不确定性引导的流形平滑,动态融合全局与子流形特征
- 在多个公开数据集上实现跨协议重建性能提升,有效缓解模型坍塌
- 适合低剂量CT临床应用,尤其适用于无配对数据场景
非理想测量计算机断层扫描(NICT)在降低辐射剂量的同时牺牲了图像质量,正逐步拓展其临床应用。尽管统一模型在NICT增强中展现出潜力,但多数方法依赖成对数据,而器官运动导致配对数据难以获取。无监督方法虽尝试突破此限制,却假设噪声均匀,忽视扫描协议差异,导致泛化能力差且易发生模型坍塌。我们观察到不同扫描协议对应不同的物理成像过程,在特征空间中形成离散子流形,违背了现有方法的假设,限制了其效果。为此,我们提出不确定性引导的流形平滑(UMS)框架,利用分类器识别子流形并预测不确定性得分,指导生成跨整个流形的多样化样本。该机制有效填补子流形间的空隙,促进特征空间连续且稠密。由于全局流形复杂,难以直接建模,我们设计一种由分类器引导的全局-子流形驱动架构,实现对子域变化的动态适应。该机制增强了网络捕捉共享与特定领域特征的能力,显著提升重建性能。我们在多个公开数据集上进行大量实验,验证了该方法在不同生成范式下的有效性。
原文摘要 · Abstract (English)
Non-ideal measurement computed tomography (NICT), which lowers radiation at the cost of image quality, is expanding the clinical use of CT. Although unified models have shown promise in NICT enhancement, most methods require paired data, which is an impractical demand due to inevitable organ motion. Unsupervised approaches attempt to overcome this limitation, but their assumption of homogeneous noise neglects the variability of scanning protocols, leading to poor generalization and potential model collapse. We further observe that distinct scanning protocols, which correspond to different physical imaging processes, produce discrete sub-manifolds in the feature space, contradicting these assumptions and limiting their effectiveness. To address this, we propose an Uncertainty-Guided Manifold Smoothing (UMS) framework to bridge the gaps between sub-manifolds. A classifier in UMS identifies sub-manifolds and predicts uncertainty scores, which guide the generation of diverse samples across the entire manifold. By leveraging the classifier's capability, UMS effectively fills the gaps between discrete sub-manifolds, and promotes a continuous and dense feature space. Due to the complexity of the global manifold, it's hard to directly model it. Therefore, we propose to dynamically incorporate the global- and sub-manifold-specific features. Specifically, we design a global- and sub-manifold-driven architecture guided by the classifier, which enables dynamic adaptation to subdomain variations. This dynamic mechanism improves the network's capacity to capture both shared and domain-specific features, thereby improving reconstruction performance. Extensive experiments on public datasets are conducted to validate the effectiveness of our method across different generation paradigms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。