Engineering Research Center of Wideband Wireless Communication Technology, Ministry of Education, Nanjing University of Posts and Telecommunications, Nanjing 210003, China.
Engineering Research Center of Wideband Wireless Communication Technology, Ministry of Education, Nanjing University of Posts and Telecommunications, Nanjing 210003, China.
Comput Med Imaging Graph. 2024 Sep;116:102421. doi: 10.1016/j.compmedimag.2024.102421. Epub 2024 Jul 26.
Intracranial aneurysm (IA) is a prevalent disease that poses a significant threat to human health. The use of computed tomography angiography (CTA) as a diagnostic tool for IAs remains time-consuming and challenging. Deep neural networks (DNNs) have made significant advancements in the field of medical image segmentation. Nevertheless, training large-scale DNNs demands substantial quantities of high-quality labeled data, making the annotation of numerous brain CTA scans a challenging endeavor. To address these challenges and effectively develop a robust IAs segmentation model from a large amount of unlabeled training data, we propose a triple learning framework (TLF). The framework primarily consists of three learning paradigms: pseudo-supervised learning, contrastive learning, and confident learning. This paper introduces an enhanced mean teacher model and voxel-selective strategy to conduct pseudo-supervised learning on unreliable labeled training data. Concurrently, we construct the positive and negative training pairs within the high-level semantic feature space to improve the overall learning efficiency of the TLF through contrastive learning. In addition, a multi-scale confident learning is proposed to correct unreliable labels, which enables the acquisition of broader local structural information instead of relying on individual voxels. To evaluate the effectiveness of our method, we conducted extensive experiments on a self-built database of hundreds of cases of brain CTA scans with IAs. Experimental results demonstrate that our method can effectively learn a robust CTA scan-based IAs segmentation model using unreliable labeled data, outperforming state-of-the-art methods in terms of segmentation accuracy. Codes are released at https://github.com/XueShuangqian/TLF.
颅内动脉瘤(IA)是一种常见疾病,对人类健康构成重大威胁。计算机断层血管造影(CTA)作为 IA 的诊断工具仍然耗时且具有挑战性。深度神经网络(DNN)在医学图像分割领域取得了重大进展。然而,训练大规模 DNN 需要大量高质量的标记数据,因此对大量脑 CTA 扫描进行注释是一项具有挑战性的工作。为了解决这些挑战,并有效地从大量未标记的训练数据中开发出强大的 IA 分割模型,我们提出了三重学习框架(TLF)。该框架主要由三种学习范式组成:伪监督学习、对比学习和置信学习。本文提出了一种改进的均值教师模型和体素选择策略,对不可靠的标记训练数据进行伪监督学习。同时,我们在高级语义特征空间中构建正、负训练对,通过对比学习提高 TLF 的整体学习效率。此外,提出了一种多尺度置信学习方法来纠正不可靠的标签,从而能够获取更广泛的局部结构信息,而不是依赖于单个体素。为了评估我们方法的有效性,我们在一个包含数百例脑 CTA 扫描的自建 IA 数据库上进行了广泛的实验。实验结果表明,我们的方法可以有效地利用不可靠的标记数据学习稳健的基于 CTA 扫描的 IA 分割模型,在分割精度方面优于最先进的方法。代码发布在 https://github.com/XueShuangqian/TLF。