Compact deep learning models for colon histopathology focusing performance and generalization challenges
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 $$\approx$$ 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.
Authors
- Fareeha Hanif
- Heba Abdelgader Mohammed
- Ali Raza
Institutions
- University of the Punjab (PK)
- University of Education (PK)
- King Khalid University (SA)
Publication Details
- Journal
- Scientific Reports
- Published
- 2026-01-19
- DOI
- https://doi.org/10.1038/s41598-026-35119-y
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- King Khalid University