A morphology-guided ViT-CNN approach for the detection of lung and colon cancer using H&E-stained biopsy images.

Abstract

The accurate categorisation of histopathological images is crucial for the reliable identification of morphologically similar cancers, including lung and colon cancer. According to the WHO and IARC, lung cancer accounted for approximately 2.48 million new cases and 1.8 million deaths worldwide in 2022, making it the leading cause of cancer-related mortality worldwide. Colorectal cancer also represents a major global health burden, with more than 1.9 million new cases and approximately 930,000 deaths reported in 2020 alone. These rising burdens highlight the need for robust computational pathology systems capable of accurate morphological discrimination. In this paper, we propose MorphoViT-CNN, a hybrid model that combines a Vision Transformer for capturing global contextual information with Convolutional Neural Networks (CNNs) for extracting fine grained local features. A morphology-aware segmentation module is designed to incorporate cellular shape, size, and texture attributes, while self-distillation improves the consistency of internal representations without requiring external supervision. Supervised contrastive learning further enhances inter-class separability, particularly in label-limited settings. The model was evaluated on the LC25000 dataset using five-fold cross validation, and ablation studies were conducted to assess the contribution of each component. MorphoViT-CNN demonstrated superior accuracy and precision-recall performance compared with leading CNN and transformer-based baselines, while significantly reducing misclassification among challenging histological subtypes. Its interpretable and multi-scale architecture provides a scalable framework for integration into computational pathology workflows. Future work will explore multimodal integration and foundation model pre-training to further enhance its applicability to precision oncology.