Arabi Hossein, Zaidi Habib
Division of Nuclear Medicine and Molecular Imaging, Geneva University Hospital, CH-1211, Geneva 4, Switzerland.
Department of Nuclear Medicine and Molecular Imaging, University of Groningen, University Medical Center Groningen, 9700 RB, Groningen, Netherlands.
J Imaging Inform Med. 2024 Dec;37(6):3217-3230. doi: 10.1007/s10278-024-01159-x. Epub 2024 Jun 10.
To develop a robust segmentation model, encoding the underlying features/structures of the input data is essential to discriminate the target structure from the background. To enrich the extracted feature maps, contrastive learning and self-learning techniques are employed, particularly when the size of the training dataset is limited. In this work, we set out to investigate the impact of contrastive learning and self-learning on the performance of the deep learning-based semantic segmentation. To this end, three different datasets were employed used for brain tumor and hippocampus delineation from MR images (BraTS and Decathlon datasets, respectively) and kidney segmentation from CT images (Decathlon dataset). Since data augmentation techniques are also aimed at enhancing the performance of deep learning methods, a deformable data augmentation technique was proposed and compared with contrastive learning and self-learning frameworks. The segmentation accuracy for the three datasets was assessed with and without applying data augmentation, contrastive learning, and self-learning to individually investigate the impact of these techniques. The self-learning and deformable data augmentation techniques exhibited comparable performance with Dice indices of 0.913 ± 0.030 and 0.920 ± 0.022 for kidney segmentation, 0.890 ± 0.035 and 0.898 ± 0.027 for hippocampus segmentation, and 0.891 ± 0.045 and 0.897 ± 0.040 for lesion segmentation, respectively. These two approaches significantly outperformed the contrastive learning and the original model with Dice indices of 0.871 ± 0.039 and 0.868 ± 0.042 for kidney segmentation, 0.872 ± 0.045 and 0.865 ± 0.048 for hippocampus segmentation, and 0.870 ± 0.049 and 0.860 ± 0.058 for lesion segmentation, respectively. The combination of self-learning with deformable data augmentation led to a robust segmentation model with no outliers in the outcomes. This work demonstrated the beneficial impact of self-learning and deformable data augmentation on organ and lesion segmentation, where no additional training datasets are needed.
为了开发一个强大的分割模型,对输入数据的潜在特征/结构进行编码对于将目标结构与背景区分开来至关重要。为了丰富提取的特征图,采用了对比学习和自学习技术,特别是当训练数据集的大小时有限的情况下。在这项工作中,我们着手研究对比学习和自学习对基于深度学习的语义分割性能的影响。为此,使用了三个不同的数据集,分别用于从磁共振图像中进行脑肿瘤和海马体描绘(分别为BraTS和Decathlon数据集)以及从计算机断层扫描图像中进行肾脏分割(Decathlon数据集)。由于数据增强技术也旨在提高深度学习方法的性能,因此提出了一种可变形数据增强技术,并将其与对比学习和自学习框架进行比较。在应用和不应用数据增强、对比学习和自学习的情况下,评估了这三个数据集的分割准确性,以单独研究这些技术的影响。自学习和可变形数据增强技术在肾脏分割中的Dice指数分别为0.913±0.030和0.920±0.022,在海马体分割中为0.890±0.035和0.898±0.027,在病变分割中为0.891±0.045和0.897±0.040,表现出可比的性能。这两种方法在肾脏分割中的Dice指数分别为0.871±0.039和0.868±0.042,在海马体分割中为0.872±0.045和0.865±0.048,在病变分割中为0.870±0.049和0.860±0.058,显著优于对比学习和原始模型。自学习与可变形数据增强的结合产生了一个强大的分割模型,结果中没有异常值。这项工作证明了自学习和可变形数据增强对器官和病变分割的有益影响,其中不需要额外的训练数据集。
J Imaging Inform Med. 2024-12
IEEE J Transl Eng Health Med. 2022
Insights Imaging. 2023-8-25
Sensors (Basel). 2023-4-23
IEEE Trans Biomed Eng. 2023-9
Comput Biol Med. 2023-3
Nat Commun. 2022-7-15
Front Digit Health. 2022-1-4