17 February 2021 A-DARTS: attention-guided differentiable architecture search for lung nodule classification
Liangxiao Hu, Qinglin Liu, Jun Zhang, Feng Jiang, Yang Liu, Shengping Zhang
Author Affiliations +
Abstract

Lung cancer has caused the most cancer deaths in the past several years. Benign–malignant lung nodule classification is vital in lung nodule detection, which can help early diagnosis of lung cancer. Most existing works extract the features of chest CT images using the well-designed networks, which require substantial effort of experts. To automate the manual process of network design, we propose an attention-guided differentiable architecture search (A-DARTS) method, which directly searches for the optimal network on chest CT images. In addition, A-DARTS utilizes an attention mechanism to alleviate the effect of the initialization-sensitive nature of the searched network while enhancing the feature presentation ability. Extensive experiments on the Lung Image Database Consortium image collection (LIDC-IDRI) benchmark dataset show that the proposed method achieves a lung nodule classification accuracy of 92.93%, which is superior to the state-of-the-art methods.

© 2021 SPIE and IS&T 1017-9909/2021/$28.00© 2021 SPIE and IS&T
Liangxiao Hu, Qinglin Liu, Jun Zhang, Feng Jiang, Yang Liu, and Shengping Zhang "A-DARTS: attention-guided differentiable architecture search for lung nodule classification," Journal of Electronic Imaging 30(1), 013012 (17 February 2021). https://doi.org/10.1117/1.JEI.30.1.013012
Received: 10 November 2020; Accepted: 20 January 2021; Published: 17 February 2021
Lens.org Logo
CITATIONS
Cited by 1 scholarly publication.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Lung

Computed tomography

Chest

Lung cancer

Network architectures

Convolution

Image classification

Back to Top