Feature Representation Using Deep Autoencoder for Lung Nodule Image Classification

• Published in: Complexity (New York, N.Y.) Vol. 2018; no. 2018; pp. 1 - 11
• Main Authors: Zhang, Weiyi, Wang, Xinqi, Tang, Renjie, Mao, Keming, Wu, Haoxiang
• Format: Journal Article
• Language: English
• Published: Cairo, Egypt Hindawi Publishing Corporation 01-01-2018; Hindawi; John Wiley & Sons, Inc; Hindawi Limited; Hindawi-Wiley
• Subjects: Algorithms; Analysis; Artificial intelligence; Classification; Diagnosis; Dictionaries; Drug dosages; Image classification; International conferences; Lung cancer; Medical imaging; Medical imaging equipment; Medical screening; Methods; Patches (structures); Representations; Statistical analysis; Tomography; China
• ISSN: 1076-2787; 1099-0526
• DOI: 10.1155/2018/3078374

This paper focuses on the problem of lung nodule image classification, which plays a key role in lung cancer early diagnosis. In this work, we propose a novel model for lung nodule image feature representation that incorporates both local and global characters. First, lung nodule images are divided into local patches with Superpixel. Then these patches are transformed into fixed-length local feature vectors using unsupervised deep autoencoder (DAE). The visual vocabulary is constructed based on the local features and bag of visual words (BOVW) is used to describe the global feature representation of lung nodule image. Finally, softmax algorithm is employed for lung nodule type classification, which can assemble the whole training process as an end-to-end mode. Comprehensive evaluations are conducted on the widely used public available ELCAP lung image database. Experimental results with regard to different parameter setting, data augmentation, model sparsity, classifier algorithms, and model ensemble validate the effectiveness of our proposed approach.