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 LUNG-PETCT-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

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

PET Image Classification for Lung Cancer Diagnosis: Deep Learning with Transfer Learning, Data Augmentation and Region-Based Prediction Explanation by Integrated Gradients

  • Diego Loja,
  • David Alvarado,
  • Remigio Hurtado

摘要

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 LUNG-PETCT-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