Lung cancer classification using convolutional neural network and DenseNet
Main Article Content
Abstract
Lung cancer is a condition that has a major impact on public health. Convolutional Neural Network (CNN) and DenseNet approaches are suggested in this study to aid lung cancer detection and classification. In various fields of pattern recognition and medical imaging, CNN and DenseNet have demonstrated their efficacy. In this study, radiology images from individuals with lung cancer were used to create a set of medical lung images. The findings show that lung cancer can be accurately classified into malignant and benign from radiological images using CNN and DenseNet architectures, with a parameter accuracy of 99.48%. This research contributes to the creation of a deep learning-based system for detecting and classifying lung cancer. The findings can be the basis for creating a more accurate and productive lung cancer diagnostic system.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
References
R. L. Siegel, K. D. Miller, H. E. Fuchs, and A. Jemal, “Cancer Statistics, 2021.,” CA. Cancer J. Clin., vol. 71, no. 1, pp. 7–33, Jan. 2021, doi: 10.3322/caac.21654.
S. Lei et al., “Global patterns of breast cancer incidence and mortality: A population‐based cancer registry data analysis from 2000 to 2020,” Cancer Commun., vol. 41, no. 11, pp. 1183–1194, 2021.
M. M. Cackowski et al., “The absence of lymph nodes removed (pNx status) impacts survival in patients with lung cancer treated surgically,” Surg. Oncol., vol. 48, p. 101941, 2023.
S. S. Raoof, M. A. Jabbar, and S. A. Fathima, “Lung Cancer Prediction using Machine Learning: A Comprehensive Approach,” in 2020 2nd International Conference on Innovative Mechanisms for Industry Applications (ICIMIA), 2020, pp. 108–115. doi: 10.1109/ICIMIA48430.2020.9074947.
Y. Chen, E. Zitello, R. Guo, and Y. Deng, “The function of LncRNAs and their role in the prediction, diagnosis, and prognosis of lung cancer,” Clin. Transl. Med., vol. 11, no. 4, p. e367, 2021.
J. A. Barta, C. A. Powell, and J. P. Wisnivesky, “Global epidemiology of lung cancer,” Ann. Glob. Heal., vol. 85, no. 1, 2019.
M. A. Heuvelmans et al., “Lung cancer prediction by Deep Learning to identify benign lung nodules,” Lung Cancer, vol. 154, no. November 2020, pp. 1–4, 2021, doi: 10.1016/j.lungcan.2021.01.027.
J. Jumanto, M. F. Mardiansyah, R. Pratama, M. F. Al Hakim, and B. Rawat, “Optimization of breast cancer classification using feature selection on neural network,” J. Soft Comput. Explor., vol. 3, no. 2, pp. 105–110, 2022, doi: 10.52465/joscex.v3i2.78.
B. He et al., “Image segmentation algorithm of lung cancer based on neural network model,” Expert Syst., vol. 39, no. 3, p. e12822, 2022.
Q. Wang, Y. Zhou, W. Ding, Z. Zhang, K. Muhammad, and Z. Cao, “Random forest with self-paced bootstrap learning in lung cancer prognosis,” ACM Trans. Multimed. Comput. Commun. Appl., vol. 16, no. 1s, pp. 1–12, 2020.
B. K. Hatuwal and H. C. Thapa, “Lung cancer detection using convolutional neural network on histopathological images,” Int. J. Comput. Trends Technol, vol. 68, no. 10, pp. 21–24, 2020.
A. B. Pawar et al., “Implementation of blockchain technology using extended CNN for lung cancer prediction,” Meas. Sensors, vol. 24, no. September, p. 100530, 2022, doi: 10.1016/j.measen.2022.100530.
R. H. Saputra and B. Prasetyo, “Improve the Accuracy of C4.5 Algorithm Using Particle Swarm Optimization (PSO) Feature Selection and Bagging Technique in Breast Cancer Diagnosis,” J. Soft Comput. Explor., vol. 1, no. 1, pp. 47–55, 2020, doi: 10.52465/joscex.v1i1.9.
A. Masood et al., “Automated Decision Support System for Lung Cancer Detection and Classification via Enhanced RFCN with Multilayer Fusion RPN,” IEEE Trans. Ind. Informatics, vol. 16, no. 12, pp. 7791–7801, 2020, doi: 10.1109/TII.2020.2972918.
S. Wankhade and S. Vigneshwari, “A novel hybrid deep learning method for early detection of lung cancer using neural networks,” Healthc. Anal., p. 100195, 2023.
S. P. Kannojia and G. Jaiswal, “Ensemble of hybrid CNN-ELM model for image classification,” in 2018 5th international conference on signal processing and integrated networks (SPIN), 2018, pp. 538–541.
A. Ajit, K. Acharya, and A. Samanta, “A review of convolutional neural networks,” in 2020 international conference on emerging trends in information technology and engineering (ic-ETITE), 2020, pp. 1–5.
S. Haggenmüller et al., “Skin cancer classification via convolutional neural networks: systematic review of studies involving human experts,” Eur. J. Cancer, vol. 156, pp. 202–216, 2021, doi: 10.1016/j.ejca.2021.06.049.
S. Kuntz et al., “Gastrointestinal cancer classification and prognostication from histology using deep learning: Systematic review,” Eur. J. Cancer, vol. 155, pp. 200–215, 2021, doi: 10.1016/j.ejca.2021.07.012.
S. Kilicarslan, K. Adem, and M. Celik, “Diagnosis and classification of cancer using hybrid model based on ReliefF and convolutional neural network,” Med. Hypotheses, vol. 137, no. December 2019, p. 109577, 2020, doi: 10.1016/j.mehy.2020.109577.
B. Fang, Y. Li, H. Zhang, and J. C.-W. Chan, “Hyperspectral images classification based on dense convolutional networks with spectral-wise attention mechanism,” Remote Sens., vol. 11, no. 2, p. 159, 2019.
A. Garcia-Diaz and H. Bersini, “DensEMANN: building a DenseNet from scratch, layer by layer and kernel by kernel,” in 2021 International Joint Conference on Neural Networks (IJCNN), 2021, pp. 1–10.
T. Takase, “Dynamic batch size tuning based on stopping criterion for neural network training,” Neurocomputing, vol. 429, pp. 1–11, 2021.
O. Rukundo, “Effects of image size on deep learning,” Electronics, vol. 12, no. 4, p. 985, 2023.
O. G. Ajayi and J. Ashi, “Effect of varying training epochs of a Faster Region-Based Convolutional Neural Network on the Accuracy of an Automatic Weed Classification Scheme,” Smart Agric. Technol., vol. 3, p. 100128, 2023.
L. Zhou, C. Zhang, F. Liu, Z. Qiu, and Y. He, “Application of deep learning in food: a review,” Compr. Rev. food Sci. food Saf., vol. 18, no. 6, pp. 1793–1811, 2019.
Ž. Kokalj and M. Somrak, “Why not a single image? Combining visualizations to facilitate fieldwork and on-screen mapping,” Remote Sens., vol. 11, no. 7, 2019, doi: 10.3390/rs11070747.
P. Kiran, B. D. Parameshachari, J. Yashwanth, and K. N. Bharath, “Offline Signature Recognition Using Image Processing Techniques and Back Propagation Neuron Network System,” SN Comput. Sci., vol. 2, no. 3, pp. 1–8, 2021, doi: 10.1007/s42979-021-00591-y.
J. Brownlee, Deep learning for computer vision: image classification, object detection, and face recognition in python. Machine Learning Mastery, 2019.
O. Ali, N. Scheidt, A. Gegov, E. Haig, M. Adda, and B. Aziz, “Automated Detection of Racial Microaggressions using Machine Learning,” 2020 IEEE Symp. Ser. Comput. Intell. SSCI 2020, no. December, pp. 2477–2484, 2020, doi: 10.1109/SSCI47803.2020.9308569.