Res-trans networks for lung nodule classification

被引:30
作者
Liu, Dongxu [1 ]
Liu, Fenghui [2 ]
Tie, Yun [1 ]
Qi, Lin [1 ]
Wang, Feng [3 ]
机构
[1] Zhengzhou Univ, Sch Informat Engn, Zhengzhou, Peoples R China
[2] Zhengzhou Univ, Dept Resp & Sleep Med, Affiliated Hosp 1, Zhengzhou, Peoples R China
[3] Zhengzhou Univ, Dept Oncol, Affiliated Hosp 1, Zhengzhou, Peoples R China
关键词
Deep learning; Transformer; Lung nodules classification; Computer-aided diagnosis;
D O I
10.1007/s11548-022-02576-5
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Purpose Lung cancer usually presents as pulmonary nodules on early diagnostic images, and accurately estimating the malignancy of pulmonary nodules is crucial to the prevention and diagnosis of lung cancer. Recently, deep learning algorithms based on convolutional neural networks have shown potential for pulmonary nodules classification. However, the size of the nodules is very diverse, ranging from 3 to 30 mm, which makes classifying them to be a challenging task. In this study, we propose a novel architecture called Res-trans networks to classify nodules in computed tomography (CT) scans. Methods We designed local and global blocks to extract features that capture the long-range dependencies between pixels to adapt to the correct classification of lung nodules of different sizes. Specifically, we designed residual blocks with convolutional operations to extract local features and transformer blocks with self-attention to capture global features. Moreover, the Res-trans network has a sequence fusion block that aggregates and extracts the sequence feature information output by the transformer block that improves classification accuracy. Results Our proposed method is extensively evaluated on the public LIDC-IDRI dataset, which contains 1,018 CT scans. A tenfold cross-validation result shows that our method obtains better performance with AUC = 0.9628 and Accuracy = 0.9292 compared with recently leading methods. Conclusion In this paper, a network that can capture local and global features is proposed to classify nodules in chest CT. Experimental results show that our proposed method has better classification performance and can help radiologists to accurately analyze lung nodules.
引用
收藏
页码:1059 / 1068
页数:10
相关论文
共 27 条
[1]   Intensity-based statistical features for classification of lungs CT scan nodules using artificial intelligence techniques [J].
Akram, Sheeraz ;
Javed, Muhammad Younus ;
Hussain, Ayyaz ;
Riaz, Farhan ;
Akram, M. Usman .
JOURNAL OF EXPERIMENTAL & THEORETICAL ARTIFICIAL INTELLIGENCE, 2015, 27 (06) :737-751
[2]   3D axial-attention for lung nodule classification [J].
Al-Shabi, Mundher ;
Shak, Kelvin ;
Tan, Maxine .
INTERNATIONAL JOURNAL OF COMPUTER ASSISTED RADIOLOGY AND SURGERY, 2021, 16 (08) :1319-1324
[3]   Lung nodule classification using deep Local-Global networks [J].
Al-Shabi, Mundher ;
Lan, Boon Leong ;
Chan, Wai Yee ;
Ng, Kwan-Hoong ;
Tan, Maxine .
INTERNATIONAL JOURNAL OF COMPUTER ASSISTED RADIOLOGY AND SURGERY, 2019, 14 (10) :1815-1819
[4]   The Lung Image Database Consortium, (LIDC) and Image Database Resource Initiative (IDRI): A Completed Reference Database of Lung Nodules on CT Scans [J].
Armato, Samuel G., III ;
McLennan, Geoffrey ;
Bidaut, Luc ;
McNitt-Gray, Michael F. ;
Meyer, Charles R. ;
Reeves, Anthony P. ;
Zhao, Binsheng ;
Aberle, Denise R. ;
Henschke, Claudia I. ;
Hoffman, Eric A. ;
Kazerooni, Ella A. ;
MacMahon, Heber ;
van Beek, Edwin J. R. ;
Yankelevitz, David ;
Biancardi, Alberto M. ;
Bland, Peyton H. ;
Brown, Matthew S. ;
Engelmann, Roger M. ;
Laderach, Gary E. ;
Max, Daniel ;
Pais, Richard C. ;
Qing, David P-Y ;
Roberts, Rachael Y. ;
Smith, Amanda R. ;
Starkey, Adam ;
Batra, Poonam ;
Caligiuri, Philip ;
Farooqi, Ali ;
Gladish, Gregory W. ;
Jude, C. Matilda ;
Munden, Reginald F. ;
Petkovska, Iva ;
Quint, Leslie E. ;
Schwartz, Lawrence H. ;
Sundaram, Baskaran ;
Dodd, Lori E. ;
Fenimore, Charles ;
Gur, David ;
Petrick, Nicholas ;
Freymann, John ;
Kirby, Justin ;
Hughes, Brian ;
Casteele, Alessi Vande ;
Gupte, Sangeeta ;
Sallam, Maha ;
Heath, Michael D. ;
Kuhn, Michael H. ;
Dharaiya, Ekta ;
Burns, Richard ;
Fryd, David S. .
MEDICAL PHYSICS, 2011, 38 (02) :915-931
[5]   GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond [J].
Cao, Yue ;
Xu, Jiarui ;
Lin, Stephen ;
Wei, Fangyun ;
Hu, Han .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, :1971-1980
[6]   AutoFormer: Searching Transformers for Visual Recognition [J].
Chen, Minghao ;
Peng, Houwen ;
Fu, Jianlong ;
Ling, Haibin .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :12250-12260
[7]   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
[8]   Learning efficient, explainable and discriminative representations for pulmonary nodules classification [J].
Jiang, Hanliang ;
Shen, Fuhao ;
Gao, Fei ;
Han, Weidong .
PATTERN RECOGNITION, 2021, 113
[9]   Attentive and ensemble 3D dual path networks for pulmonary nodules classification [J].
Jiang, Hanliang ;
Gao, Fei ;
Xu, Xingxin ;
Huang, Fei ;
Zhu, Suguo .
NEUROCOMPUTING, 2020, 398 :422-430
[10]   Random forest based lung nodule classification aided by clustering [J].
Lee, S. L. A. ;
Kouzani, A. Z. ;
Hub, E. J. .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2010, 34 (07) :535-542