Publication Details
Keywords: Explainable AI (XAI), LLM, Nodule Detection, Benign, Malignant, Deep learning, Ensemble.
Abstract
Lung cancer stands as the most prevalent form of cancer while maintaining its position as the main reason for cancer deaths worldwide. Detecting lung cancer early and precisely plays an essential role in achieving better clinical results. Medical imaging has reached remarkable success because of deep learning applications whose main focus is lung nodule detection along with classification tasks. The current research extends existing ensemble Convolutional Neural Networks (CNNs) detection methods for lung cancer by adopting modern segmentation protocols and classification frameworks and explainable features. The LUNA16 dataset trained U-Net model performs segmentation of lung nodules in CT scan images before the classification process focuses exclusively on relevant regions. Segmented nodules undergo classification either as malignant or benign through the application of ResNet101, MobileNetV3-small, and a custom CNN classifier, all pre-trained on the LIDC-IDRI dataset. Performance comparisons are made between the ensemble approach and each individual model when their predictions are combined through a soft ensemble framework. Additionally, SHAP (Shapley Additive Explanations) provides important feature explanations that influence classification. The integration of a Large Language Model (LLM) happens subsequent to explainability for the generation of contextual information which aids both clinicians' decision-making and improves interpretation outcomes. The ensemble system achieves superior performance results when compared to standalone networks and its combination with explainable AI and LLM-based interpretation generates a transparent diagnostic tool for lung cancers.
References
- Ahmed, M. S., Iqbal, K. N., & Alam, M. G. R., Interpretable Lung Cancer Detection using Explainable AI Methods. 2023 International Conference for Advancement in Technology (ICONAT), pp. 1-6, Oct. 2023.
- Ali, Zeeshan, Irtaza, Aun, & Maqsood, Muazzam, An efficient U-Net framework for lung nodule detection using densely connected dilated convolutions, The Journal of Supercomputing, Vol.78, pp.1602-1616, Feb. 2022.
- American Cancer Society, Lung cancer. https://www.cancer.org/cancer/lung-cancer.html/, 2022.
- Arnaud Arindra Adiyoso Setio, Alberto Traverso, Thomas de Bel, Moira S.N. Berens, Cas van den Bogaard, Piergiorgio Cerello, Hao Chen, Qi Dou, Maria Evelina Fantacci, Bart Jansen, Nicole Walasek, Guido C.A. Zuidhof, Bram van Ginneken, Colin Jacobs et al, Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: The LUNA16 challenge. Medical Image Analysis, Vol.42, pp.1-13, June 2017.
- Bach, P. B., Jett, J. R., Pastorino, U., Tockman, M. S., Swensen, S. J., & Begg, C. B., Computed tomography screening and lung cancer outcomes, JAMA, Vol. 297 No. 9, pp.953β961, Mar. 2007.
- Claudia I. Henschke, David F. Yankelevitz, William J. Kostis. CT Screening for Lung Cancer, Seminars in Ultrasound, CT and MRI, Vol. 24 No.1, pp.23-32, Feb 2023.
- Esteva, A., Robicquet, A., Ramsundar, B., Kuleshov, V., DePristo, M., Chou, K., Cui, C., Corrado, G., Thrun, S., & Dean, J, A guide to deep learning in healthcare. Nature medicine, Vol. 25 No. 1, pp. 24β29, Aug. 2019.
- Hao R, Namdar K, Liu L, Haider MA, Khalvati F., A Comprehensive Study of Data Augmentation Strategies for Prostate Cancer Detection in Diffusion-Weighted MRI Using Convolutional Neural Networks, J Digit Imaging, Vol. 34 No. 4, pp. 862-876, May 2021.
- M. B. A. Miah & M. A. Yousuf., Detection of lung cancer from CT image using image processing and neural network. 2015 International Conference on Electrical Engineering and Information Communication Technology (ICEEICT), pp.1-6, Oct. 2015.
- Nakrani, Mahender, Sable, Ganesh, & Shinde, Ulhas, ResNet based Lung Nodules Detection from Computed Tomography Images, International Journal of Innovative Technology and Exploring Engineering, Vol. 9, No. 4, pp. 2356-2360, April 2020.
- Siegel, R. L., Miller, K. D., & Jemal, A., Cancer statistics, 2020, CA: a cancer journal for clinicians, Vol. 70, No. 1, pp. 7β30, Jan. 2020.
- S. Niranjan Kumar et al., Lung Nodule Segmentation Using UNet. 2021 7th International Conference on Advanced Computing and Communication Systems (ICACCS), pp.420-424, Mar. 2021.
- World Health Organization, Lung cancer fact sheet. https://www.who.int., 2021.
- J. Gera, A. R. Palakayala, V. K. K. Rejeti, and T. Anusha, β Blockchain Technology for Fraudulent Practices in Insurance Claim Process,β in 2020 5th International Conference on Communication and Electronics Systems (ICCES) (IEEE, 2020), 1068β1075.
- Rejeti, Kishore, G. Murali, and B. Suresh Kumar. βAn Accurate Methodology to Identify the Explosives Using Wireless Sensor Networks.β In Proceedings of International Conference on Sustainable Computing in Science, Technology and Management (SUSCOM), Amity University Rajasthan, Jaipur-India. 2019.
- V. K. K. Rejeti, S. Abdul, P. H. Prasanth, S. Ashok, and S. M. Sameer, βVirtual fit using computer vision and trimesh,β in 2023 Second International Conference on Electronics and Renewable Systems (ICEARS). IEEE, 2023, pp. 1590β1595.
- Kishore, N. G. Manthru, and Gudipati, βWireless nano senor Network (WNSN) for trace detection of explosives: The case of RDX and TNT,β Instrum. Mes. Metrol., vol. 18, no. 2, pp. 153 - 158, 2019, doi: 10.18280/i2m.180209.
- Krishnaiah, V. J. R., Prakash, V. S., Chandra, G. R., Sirisha, P. G. K., Mohan, K. J., Rejeti, V. K. K., & Sundari, P. N. (2024). Optimizing ZnO/CdS/CdTe bilayer structures for enhanced CdTe solar cell efficiency: A machine learning approach: VVJR Krishnaiah et al. MRS Advances, 9(9), 640-645.
- Sai M S, Rejeti V K K, Gera J and Raju M N 2021 Effective routing protocol in mobile ad-hoc network using individual node energy. Int. J. Adv. Res. Eng. Tech. 12: 445β453.
- Rejeti, Kishore, G. Murali, and B. Suresh Kumar. βAn Accurate Methodology to Identify the Explosives Using Wireless Sensor Networks.β In Proceedings of International Conference on Sustainable Computing in Science, Technology and Management (SUSCOM), Amity University Rajasthan, Jaipur-India. 2019.