Automated Esophageal Tumor Detection Using Deep Learning

Abstract

Esophageal cancer remains one of the most lethal malignancies worldwide, where early detection is essential for improving survival outcomes. Traditional diagnostic methods such as endoscopy and histopathology are time-consuming, resource-intensive, and subject to human variability. This study presents a deep learning-based end-to-end diagnostic system for esophageal cancer detection using image classification. The proposed model integrates a hybrid architecture combining Swin Transformer and ResNet-50, capturing both global contextual information and fine-grained local features to enhance classification accuracy. Due to the absence of pixel-level annotated segmentation masks, a Grad-CAM-based visualization technique is employed to localize cancer-affected regions, providing interpretability and visual support for clinical decisions. A confidence-based grading module is included to estimate cancer severity levels—Low, Medium, or High—using model prediction probabilities, thereby compensating for the lack of explicitly labeled grading data. The model is trained and optimized under low-memory constraints, ensuring efficient deployment in real-world environments, including low-resource clinical settings. It is saved in a portable PyTorch .pth format, enabling consistent inference across platforms. Additionally, a web interface built with Flask allows users to upload endoscopic images and receive real-time predictions, visual heatmaps, and grading feedback. Experimental results on a dataset of cancerous and non-cancerous esophageal images demonstrate high classification accuracy and reliable visual explanations, validating the system's effectiveness. This work highlights the potential of artificial intelligence in advancing diagnostic tools for esophageal cancer and offers a practical solution for resource-limited healthcare settings.

Country : India

1 Er. Ashish P. Mohod2 Pratham Anil Bangre3 Sahil Ramdas Borkar4 Jayant Dhanraj Bajirao5 Harshal Harichandra Bhanarkar6 Himanshu Dadarao Ramteke

  1. Artificial Intelligence, Priyadarshini J. L. College of Engineering, Nagpur, Maharashtra, India
  2. Artificial Intelligence, Priyadarshini J. L. College of Engineering, Nagpur, Maharashtra, India
  3. Artificial Intelligence, Priyadarshini J. L. College of Engineering, Nagpur, Maharashtra, India
  4. Artificial Intelligence, Priyadarshini J. L. College of Engineering, Nagpur, Maharashtra, India
  5. Artificial Intelligence, Priyadarshini J. L. College of Engineering, Nagpur, Maharashtra, India
  6. Artificial Intelligence, Priyadarshini J. L. College of Engineering, Nagpur, Maharashtra, India

IRJIET, Volume 9, Issue 5, May 2025 pp. 35-41

doi.org/10.47001/IRJIET/2025.905004

References

  1. Y. Zhu et al., “Automated esophageal cancer detection and classification using convolutional neural networks,” Scientific Reports, vol. 10, 2020.
  2. K. Zhang et al., “Diagnosis of esophageal cancer using deep learning models trained with endoscopic images,” Gastroenterology Journal, vol. 12, 2020.
  3. Y. Yang et al., “AI-enhanced classification of esophageal cancer using transfer learning and endoscopic images,” Artificial Intelligence in Medicine, vol. 115, 2021.
  4. J. Wang et al., “Improving early esophageal cancer diagnosis with machine learning-based endoscopy,” Computer Methods and Programs in Biomedicine, vol. 198, 2021.
  5. H. Sun et al., “Endoscopic image classification for esophageal cancer using a convolutional neural network with transfer learning,” Journal of Digestive Diseases, vol. 23, no. 2, 2022.
  6. Y. Liu et al., “Barium esophagram-based deep learning system for detecting esophageal cancer,” Frontiers in Oncology, vol. 13, 2023.
  7. T. Yamada et al., “Histopathological image analysis of esophageal cancer with deep learning-enhanced segmentation,” Biomedical Signal Processing and Control, vol. 85, 2023.
  8. X. Huang et al., “Transfer learning in deep learning models for esophageal cancer classification on endoscopic images,” Journal of Biomedical Informatics, vol. 138, 2023.
  9. P. Gupta et al., “Multi-modal deep learning approach for esophageal cancer staging and classification,” Computers in Biology and Medicine, vol. 155, 2023.
  10. R. Tan et al., “Data augmentation in endoscopic imaging: Enhancing esophageal cancer classification in AI models,” Artificial Intelligence in Healthcare, vol. 7, 2024.
  11. A.Shakil, “Endoscopy Classification: Eso and Non-Eso,” Kaggle, 2023. [Online]. Available: https://www.kaggle.com/code/azhanshakil/endoscopyclassification-eso-and-non-eso/notebook