Hybrid Swarm-Driven Vision Transformer (HSViT) for Lung Cancer Segmentation and Classification from CT Scans

Kavithamani V; V. Kavya; R. Suganthi; Yuvaraj S.; P. Monisha; Arun Patrick

doi:10.35882/jeeemi.v8i1.1384

Kavithamani V Department of Electronics and Communication Engineering, Jai Shriram Engineering College, Tamilnadu, India. https://orcid.org/0000-0003-4410-2060
V. Kavya Department of Electronics and Communication Engineering, Jai Shriram Engineering College, Tamilnadu, India. https://orcid.org/0009-0005-9498-9298
R. Suganthi Department of Electronics and Communication Engineering, Panimalar Engineering College, Tamil Nadu, India https://orcid.org/0000-0002-7045-5321
Yuvaraj S. Department of Computer Science and Engineering at Sri Eshwar College of Engineering, Coimbatore, India https://orcid.org/0000-0001-5002-7285
P. Monisha Department of ECE, Karunya Institute of Technology and Sciences,Coimbatore, Tamilnadu, India https://orcid.org/0000-0003-0058-5988
Arun Patrick Department of Computer Science and Engineering, Nehru Institute of Technology, Jawahar Gardens, Tamil Nadu, India https://orcid.org/0009-0009-0575-7663

DOI: https://doi.org/10.35882/jeeemi.v8i1.1384

Keywords: Hybrid Swarm Driven Vision Transformer, Coyote Optimization Algorithm, Vision Transformer, Dual Stage Attention Fusion, Lung Cancer Segmentation

Abstract

Lung cancer segmentation and classification from computed tomography (CT) images play a vital role in early diagnosis, prognosis assessment, and effective treatment planning. Despite significant progress in medical image analysis, accurate lung lesion analysis remains highly challenging due to overlapping anatomical structures, heterogeneous tissue intensity distributions, irregular and complex tumor shapes, and poorly defined lesion boundaries. These factors often limit the reliability and generalization capability of conventional deep learning models when applied to real-world clinical data. To address these challenges, this paper proposes a Hybrid Swarm-Driven Vision Transformer (HSViT) framework that synergistically combines swarm intelligence with transformer-based deep learning. The processing pipeline begins with Contrast Limited Adaptive Histogram Equalization (CLAHE), which enhances local contrast while suppressing noise amplification, thereby improving the visibility of subtle pulmonary nodules and lesion regions. Subsequently, a U-Net segmentation model optimized using the Coyote Optimization Algorithm (COA) is employed to accurately delineate lung lesions. COA, a swarm-based metaheuristic, adaptively fine-tunes U-Net parameters, enabling improved convergence and more precise boundary detection compared to gradient-based optimization alone. Following segmentation, discriminative lesion features are extracted and passed to the HSViT classifier. The proposed classifier integrates a Dual-Stage Attention Fusion (DSAF) mechanism, which effectively captures both fine-grained local spatial features and long-range global contextual dependencies. The framework achieves a Dice Coefficient of 0.95, an overall classification accuracy of 98.7%, and a minimized training loss of 0.04. These results highlight the strong potential of HSViT for reliable automated lung cancer diagnosis and for supporting clinical decision-making systems in real-world healthcare environments.

Downloads

Download data is not yet available.

References

Ahmed, B. T. (2019). Lung cancer prediction and detection using image processing mechanisms: an overview. Signal and Image Processing Letters, 1(3), 20-31, DOI:10.31763/simple.v1i3.11.

Wu, J. T. Y., Wakelee, H. A., & Han, S. S. (2023). Optimizing lung cancer screening with risk prediction: current challenges and the emerging role of biomarkers. Journal of Clinical Oncology, 41(27), 4341-4347, DOI: 10.1200/JCO.23.01060

Balasamy, K., & Suganyadevi, S. “Multi-dimensional fuzzy based diabetic retinopathy detection in retinal images through deep CNN method”. Multimedia Tools and Applications, Vol 83, no. 5, pp.1–23. 2024, doi: 10.1007/s11042-024-19798-1

Huang, D., Li, Z., Jiang, T., Yang, C., & Li, N. (2024). Artificial intelligence in lung cancer: current applications, future perspectives, and challenges. Frontiers in Oncology, 14, 1486310, doi: 10.3389/fonc.2024.1486310

Sakoda, L. C., Henderson, L. M., Caverly, T. J., Wernli, K. J., & Katki, H. A. (2017). Applying risk prediction models to optimize lung cancer screening: current knowledge, challenges, and future directions. Current epidemiology reports, 4(4), 307-320, DOI: 10.1007/s40471-017-0126-8

SHARIFF, V., Chiranjeevi, P., & Krishna, M. A. (2023). An analysis on advances in lung cancer diagnosis with medical imaging and deep learning techniques: Challenges and opportunities. Journal of Theoretical and Applied Information Technology, 101(17), 7083-7095.

Dataset collection: https://www.kaggle.com/code/ahmednagdiii/lung-cancer-classification-with-pre-trained-cnn/input

Rawashdeh, M., Obaidat, M., Abouali, M., Salhi, D., & Thakur, K. (2025). An effective lung Cancer diagnosis model using Pre-Trained CNNs. Computer Modeling in Engineering & Sciences, 143(1), 1129, https://doi.org/10.32604/cmes.2025.063765

Saha, N., Mondal, R., Banerjee, A., Debnath, R., & Chatterjee, S. (2025). Advanced Deep Lung Care Net: A Next Generation Framework for Lung Cancer Prediction. International Journal of Innovative Science and Research Technology, 10(6), 2312-2320, https://doi.org/10.38124/ijisrt/25jun1801

Shatnawi, M. Q., Abuein, Q., & Al-Quraan, R. (2025). Deep learning-based approach to diagnose lung cancer using CT-scan images. Intelligence-Based Medicine, 11, 100188, DOI:10.1016/j.ibmed.2024.100188

Bhattacharyya, S., Khattar, S., & Goel, P. (2025, May). Investigation of Deep Learning based Lung Cancer Detection using Histopathological Images. In 2025 IEEE International Conference on Computer, Electronics, Electrical Engineering & their Applications (IC2E3) (pp. 1-6). IEEE, DOI:10.1109/IC2E365635.2025.11167720

Jiang, Y., Ebrahimpour, L., Després, P., & Manem, V. S. (2025). A benchmark of deep learning approaches to predict lung cancer risk using national lung screening trial cohort. Scientific reports, 15(1), 1736, https://doi.org/10.1038/s41598-024-84193-7

Tusher, M. I., Hasan, M. M., Akter, S., Haider, M., Chy, M. S. K., Akhi, S. S., ... & Shaima, M. (2025). Deep learning meets early diagnosis: A hybrid CNN-DNN framework for lung cancer prediction and clinical translation. International Journal of Medical Science and Public Health Research, 6(05), 63-72, https://doi.org/10.37547/ijmsphr/Volume06Issue05-04

Lakide, V., & Ganesan, V. (2025). Precise Lung Cancer Prediction using ResNet–50 Deep Neural Network Architecture. Journal of Electronics, Electromedical Engineering, and Medical Informatics, 7(1), 38-46, https://doi.org/10.35882/jeeemi.v7i1.518

Priya, A., & Bharathi, P. S. (2025). SE-ResNeXt-50-CNN: A deep learning model for lung cancer classification. Applied Soft Computing, 171, 112696, DOI:10.1016/j.asoc.2025.112696

Ansari, M. M., Kumar, S., Chola, C., Heyat, M. B. B., Akhtar, F., Hayat, M. A. B., ... & Pomary, D. (2025). A Novel Machine and Deep Learning–Based Ensemble Techniques for Automatic Lung Cancer Detection. BioMed Research International, 2025(1), 6666688, https://doi.org/10.1155/bmri/6666688

Kumar, V., Prabha, C., Sharma, P., Mittal, N., Askar, S. S., & Abouhawwash, M. (2024). Unified deep learning models for enhanced lung cancer prediction with ResNet-50–101 and EfficientNet-B3 using DICOM images. BMC medical imaging, 24(1), 63, https://doi.org/10.1186/s12880-024-01241-4

Suganyadevi, S., & Seethalakshmi, V. “CVD-HNet: Classifying Pneumonia and COVID-19 in Chest X-ray Images Using Deep Network”. Wireless Personal Communications, Vol.126, no. 4, pp.3279–3303, 2022, doi: 10.1007/s11277-022-09864-y

Shamia, D., Balasamy, K., and Suganyadevi, S. “A secure framework for medical image by integrating watermarking and encryption through fuzzy based roi selection”, Journal of Intelligent & Fuzzy systems, 2023, Vol. 44, no.5, pp.7449-7457, doi: 10.3233/JIFS-222618.

Balasamy, K., Seethalakshmi, V. & Suganyadevi, S. Medical Image Analysis Through Deep Learning Techniques: A Comprehensive Survey. Wireless Pers Commun 137, 1685–1714 (2024). https://doi.org/10.1007/s11277-024-11428-1.

Suganyadevi, S., Seethalakshmi, V. Deep recurrent learning based qualified sequence segment analytical model (QS2AM) for infectious disease detection using CT images. Evolving Systems 15, 505–521 (2024). https://doi.org/10.1007/s12530-023-09554-5.

Balasamy, K., Seethalakshmi, V. & Suganyadevi, S. “Medical image analysis through deep learning techniques: a comprehensive survey”, Wireless Pers Commun, Vol.137, pp.1685–1714, 2024, doi:10.1007/s11277-024-11428-1

Suganyadevi S, Pershiya AS, Balasamy K, Seethalakshmi V, Bala S, Arora K (2024) Deep learning based Alzheimer disease diagnosis: a comprehensive review. SN Comput Sci 5(4):391, DOI:10.1007/s42979-024-02743-2

M.F. Mridha, et al., A comprehensive survey on the progress, process, and challenges of lung cancer detection and classification, J. Healthc. Eng. 2022 (2022) 5905230, https://doi.org/10.1155/2022/5905230.

S.R. Rezaei, A. Ahmadi, A hierarchical GAN method with ensemble CNN for accurate nodule detection, Int. J. Comput. Assist. Radiol. Surg. 18 (4) (2023) 695–705, https://doi.org/10.1007/s11548-022-02807-9.

M.S. Bhuiyan, et al., Advancements in early detection of lung cancer in public health: a comprehensive study utilizing machine learning algorithms and predictive models, J. Comput. Sci. Technol. Stud. 6 (1) (2024) 113–121, https://

doi.org/10.32996/jcsts.2024.6.1.12.

H.T. Gayap, M.A. Akhloufi, Deep machine learning for medical diagnosis, application to lung cancer detection: a review, BioMedInformatics 4 (1) (2024) 236–284, https://doi.org/10.3390/biomedinformatics4010015.

S. Zou, S. Wei, H. Liu, An integrated cell region reconstruction method based upon mask R-CNN model and improved voronoi algorithm, J. Phys. Conf. Ser. 1453 (1) (2020) 12034, https://doi.org/10.1088/1742-6596/1453/1/012034.

K. C P.K. Balasubramanian, W.C. Lai, G.H. Seng, J. Selvaraj, APESTNet with mask R-CNN for liver tumor segmentation and classification, Cancers. (Basel) 15 (2) (2023) 330, https://doi.org/10.3390/cancers15020330.

Kaur, G., Prabha, C., Chhabra, D., Kaur, N., Veeramanickam, M. R. M., & Gill, S. K. (2022). A systematic approach to machine learning for cancer classification. 2022 5th International Conference on Contemporary Computing and Informatics (IC3I). IEEE, DOI:10.1109/IC3I56241.2022.10072474.