Skip to main navigation Skip to search Skip to main content

Lung Nodule Classification of CT images Using Channel and Spatial Attention CNN with Bayesian Optimization

  • Harbin Institute of Technology
  • Institute of Industrial and Electrical Testing

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Lung cancer is the highest cause of mortality among cancers related diseases. Early diagnosis and treatment of lung cancer patients can improve their survival rates. Usually, for patients' diagnosis, Computed Tomography (CT) images are manually diagnosed by radiologists, which is a huge burden to them and sometimes leads to inaccuracy. Deep learning techniques have been proved as a popular and influential method in many medical imaging diagnosis areas. This paper proposes an automated computer-aided diagnosis (CAD) algorithm that uses the channel and spatial attention Convolution Neural Network (CNN) with Bayesian optimization to classify benign and malignant lung nodules from chest CT scans. We applied Bayesian optimization to determine the optimal deep CNN architecture by searching hyperparameters values such as number of training epochs, training batch size, number of layers, filter size, and number of filters. We validated the proposed network on CT scan from The Lung Nodule Analysis 2016 (LUNA16) public dataset, which comprises 888 CT scans. Experimental results show that the proposed algorithm gives accuracy, sensitivity, specificity, and ${F}{1}-$ score of 9S.2l%, 96.95%, 99.14%, and 9S.03%, respectively. We found that the proposed system obtained competent results compared with the network algorithms in the literature review.

Original languageEnglish
Title of host publication2021 Global Reliability and Prognostics and Health Management, PHM-Nanjing 2021
EditorsWei Guo, Steven Li
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665401302
DOIs
StatePublished - 2021
Event12th IEEE Global Reliability and Prognostics and Health Management, PHM-Nanjing 2021 - Nanjing, China
Duration: 15 Oct 202117 Oct 2021

Publication series

Name2021 Global Reliability and Prognostics and Health Management, PHM-Nanjing 2021

Conference

Conference12th IEEE Global Reliability and Prognostics and Health Management, PHM-Nanjing 2021
Country/TerritoryChina
CityNanjing
Period15/10/2117/10/21

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 3 - Good Health and Well-being
    SDG 3 Good Health and Well-being

Keywords

  • Bayesian optimization
  • CT images
  • Computer-aided diagnosis
  • Deep convolutional neural network
  • Lung nodules

Fingerprint

Dive into the research topics of 'Lung Nodule Classification of CT images Using Channel and Spatial Attention CNN with Bayesian Optimization'. Together they form a unique fingerprint.

Cite this