Genetic Algorithm Optimized Vision Transformer Based Transfer Learning Approach for Early Prediction of Chronic Kidney Disease


Authors : Arindam Roy; Avishek Gupta; Ram Prasad Chakraborty

Volume/Issue : Volume 10 - 2025, Issue 11 - November


Google Scholar : https://tinyurl.com/4v2vf26d

Scribd : https://tinyurl.com/ywaf4kmn

DOI : https://doi.org/10.38124/ijisrt/25nov076

Note : A published paper may take 4-5 working days from the publication date to appear in PlumX Metrics, Semantic Scholar, and ResearchGate.

Note : Google Scholar may take 30 to 40 days to display the article.


Abstract : Chronic Kidney Disease (CKD) is an ongoing health issue defined by the slow decline in kidney performance over a prolonged period. This condition can escalate to kidney failure, which is life-threatening without the absence of dialysis or a kidney transplant. The common key factors leading to CKD include diabetes, hypertension, Glomerulonephritis. Standard diagnostic practices tend to be laborious, require considerable resources, and are vulnerable to errors made by humans. With the ongoing advancements in artificial intelligence within the healthcare domain, machine learning and deep learning algorithms are playing a pivotal role in the accurate and effective detection of CKD. The purpose of the proposed research is to construct and authenticate a predictive model aimed at diagnosing chronic kidney disease. Image processing-based diagnostic approaches have shown a greater success rate compared to other detection methods. To tackle these challenges, our investigation proposes a new methodology that merges Genetic Algorithms (GA) with a Vision Transformer (ViT) model that employs Hierarchical Attention in a transfer learning framework, thereby improving both feature selection and classification accuracy. In this scholarly work, we have scrutinized the performance of two leading architectures, VGG16 and ResNet50, and have proposed an attention-centric approach that employs the Vision Transformer model optimized by evolutionary algorithm. The attention mechanism has the potential to grasp long-term dependencies in images. Extraction of complex features is to be done using ViT Model, preceded by several hyper parameters like number of epochs, learning rate, batch size, number of layers, layer size, number of attention heads, and attention window size should be optimized through a genetic algorithm to enhance performance and also feature selection is done by GA to get the optimized result. A dataset available to the public consists of four types of image data of kidneys: Cyst, Normal, Tumor, and Stone, which are used in the three architectures discussed above. In total, there are 12,446 images that are segmented for training, testing, and validation purposes. Experimental results demonstrate that our GA optimized ViT Model surpasses state-of-the-art traditional models, achieving a 98.05% F1-score, and the model also shows superiority in terms of trainable parameters

Keywords : Chronic Kidney Disease, Fast ViT (Vision Transformer), Genetic Algorithm (GA), Transfer Learning, ResNet50, VGG16.

References :

  1. A. K. Bello, I. G. Okpechi, A. Levin, F. Ye, S. Damster, S. Arruebo, J.-A. Donner, F. J. Caskey, Y. Cho, M. R. Davids et al., “An update on the global disparities in kidney disease burden and care across world countries and regions,” The Lancet Global Health, vol. 12, no. 3, pp.e382–e395, 2024.
  2.  A. C. Webster, E. V. Nagler, R. L. Morton, and P. Masson, “Chronickidney disease,” The lancet, vol. 389, no. 10075, pp. 1238–1252, 2017.
  3. N. Bhaskar and M. Suchetha, “Analysis of salivary components as non-invasive biomarkers for monitoring chronic kidney disease,” International Journal of Medical Engineering and Informatics, vol. 12, no. 2, pp. 95–107, 2020, doi: 10.1504/IJMEI.2020.106896.
  4. K. Suzuki, Overview of deep learning in medical imaging, Radiol. Phys. Technol. 10 (3) (2017) 257–273.
  5. J.-G. Lee, S. Jun, Y.-W. Cho, H. Lee, G.B. Kim, J.B. Seo, N. Kim, Deep learning in medical imaging: general overview, Korean J. Radiol. 18 (4) (2017) 570–584.
  6. G. Litjens, T. Kooi, B.E. Bejnordi, A.A.A. Setio, F. Ciompi, M. Ghafoorian, J.A. Van Der Laak, B. Van Ginneken, C.I. Sánchez, A survey on deep learning in medical image analysis, Med. Image Anal. 42 (2017) 60–88.
  7. G. Wang, W. Li, M.A. Zuluaga, R. Pratt, P.A. Patel, M. Aertsen, T. Doel, A.L. David, J. Deprest, S. Ourselin, et al., Interactive medical image segmentation using deep learning with image-specific fine tuning, IEEE Trans. Med. Imaging 37 (7) (2018) 1562–1573.
  8. Maha, Gharaibeh., Dalia, Alzu'bi., Malak, Abdullah., Ismail, Hmeidi., Mohammad, Rustom, Al, Nasar., Laith, Abualigah., Amir, H., Gandomi. (2022). Radiology Imaging Scans for Early Diagnosis of Kidney Tumors: A Review of Data Analytics-Based Machine Learning and Deep Learning Approaches. Big data and cognitive computing, doi: 10.3390/bdcc6010029
  9.  Agarwal, Anush., Gaikar, Rohini., Schieda, Nicola., Elfaal, Mohamed, WalaaEldin., Ukwatta, Eranga. (2023). Deep-learning-based ensemble method for fully automated detection of renal masses on magnetic resonance images. Journal of medical imaging, doi:10.1117/1.JMI.10.2.024501
  10. Khan, A., Das, R., & Parameshwara, M. C. (2022). “Detection of kidney stones using digital image processing: A holistic approach. Engineering Research Express,” 4(3), 035040. https://doi.org/10.1088/2634- Fire
  11. M. Yang, X. He, L. Xu, et al., “Ct-based transformer model for non-invasively predicting the fuhrman nuclear grade of clear cell renal cell carcinoma,” Frontiers in Oncology, vol. 12, p. 961 779, 2022.
  12. Sri, V. S., & Lakshmi, G. J. (2023, April). Detection Analysis of Abnormality in Kidney using Deep Learning Techniques and its Optimization. In 2023 Second International Conference on Electrical, Electronics, Information and Communication Technologies (ICEEICT) (pp.1-6). IEEE.
  13. Mehmet, Çifci., Sadiq, Hussain., Peren, Jerfi, Canatalay.(2023). Hybrid Deep Learning Approach for Accurate Tumor Detection in Medical Imaging Data. Diagnostics, doi: 10.3390/diagnostics13061025
  14. K. Nagawa et al., 'Three-dimensional convolutional neural network-based classification of chronic kidney disease severity using kidney MRI,' Scientific Reports, 2024.
  15. Bittencourt, J. A. S., Sousa, C. M., Santana, E. E. C., Moraes, Y. A. C. D., Carneiro, E. C. R. D. L., Fontes, A. J. C., ... & Nascimento, M. D. D. S. B. (2024). Prediction of metabolic syndrome and its associated risk factors in patients with chronic kidney disease using machine learning techniques. Brazilian Journal of Nephrology, 46(4), e20230135.
  16. Satukumati, S. B., & Bhat, M. N. (2024). Early detection of chronic kidney disease using data mining methods. International Journal of Advanced Intelligence Paradigms, 28(1-2), 86-99.
  17. L. Maria Michael Visuwasam, “NMA: Integrating Big Data Into A Novel Mobile Application Using Knowledge Extraction for Big Data Analytics Cluster Computing: The Journal Of Networks, Software Tools And Applications,22 (1),2018.
  18. Ghosh, B. P., Imam, T., Anjum, N., Mia, M. T., Siddiqua, C. U., Sharif, K. S., ... & Hossain, M. Z. (2024). Advancing chronic kidney disease prediction: Comparative analysis of machine learning algorithms and a hybrid model. Journal of Computer Science and Technology Studies, 6(3), 15-21.
  19. Singh, V., Asari, V. K., & Rajasekaran, R. (2022). A deep neural network for early detection and prediction of chronic kidney disease. Diagnostics, 12(1), 116.
  20. https://www.kaggle.com/datasets/nazmul0087/ct-kidney-dataset-normal-cyst-tumor-and-stone

Chronic Kidney Disease (CKD) is an ongoing health issue defined by the slow decline in kidney performance over a prolonged period. This condition can escalate to kidney failure, which is life-threatening without the absence of dialysis or a kidney transplant. The common key factors leading to CKD include diabetes, hypertension, Glomerulonephritis. Standard diagnostic practices tend to be laborious, require considerable resources, and are vulnerable to errors made by humans. With the ongoing advancements in artificial intelligence within the healthcare domain, machine learning and deep learning algorithms are playing a pivotal role in the accurate and effective detection of CKD. The purpose of the proposed research is to construct and authenticate a predictive model aimed at diagnosing chronic kidney disease. Image processing-based diagnostic approaches have shown a greater success rate compared to other detection methods. To tackle these challenges, our investigation proposes a new methodology that merges Genetic Algorithms (GA) with a Vision Transformer (ViT) model that employs Hierarchical Attention in a transfer learning framework, thereby improving both feature selection and classification accuracy. In this scholarly work, we have scrutinized the performance of two leading architectures, VGG16 and ResNet50, and have proposed an attention-centric approach that employs the Vision Transformer model optimized by evolutionary algorithm. The attention mechanism has the potential to grasp long-term dependencies in images. Extraction of complex features is to be done using ViT Model, preceded by several hyper parameters like number of epochs, learning rate, batch size, number of layers, layer size, number of attention heads, and attention window size should be optimized through a genetic algorithm to enhance performance and also feature selection is done by GA to get the optimized result. A dataset available to the public consists of four types of image data of kidneys: Cyst, Normal, Tumor, and Stone, which are used in the three architectures discussed above. In total, there are 12,446 images that are segmented for training, testing, and validation purposes. Experimental results demonstrate that our GA optimized ViT Model surpasses state-of-the-art traditional models, achieving a 98.05% F1-score, and the model also shows superiority in terms of trainable parameters

Keywords : Chronic Kidney Disease, Fast ViT (Vision Transformer), Genetic Algorithm (GA), Transfer Learning, ResNet50, VGG16.

CALL FOR PAPERS


Paper Submission Last Date
30 - November - 2025

Video Explanation for Published paper

Never miss an update from Papermashup

Get notified about the latest tutorials and downloads.

Subscribe by Email

Get alerts directly into your inbox after each post and stay updated.
Subscribe
OR

Subscribe by RSS

Add our RSS to your feedreader to get regular updates from us.
Subscribe