Research Article |
PSbBO-Net: A Hybrid Particle Swarm and Bayesian Optimization-based DenseNet for Lung Cancer Detection using Histopathological and CT Images
Author(s): Saurabh Singh Raghuvanshi*, K. V. Arya and Vinal Patel
Published In : International Journal of Electrical and Electronics Research (IJEER) Volume 12, Issue 3
Publisher : FOREX Publication
Published : 25 September 2024
e-ISSN : 2347-470X
Page(s) : 1074-1086
Abstract
Lung cancer remains a substantial global fatality; early detection is imperative for successful intervention and treatment. Deep learning (DL) models have shown promise in predicting lung cancer from medical images, but optimizing their parameters remains a challenging task. To improve prediction capability, this study introduces an approach by merging Particle Swarm Optimization and Bayesian Optimization (PSbBO) to optimize deep learning parameters. PSO provides an effective way for exploring the hyperparameter space, while Bayesian optimization provides a probabilistic framework for the effective evaluation and refining of a DL network. The simulation study showcases the effectiveness of the proposed model, achieving notable metrics for histopathological images, including an accuracy of 99.5%, precision of 98.3%, recall of 99.2%, F1-score of 99.4%, and an error rate of 1.19%. Furthermore, when applied to lung CT images, the proposed PSbBO demonstrates an accuracy of 98.8%, precision of 97.4%, recall of 98.3%, F1-score of 98.6%, and an error rate of 1.21%.
Keywords: Lung cancer detection
, Deep learning
, Particle Swarm Optimization
, Bayesian Optimization
.
Saurabh Singh Raghuvanshi*, Multimedia and Information Security Research Group, Department of Computer Science and Engineering, ABV-Indian Institute of Information Technology and Management, Gwalior-474015, India; Email: saurabhr@iiitm.ac.in
K. V. Arya, Multimedia and Information Security Research Group, Department of Computer Science and Engineering, ABV-Indian Institute of Information Technology and Management, Gwalior-474015, India; Email: kvarya@iiitm.ac.in
Vinal Patel, Department of Electrical and Electronics Engineering, ABV-Indian Institute of Information Technology and Management, Gwalior-474015, India; Email: vp@iiitm.ac.in
-
[1] Sharma, P., Mehta, M., Dhanjal, D. S., Kaur, S., Gupta, G., Singh, H.,& Satija, S. (2019). Emerging trends in the novel drug delivery approaches for the treatment of lung cancer. Chemico-biological interactions, 309, p.108720.
-
[2] Huang, S., Yang, J., Fong, S., & Zhao, Q. (2020). Artificial intelligence in cancer diagnosis and prognosis: Opportunities and challenges. Cancer Letters, 471, pp.61-71.
-
[3] Dhiman, P., Ma, J., Navarro, C. A., Speich, B., Bullock, G., Damen, J. A.,& Collins, G. S. (2021). Reporting of prognostic clinical prediction models based on machine learning methods in oncology needs to be improved. Journal of Clinical Epidemiology, 138, pp.60-72.
-
[4] Ibrahim, D. M., Elshennawy, N. M., & Sarhan, A. M. (2021). Deep-chest: Multi-classification deep learning model for diagnosing COVID-19, pneumonia, and lung cancer chest diseases. Computers in biology and medicine, 132, p.104348.
-
[5] Toh, T. S., Dondelinger, F., & Wang, D. (2019). Looking beyond the hype: applied AI and machine learning in translational medicine. EBioMedicine, 47, pp.607-615.
-
[6] Vaidya, P., Bera, K., Gupta, A., Wang, X., Corredor, G., Fu, P.,& Madabhushi, A. (2020). CT derived radiomic score for predicting the added benefit of adjuvant chemotherapy following surgery in stage I, II resectable non-small cell lung cancer: a retrospective multicohort study for outcome prediction. The Lancet Digital Health, 2(3), pp.116-128.
-
[7] Lakshmanaprabu, S. K., Mohanty, S. N., Shankar, K., Arunkumar, N., & Ramirez, G. (2019). Optimal deep learning model for classification of lung cancer on CT images. Future Generation Computer Systems, 92, pp.374-382.
-
[8] Zade, A. E., Haghighi, S. S., & Soltani, M. (2022). Deep neural networks for neuro-oncology: Towards patient individualized design of chemo-radiation therapy for Glioblastoma patients. Journal of Biomedical Informatics, 127, p.104006.
-
[9] Ilhan, B., Guneri, P., & Wilder-Smith, P. (2021). The contribution of artificial intelligence to reducing the diagnostic delay in oral cancer. Oral oncology, 116, p.105254.
-
[10] Xie, Y., Meng, W. Y., Li, R. Z., Wang, Y. W., Qian, X., Chan, C., & Leung, E. L. H. (2021). Early lung cancer diagnostic biomarker discovery by machine learning methods. Translational oncology, 14(1), p.100907.
-
[11] Seijo, L. M., Peled, N., Ajona, D., Boeri, M., Field, J. K., Sozzi, G., & Montuenga, L. M. (2019). Biomarkers in lung cancer screening: achievements, promises, and challenges. Journal of Thoracic Oncology, 14(3), pp.343-357.
-
[12] Dunn, B. K., Woloshin, S., Kramer, B. S., & Xie, H. (2022). Cancer overdiagnosis: a challenge in the era of screening. Journal of the National Cancer Center.
-
[13] Curigliano, G., Lenihan, D., Fradley, M., Ganatra, S., Barac, A., Blaes, A., & ESMO Guidelines Committee. (2020). Management of cardiac disease in cancer patients throughout oncological treatment: ESMO consensus recommendations. Annals of Oncology, 31(2), pp.171-190.
-
[14] Dalamaga, M., Christodoulatos, G. S., Karampela, I., Vallianou, N., & Apovian, C. M. (2021). Understanding the co-epidemic of obesity and COVID-19: current evidence, comparison with previous epidemics, mechanisms, and preventive and therapeutic perspectives. Current obesity reports, 10(3), pp.214-243.
-
[15] Bohr, A., & Memarzadeh, K. (2020). The rise of artificial intelligence in healthcare applications. Artificial Intelligence in Healthcare, pp. 25-60.
-
[16] Bode, G., Thul, S., Baranski, M., & Müller, D. (2020). Real-world application of machine-learning-based fault detection trained with experimental data. Energy, 198, p.117323.
-
[17] Subrahmanya, S. V. G., Shetty, D. K., Patil, V., Hameed, B. Z., Paul, R., Smriti, K., & Somani, B. K. (2022). The role of data science in healthcare advancements: applications, benefits, and prospects. Irish Journal of Medical Science (1971- ), 191(4), pp.1473-1483.
-
[18] Jiang, H., Shen, F., Gao, F., & Han, W. (2021). Learning efficient, explainable, and discriminative representations for pulmonary nodules classification. Pattern Recognition, 113, p.107825.
-
[19] Kumar, R., Wang, W., Kumar, J., Yang, T., Khan, A., Ali, W., & Ali, I. (2021). An integration of blockchain and AI for secure data sharing and detection of CT images for hospitals. Computerized Medical Imaging and Graphics, 87, p.101812.
-
[20] Zheng, S., Shen, Z., Pei, C., Ding, W., Lin, H., Zheng, J.,& Huang, L. (2021). Interpretative computer-aided lung cancer diagnosis: From radiology analysis to malignancy evaluation. Computer Methods and Programs in Biomedicine, 210, p.106363.
-
[21] Nanglia, P., Kumar, S., Mahajan, A. N., Singh, P., & Rathee, D. (2021). A hybrid algorithm for lung cancer classification using SVM and Neural Networks. ICT Express, 7(3), pp.335-341.
-
[22] Ananthajothi, K., Rajasekar, P., & Amanullah, M. (2023). Enhanced U-Net-based segmentation and heuristically improved deep neural network for pulmonary emphysema diagnosis. Sādhanā, 48(1), p.33.
-
[23] Ali, I., Muzammil, M., Haq, I. U., Khaliq, A. A., & Abdullah, S. (2020). Efficient lung nodule classification using transferable texture convolutional neural network. IEEE Access, 8, pp.175859-175870.
-
[24] Masood, A., Sheng, B., Yang, P., Li, P., Li, H., Kim, J., & Feng, D. D. (2020). Automated decision support system for lung cancer detection and classification via enhanced RFCN with multilayer fusion RPN. IEEE Transactions on Industrial Informatics, 16(12), pp.7791-7801.
-
[25] Shakeel, P.M., Burhanuddin, M.A. and Desa, M.I., (2019). Lung cancer detection from CT images using improved profuse clustering and deep learning instantaneously trained neural networks. Measurement, 145, pp.702-712.
-
[26] ALzubi, J.A., Bharathikannan, B., Tanwar, S., Manikandan, R., Khanna, A. and Thaventhiran, C., (2019). Boosted neural network ensemble classification for lung cancer disease diagnosis. Applied Soft Computing, 80, pp.579-591.
-
[27] Maleki, N., Zeinali, Y. and Niaki, S.T.A., (2021). A k-NN method for lung cancer prognosis with the use of a genetic algorithm for feature selection. Expert Systems with Applications, 164, p.113981.
-
[28] Tian, Q., Wu, Y., Ren, X. and Razmjooy, N., (2021). A new optimized sequential method for lung tumor diagnosis based on deep learning and converged search and rescue algorithm. Biomedical Signal Processing and Control, 68, p.102761.
-
[29] Ding, L., Zhang, X.Y., Wu, D.Y. and Liu, M.L., (2021). Application of an extreme learning machine network with particle swarm optimization in syndrome classification of primary liver cancer. Journal of Integrative Medicine, 19(5), pp.395-407.
-
[30] Atteia, G., Alhussan, A.A. and Samee, N.A., (2022). Bo-Tallinn: Bayesian-based optimized CNN for acute lymphoblastic leukemia detection in microscopic blood smear images. Sensors, 22(15), p.5520.
-
[31] Shakeel, P.M., Burhanuddin, M.A. and Desa, M.I., (2022). Automatic lung cancer detection from CT image using improved deep neural network and ensemble classifier. Neural Computing and Applications, 34, pp.9579-9592.
-
[32] Ghoniem, R.M., Algarni, A.D., Refky, B. and Ewees, A.A., (2021). Multi-modal evolutionary deep learning model for ovarian cancer diagnosis. Symmetry, 13(4), p.643.
-
[33] Sebastian, A.E., Dua, D., (2023). Lung Nodule Detection via Optimized Convolutional Neural Network: Impact of Improved Moth Flame Algorithm. Sens Imaging, 24, p.11.
-
[34] Bilal, Anas, et al. (2022). Lung nodules detection using grey wolf optimization by weighted filters and classification using CNN. Journal of the Chinese Institute of Engineers, 45(2), pp.175-186.
-
[35] Vidhya, R., and T. T. Mirnalinee (2022). Hybrid Optimized Learning for Lung Cancer Classification. Intelligent Automation & Soft Computing 34(2).
-
[36] Mkindu, Hassan, Longwen Wu, and Yaqin Zhao (2023). Lung nodule detection in chest CT images based on vision transformer network with Bayesian optimization. Biomedical Signal Processing and Control 85, p.104866.
-
[37] Toğaçar, M., Ergen, B. and Cömert, Z., (2020). Detection of lung cancer on chest CT images using minimum redundancy maximum relevance feature selection method with convolutional neural networks. Biocybernetics and Biomedical Engineering, 40(1), pp.23-39.
-
[38] SU, A., PP, F.R., Abraham, A. and Stephen, D., (2022). Deep Learning-Based BoVW–CRNN Model for Lung Tumor Detection in Nano-Segmented CT Images. Electronics, 12(1), p.14.
-
[39] Guan, X., Du, Y., Ma, R., Teng, N., Ou, S., Zhao, H. and Li, X., (2023). Construction of the XGBoost model for early lung cancer prediction based on metabolic indices. BMC Medical Informatics and Decision Making, 23(1), pp.1-16.
-
[40] Albertina, B., Watson, M., Holback, C., Jarosz, R., Kirk, S., Lee, Y., Lemmerman, J. (2016). Radiology Data from The Cancer Genome Atlas Lung Adenocarcinoma [TCGA-LUAD] collection. The Cancer Imaging Archive available at URL: http://doi.org/10.7937/K9/TCIA.2016.JGNIHEP5.
-
[41] Dansana, J., Kabat, M.R. and Pattnaik, P.K., (2023). A Novel Optimized Perturbation-Based Machine Learning for Preserving Privacy in Medical Data. Wireless Personal Communications, 130(3), pp.1905-1927.
-
[42] Borkowski AA, Bui MM, Thomas LB, Wilson CP, DeLand LA, Mastorides SM. Lung and colon cancer histopathological image dataset (LC25000). The LC25000 dataset is available at URL: https://www.kaggle.com/datasets/andrewmvd/lung-and-colon-cancer-histopathological-images.
-
[43] Hany M. Chest CT-scan images dataset. Kaggle. Retrieved November. 2020;13:2022.
-
[44] Akhand ZE, Rahman AI, Sarda A, Fahim MZ, Tushi LT, Azad K, Tahiat HT (2023). Lung Cancer Detection Using Ensemble Technique of CNN. In Proceedings of International Conference on Information and Communication Technology for Development: ICICTD, pp. 497-507.
-
[45] Ashraf, Syed Faaz, Yin K, X. Meng Cindy, Qang Qi, Wang Qiong, Pu Jiantao, Dhupar Rajeev, (2022). Predicting benign, preinvasive, and invasive lung nodules on computed tomography scans using machine learning. The Journal of Thoracic and Cardiovascular Surgery 163.4, pp.1496-1505.
-
[46] El-Askary, Nada S., Mohammed A-M. Salem, and Mohamed I. Roushdy (2022). "Features processing for random forest optimization in lung nodule localization." Expert Systems with Applications 193, p.116489.