An End-to-End Concatenated CNN Attention Model for the Classification of Lung Cancer With XAI Techniques
Fariha Haque, Mohammad Asif Hasan, Md. Abu Ismail Siddique, Tonmoy Roy, Tonmoy Kanti Shaha, Yamina Islam, Avijit Paul, Muhammad E. H. Chowdhury
Abstract
In the field of medical imaging, deep learning (DL) techniques have made significant contributions to the detection and classification of various cancers. Identifying the precise regions in medical images containing cancerous cells plays a crucial role in the diagnostic process. Early and accurate cancer detection is essential for effective treatment and improved patient outcomes. However, manual diagnosis is labor-intensive, requiring the specialized expertise of radiologists, and the increasing number of cancer cases presents challenges in processing large volumes of image data efficiently. To address these challenges, an end-to-end concatenated Convolutional Neural Network (CNN) attention model has been proposed for automatic lung cancer classification. This approach integrates two distinct CNNs, followed by a multi-layer perceptron (MLP) and a multi-head attention (MHA) mechanism, to enhance performance. The model leverages explainable AI techniques, such as gradient-weighted class activation mapping (grad-CAM) and Shapley additive explanations (SHAP), to highlight critical regions within the images that influence the decision-making process. This model achieves impressive performance, with an accuracy of 99.54%, precision of 99.31%, recall of 99.95%, F1-score of 99.66%, and an AUC of 99.97%. These results demonstrate that Thisapproach not only surpasses existing methods but also provides a highly accurate and interpretable solution. By reducing the need for extensive manual intervention, this model enables faster and more reliable lung cancer diagnosis, paving the way for timely and effective treatments.