IEEE Trans Med Imaging. 2023 Aug;42(8):2133-2145. doi: 10.1109/TMI.2023.3244252. Epub 2023 Aug 1.
CT metal artefact reduction (MAR) methods based on supervised deep learning are often troubled by domain gap between simulated training dataset and real-application dataset, i.e., methods trained on simulation cannot generalize well to practical data. Unsupervised MAR methods can be trained directly on practical data, but they learn MAR with indirect metrics and often perform unsatisfactorily. To tackle the domain gap problem, we propose a novel MAR method called UDAMAR based on unsupervised domain adaptation (UDA). Specifically, we introduce a UDA regularization loss into a typical image-domain supervised MAR method, which mitigates the domain discrepancy between simulated and practical artefacts by feature-space alignment. Our adversarial-based UDA focuses on a low-level feature space where the domain difference of metal artefacts mainly lies. UDAMAR can simultaneously learn MAR from simulated data with known labels and extract critical information from unlabeled practical data. Experiments on both clinical dental and torso datasets show the superiority of UDAMAR by outperforming its supervised backbone and two state-of-the-art unsupervised methods. We carefully analyze UDAMAR by both experiments on simulated metal artefacts and various ablation studies. On simulation, its close performance to the supervised methods and advantages over the unsupervised methods justify its efficacy. Ablation studies on the influence from the weight of UDA regularization loss, UDA feature layers, and the amount of practical data used for training further demonstrate the robustness of UDAMAR. UDAMAR provides a simple and clean design and is easy to implement. These advantages make it a very feasible solution for practical CT MAR.
基于监督式深度学习的 CT 金属伪影降低 (MAR) 方法常常受到模拟训练数据集与实际应用数据集之间的域差距的困扰,即,在模拟中训练的方法无法很好地推广到实际数据中。无监督 MAR 方法可以直接在实际数据上进行训练,但是它们使用间接指标来学习 MAR,并且常常表现不佳。为了解决域差距问题,我们提出了一种名为 UDAMAR 的新型 MAR 方法,该方法基于无监督域自适应 (UDA)。具体来说,我们在典型的图像域监督 MAR 方法中引入了 UDA 正则化损失,通过特征空间对齐来减轻模拟和实际伪影之间的域差异。我们基于对抗的 UDA 专注于金属伪影的域差异主要存在的低水平特征空间。UDAMAR 可以同时从具有已知标签的模拟数据中学习 MAR,并从未标记的实际数据中提取关键信息。在临床牙科和躯干数据集上的实验表明,UDAMAR 通过优于其监督式骨干网络和两种最先进的无监督方法来展现其优越性。我们通过对模拟金属伪影的实验和各种消融研究来仔细分析 UDAMAR。在模拟中,其接近监督式方法的性能和优于无监督式方法的优势证明了其有效性。关于 UDA 正则化损失的权重、UDA 特征层和用于训练的实际数据量的影响的消融研究进一步证明了 UDAMAR 的稳健性。UDAMAR 提供了简单而整洁的设计,易于实现。这些优势使其成为实用 CT MAR 的一种非常可行的解决方案。