Swarm Intelligence-Guided Hybrid Transfer Learning for Gastrointestinal Polyp Classification
Swarm Intelligence-Guided Hybrid Transfer Learning for Gastrointestinal Polyp Classification
Autori:
Časopis: Biomimetics
Volume, no: 11 , 8
ISSN: 2313-7673
DOI: 10.3390/biomimetics11080541
Stranice: 541-541
Apstrakt:
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.
Ključne reči: gastrointestinal polyp classification; swarm intelligence; metaheuristic optimization; multi-backbone feature fusion; transfer learning; explainable AI; Grad-CAM; convolutional neural networks; Kvasir; colorectal cancer
Kategorije objave:
Bibliografske reference nastavnika Univerziteta Singidunum
Zahvaljujemo se što ste preuzeli publikaciju sa portala Singipedia.
Ukoliko želite da se prijavite za obaveštenja o sadržajima iz oblasti ove publikacije, možete nam ostaviti adresu svoje elektronske pošte.