Improvement in the Convolutional Neural Network for Computed Tomography Images
Open Access
- 7 February 2021
- journal article
- research article
- Published by MDPI AG in Applied Sciences
- Vol. 11 (4), 1505
- https://doi.org/10.3390/app11041505
Abstract
Background and purpose. This study evaluated a modified specialized convolutional neural network (CNN) to improve the accuracy of medical images. Materials and Methods. We defined computed tomography (CT) images as belonging to one of the following 10 classes: head, neck, chest, abdomen, and pelvis with and without contrast media, with 10,000 images per class. We modified the CNN based on the AlexNet with an input size of 512 × 512. We resized the filter sizes of the convolution layer and max pooling. Using these modified CNNs, various models were created and evaluated. The improved CNN was evaluated to classify the presence or absence of the pancreas in the CT images. We compared the overall accuracy, which was calculated from images not used for training, to that of the ResNet. Results. The overall accuracies of the most improved CNN and ResNet in the 10 classes were 94.8% and 89.3%, respectively. The filter sizes of the improved CNN for the convolution layer were (13, 13), (7, 7), (5, 5), (5, 5), and (5, 5) in order from the first layer, and that of max-pooling was (7, 7). The calculation times of the most improved CNN and ResNet were 56 and 120 min, respectively. Regarding the classification of the pancreas, the overall accuracies of the most improved CNN and ResNet were 75.75% and 58.25%, respectively. The calculation times of the most improved CNN and ResNet were 36 and 55 min, respectively. Conclusion. By optimizing the filter size of the convolution layer and max-pooling of 512 × 512 images, we quickly obtained a highly accurate medical image classification model. This improved CNN can be useful for classifying lesions and anatomies for related diagnostic aid applications.This publication has 21 references indexed in Scilit:
- Deep learning algorithms for detection of critical findings in head CT scans: a retrospective studyThe Lancet, 2018
- Progressive Neural Architecture SearchPublished by Springer Science and Business Media LLC ,2018
- Computer-aided diagnosis of lung nodule classification between benign nodule, primary lung cancer, and metastatic lung cancer at different image size using deep convolutional neural network with transfer learningPLOS ONE, 2018
- Classification of Computed Tomography Images in Different Slice Positions Using Deep LearningJournal of Healthcare Engineering, 2018
- Deep neural networks for automatic detection of osteoporotic vertebral fractures on CT scansComputers in Biology and Medicine, 2018
- Deep Learning at Chest Radiography: Automated Classification of Pulmonary Tuberculosis by Using Convolutional Neural NetworksRadiology, 2017
- ImageNet classification with deep convolutional neural networksCommunications of the ACM, 2017
- Automatic Image-Based Plant Disease Severity Estimation Using Deep LearningComputational Intelligence and Neuroscience, 2017
- Deep Residual Learning for Image RecognitionPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2016
- DeepOrgan: Multi-level Deep Convolutional Networks for Automated Pancreas SegmentationPublished by Springer Science and Business Media LLC ,2015