10th International Congress on Information and Communication Technology

Authors - Diego Loja, David Alvarado, Remigio Hurtado
Abstract - Lung cancer, one of the leading causes of death worldwide, accounts for more than 2.2 million cases and nearly 1.8 million deaths. This type of cancer is classified into non-small cell lung carcinoma (NSCLC), the most common and slow-progressing type, and small cell lung carcinoma (SCLC), which is less common but highly aggressive [1]. In response to the urgency for rapid and accurate diagnosis, this work presents an innovative method for classifying PET images using the EfficientV2S model, combined with advanced data augmentation and normalization techniques. Unlike traditional methods, this approach incorporates visual explanations based on integrated gradients, enabling the justification of model predictions. The proposed method consists of three phases: data preprocessing, experimentation, and prediction explanation. The LUNGPETCT- DX dataset is utilized, comprising 133 patients distributed across three main classes: adenocarcinoma, small cell carcinoma, and squamous cell carcinoma. The models are evaluated using quality metrics such as accuracy (78%), precision (82%), recall (78%), and F1-score (76%), highlighting the superior performance of EfficientV2S compared to other approaches. Additionally, integrated gradients are employed to visually justify predictions, providing critical interpretability in the medical context. For future work, the integration of CT images is proposed to enhance predictions, along with validation on larger datasets and optimization through fine-tuning, aiming to improve the model’s generalization and robustness