Multiple representation contrastive self-supervised learning for pulmonary nodule detection

被引:0
作者
Torki, Asghar [1 ]
Adibi, Peyman [1 ]
Kashani, Hamidreza Baradaran [1 ]
机构
[1] Univ Isfahan, Fac Comp Engn, Artificial Intelligence Dept, Esfahan, Iran
关键词
Contrastive learning; Self-supervised learning; Transformation invariant subspaces; Representation learning; Pulmonary nodule detection;
D O I
10.1016/j.knosys.2024.112307
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Self-supervised learning aims to create semantic-enriched representation from unannotated data. A prevalent strategy in this field involves training a unified representation space that is invariant to various transformation combinations. However, creating a single invariant representation to multiple transformations poses several challenges. The efficacy of such a representation space depends on factors such as the intensity, sequence, and various combination scenarios of transformations. As a result, features generated in single representation space may exhibit limited adaptability for subsequent tasks. In contrast to the conventional SSL training approach, we introduce a novel method that involves constructing multiple atomic transformation-invariant representation subspaces. Each subspace in the proposed method is invariant to a specific atomic transformation from a predefined reference set. Our method offers increased flexibility by enabling the downstream task to weigh every atomic transformation-invariant subspace based on the desired feature space. A series of experiments were conducted to compare our approach to traditional self-supervised learning methods in order to assess its effectiveness. This evaluation encompassed diverse data regimes, datasets, evaluation protocols, and perspectives on source-destination data distribution. Our results highlight the superiority of our method compared to training strategies based on single transformation-invariant representation spaces. Additionally, our proposed method demonstrated superior performance in reducing false positives in the context of pulmonary nodule detection when compared to several recent supervised and self-supervised approaches.
引用
收藏
页数:13
相关论文
共 60 条
[11]  
Dosovitskiy A, 2014, ADV NEUR IN, V27
[12]   Self-Supervised Representation Learning: Introduction, advances, and challenges [J].
Ericsson, Linus ;
Gouk, Henry ;
Loy, Chen Change ;
Hospedales, Timothy M. .
IEEE SIGNAL PROCESSING MAGAZINE, 2022, 39 (03) :42-62
[13]   Recurrent attention network for false positive reduction in the detection of pulmonary nodules in thoracic CT scans [J].
Farhangi, M. Mehdi ;
Petrick, Nicholas ;
Sahiner, Berkman ;
Frigui, Hichem ;
Amini, Amir A. ;
Pezeshk, Aria .
MEDICAL PHYSICS, 2020, 47 (05) :2150-2160
[14]   NiftyNet: a deep-learning platform for medical imaging [J].
Gibson, Eli ;
Li, Wenqi ;
Sudre, Carole ;
Fidon, Lucas ;
Shakir, Dzhoshkun I. ;
Wang, Guotai ;
Eaton-Rosen, Zach ;
Gray, Robert ;
Doel, Tom ;
Hu, Yipeng ;
Whyntie, Tom ;
Nachev, Parashkev ;
Modat, Marc ;
Barratt, Dean C. ;
Ourselin, Sebastien ;
Cardoso, M. Jorge ;
Vercauteren, Tom .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2018, 158 :113-122
[15]  
Glorot X., 2010, P 13 INT C ART INT S, P249
[16]  
Grill J.-B., 2020, P 34 INT C NEUR INF, P21271, DOI DOI 10.48550/ARXIV.2006.07733
[17]  
Haghighi F, 2022, PROC CVPR IEEE, P20792, DOI [10.1109/CVPR52688.2022.02016, 10.1109/cvpr52688.2022.02016]
[18]   Transferable Visual Words: Exploiting the Semantics of Anatomical Patterns for Self-Supervised Learning [J].
Haghighi, Fatemeh ;
Taher, Mohammad Reza Hosseinzadeh ;
Zhou, Zongwei ;
Gotway, Michael B. ;
Liang, Jianming .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2021, 40 (10) :2857-2868
[19]  
Haghighi Fatemeh, 2020, Med Image Comput Comput Assist Interv, V12261, P137, DOI 10.1007/978-3-030-59710-8_14
[20]  
Hataya Ryuichiro, 2020, EUR C COMP VIS, P1