학술논문
Evaluation of data balancing techniques in 3D CNNs for the classification of pulmonary nodules in CT images
Document Type
Conference
Author
Source
2020 IEEE Symposium on Computers and Communications (ISCC) Computers and Communications (ISCC), 2020 IEEE Symposium on. :1-6 Jul, 2020
Subject
Language
ISSN
2642-7389
Abstract
Lung cancer is the most prevalent cancer in the world and early detection and diagnosis enable more treatment options and a far greater chance of survival. In this work, we propose an algorithm based on 3D Convolutional Neural Network (CNN) to classify pulmonary nodules as benign or malignant in computed tomography images. Three architecture of 3D CNNs are proposed, containing different input sizes and numbers of convolutional layers. In addition, we investigated data augmentation techniques and modifications in the network training cost function to address the problem of imbalanced data. The best result was achieved for input size of 32×32×32 pixels, 2 blocks of convolutional layers and 2 pooling layers. Also, the modification of cost function achieved promising results, with accuracy of 0.9188, kappa of 0.8019, sensitivity of 0.8481, specificity of 0.9479 and AUC of 0.8980 in the test set during malignant nodule detection.