Enhancing image-based diagnosis of gastrointestinal tract diseases through deep learning with EfficientNet and advanced data augmentation techniques

Abstract

Abstract The early detection and diagnosis of gastrointestinal tract diseases, such as ulcerative colitis, polyps, and esophagitis, are crucial for timely treatment. Traditional imaging techniques often rely on manual interpretation, which is subject to variability and may lack precision. Current methodologies leverage conventional deep learning models that, while effective to an extent, often suffer from overfitting and generalization issues on medical image datasets due to the intricate and subtle variations in disease manifestations. These models typically do not fully utilize the potential of transfer learning or advanced data augmentation, leading to less-than-optimal performance, especially in diverse real-world scenarios where data variability is high. This study introduces a robust model using the EfficientNetB5 architecture combined with a sophisticated data augmentation strategy. The model is tailored for the high variability and intricate details present in gastrointestinal tract disease images. By integrating transfer learning with maximal pooling and extensive regularization, the model aims to enhance diagnostic accuracy and reduce overfitting. The proposed model achieved a test accuracy of 98.89%, surpassing traditional methods by incorporating advanced regularization and augmentation techniques. The application of horizontal flipping and dynamic scaling during training significantly improved the model’s ability to generalize, evidenced by a low-test loss of 0.230 and high precision metrics across all classes. The proposed deep learning framework demonstrates superior performance in the automated classification of gastrointestinal diseases from image data. By addressing key limitations of existing models through innovative techniques, this study contributes to the enhancement of diagnostic processes in medical imaging, potentially leading to more accurate and timely disease interventions.

Keywords

• Deep learning
• Gastrointestinal diseases
• EfficientNet
• Image augmentation
• Medical imaging
• Transfer learning