Compact deep learning models for colon histopathology focusing performance and generalization challenges

Read full text
Authors: Fareeha Hanif, Ali Raza, Heba Abdelgader Mohammed
Cited by: 5
References: 84

Abstract

Colorectal cancer is a leading cause of cancer-related mortality, and accurate analysis of histopathological images is critical for early diagnosis and improved patient outcomes. This study proposes and systematically evaluates four purpose-built lightweight convolutional neural network (CNN) variants (Lite-V0, Lite-V1, Lite-V2, and Lite-V4) for binary classification of colon histopathology images into Colon_Adenocarcinoma and Colon_Benign_Tissue. Experiments were conducted on a balanced dataset (24,000 images) with fixed train/validation/test splits and comprehensive evaluation using accuracy and macro-F1, supported by confusion matrices and ROC/precision–recall analyses. Among all variants, Lite-V2 achieved the best validation performance (macro-F1 ≈ 0.999), while remaining highly compact (1.53 MB; 127,682 parameters), indicating a favorable accuracy–efficiency trade-off for deployment-oriented diagnostic support. On the independent test set, however, Lite-V2 exhibited a marked generalization drop, achieving approximately 50% accuracy and macro-F1 = 0.33, suggesting a domain-shift effect between validation and test samples. These findings demonstrate that lightweight CNNs can achieve near-perfect internal validation performance for colon histopathology classification, but robust cross-domain generalization remains essential; future work will focus on domain adaptation and stain-robust training strategies to improve reliability on unseen clinical data.

Fields of Study & Subjects

AI in cancer detection Colorectal Cancer Screening and Detection Advanced Neural Network Applications

Publication details

Journal / source
Scientific Reports
Publisher
Springer Science and Business Media LLC
Volume
16
Issue
1
Pages
5489–5489
Publication date
19 January 2026
Language
EN