DOI: 10.3390/biomimetics11080541 ISSN: 2313-7673

Swarm Intelligence-Guided Hybrid Transfer Learning for Gastrointestinal Polyp Classification

Una Tuba, Mladen Veinovic, Eva Tuba, Adis Alihodzic, Milan Tuba

Colorectal cancer remains a leading cause of cancer-related mortality worldwide, with automated polyp classification from endoscopic images offering a promising avenue for improving early detection. Existing approaches rely on single convolutional neural network (CNN) backbones with manually designed classification heads, limiting both representational capacity and deployment flexibility. This paper presents a swarm intelligence-augmented multi-backbone deep learning framework for eight-class gastrointestinal lesion classification on the Kvasir benchmark. Four CNN backbones (ResNet50, DenseNet121, MobileNetV2, EfficientNetB3) are independently fine-tuned using a two-phase transfer learning protocol and their penultimate-layer features concatenated into a 5888-dimensional representation, reduced to 256 dimensions via PCA. Five swarm intelligence algorithms—Particle Swarm Optimization, Artificial Bee Colony, JADE, L-SHADE, and CMA-ES—are benchmarked on the classification head architecture search task; all independently converge to tanh activation, a consistent pattern across independently initialized algorithms that is suggestive of, though not conclusive evidence for, particular geometric properties of PCA-transformed deep feature spaces. The PSO-optimized single-layer head (284 units, tanh) outperforms a manually designed three-layer baseline by 0.75% while using 67% fewer parameters. SI-guided class weight optimization yields targeted F1 improvements on the two most clinically significant classes (polyps: +0.015, ulcerative-colitis: +0.013). The fixed-head classifier trained on fused four-backbone features achieves 91.08% accuracy on Kvasir v2 (multi-seed mean 91.47% ± 0.49 across nine converging seeds; one seed failed to converge and is disclosed rather than excluded), below end-to-end DenseNet121 (92.25%; Wilcoxon p = 0.31, not statistically significant), while enabling classifier updates in under 30 s; a three-backbone subset dropping the weakest backbone (EfficientNetB3) reaches 92.33%, exceeding the full four-backbone fusion. Cross-dataset evaluation on Kvasir v1-to-v2 confirms near-zero generalization gaps across dataset scales; a restricted two-class evaluation on HyperKvasir (the only two of eight classes with usable labeled data) reaches 96.28% accuracy, and dual Grad-CAM with SI minimal sufficient region analysis, validated quantitatively against Kvasir-SEG ground-truth masks, provides spatially grounded, clinically interpretable explanations.

More from our Archive