Attention-guided deep neural network with a multichannel architecture for lung nodule classification

被引:6
作者
Zheng, Rong [1 ]
Wen, Hongqiao [2 ]
Zhu, Feng [3 ,4 ]
Lan, Weishun [5 ]
机构
[1] Huazhong Univ Sci & Technol, Maternal & Child Hlth Hosp Hubei Prov, Tongji Med Coll, Dept Gynecol, Wuhan 430070, Peoples R China
[2] Wuhan Univ Technol, Sch Informat Engn, Wuhan 430070, Peoples R China
[3] Huazhong Univ Sci & Technol, Union Hosp, Tongji Med Coll, Dept Cardiol, Wuhan, Peoples R China
[4] Huazhong Univ Sci & Technol, Union Hosp, Tongji Med Coll, Clin Ctr Human Gene Res, Wuhan, Peoples R China
[5] Huazhong Univ Sci & Technol, Maternal & Child Hlth Hosp Hubei Prov, Tongji Med Coll, Dept Med Imaging, Wuhan 430070, Peoples R China
基金
中国国家自然科学基金;
关键词
Attention mechanism; Deep learning; lung nodule classification; COMPUTER-AIDED DIAGNOSIS; FALSE-POSITIVE REDUCTION; PULMONARY NODULES; TEXTURE; INFORMATION; SHAPE;
D O I
10.1016/j.heliyon.2023.e23508
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Detecting and accurately identifying malignant lung nodules in chest CT scans in a timely manner is crucial for effective lung cancer treatment. This study introduces a deep learning model featuring a multi-channel attention mechanism, specifically designed for the precise diagnosis of malignant lung nodules. To start, we standardized the voxel size of CT images and generated three RGB images of varying scales for each lung nodule, viewed from three different angles. Subsequently, we applied three attention submodels to extract class-specific characteristics from these RGB images. Finally, the nodule features were consolidated in the model's final layer to make the ultimate predictions. Through the utilization of an attention mechanism, we could dynamically pinpoint the exact location of lung nodules in the images without the need for prior segmentation. This proposed approach enhances the accuracy and efficiency of lung nodule classification. We evaluated and tested our model using a dataset of 1018 CT scans sourced from the Lung Image Database Consortium and Image Database Resource Initiative (LIDC-IDRI). The experimental results demonstrate that our model achieved a lung nodule classification accuracy of 90.11 %, with an area under the receiver operator curve (AUC) score of 95.66 %. Impressively, our method achieved this high level of performance while utilizing only 29.09 % of the time needed by the mainstream model.
引用
收藏
页数:10
相关论文
共 37 条
[11]  
Han F., 2013, A Texture Feature Analysis for Diagnosis of Pulmonary Nodules Using LIDC-IDRI Database
[12]   Texture Feature Analysis for Computer-Aided Diagnosis on Pulmonary Nodules [J].
Han, Fangfang ;
Wang, Huafeng ;
Zhang, Guopeng ;
Han, Hao ;
Song, Bowen ;
Li, Lihong ;
Moore, William ;
Lu, Hongbing ;
Zhao, Hong ;
Liang, Zhengrong .
JOURNAL OF DIGITAL IMAGING, 2015, 28 (01) :99-115
[13]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[14]   Deep Neural Networks for Acoustic Modeling in Speech Recognition [J].
Hinton, Geoffrey ;
Deng, Li ;
Yu, Dong ;
Dahl, George E. ;
Mohamed, Abdel-rahman ;
Jaitly, Navdeep ;
Senior, Andrew ;
Vanhoucke, Vincent ;
Patrick Nguyen ;
Sainath, Tara N. ;
Kingsbury, Brian .
IEEE SIGNAL PROCESSING MAGAZINE, 2012, 29 (06) :82-97
[15]   Deep learning [J].
LeCun, Yann ;
Bengio, Yoshua ;
Hinton, Geoffrey .
NATURE, 2015, 521 (7553) :436-444
[16]   Dense Convolutional Binary-Tree Networks for Lung Nodule Classification [J].
Liu, Yijing ;
Hao, Pengyi ;
Zhang, Peng ;
Xu, Xinnan ;
Wu, Jian ;
Chen, Wei .
IEEE ACCESS, 2018, 6 :49080-49088
[17]   Lung Nodule Classification via Deep Transfer Learning in CT Lung Images [J].
Medeiros da Nobrega, Raul Victor ;
Peixoto, Solon Alves ;
da Silva, Suane Pires P. ;
Reboucas Filho, Pedro Pedrosa .
2018 31ST IEEE INTERNATIONAL SYMPOSIUM ON COMPUTER-BASED MEDICAL SYSTEMS (CBMS 2018), 2018, :244-249
[18]  
Mnih V, 2014, ADV NEUR IN, V27
[19]  
Mustofa F., 2023, Journal of Computing Theories and Applications, V1, P41, DOI 10.33633/jcta.v1i1.9190
[20]  
Papanastasiou G, 2023, Arxiv, DOI arXiv:2307.12775