A LIGHTWEIGHT CUSTOM CNN FOR ROBUST AND CLINICALLY SAFE CT KIDNEY DISEASE CLASSIFICATION
Article Sidebar
-
CT Kidney Classification, Deep Learning, CNN,, Tumor Detection
Abstract
ABSTRACT:
Kidney diseases such as renal cysts, stone, and malignant tumors represent a significant and increasing health burden worldwide. Manual CT diagnosis is labour-intensive and has high inter-observer variability, which requires the support of automatic diagnosis. In this paper, a lightweighted custom Convolutional Neural Network (CNN) is proposed, which is used to automatically classify kidney computed tomography (CT) images into four classes: Cyst, Normal, Stone, and Tumor. It includes residual skip connections, a dual-pooling head (Global Average and Global Max Pooling), Spatial Dropout2D regularization, and a composite training objective that explicitly favors Tumor Recall. The CT-KIDNEY-DATASET was mad leakage-safe through MD5 exact deduplication and perceptual-hash grouping, so that all reported evaluations are free of data leakage. Across the five validation folds, KidneyCNN achieved a mean accuracy of 99.82 ± 0.06% and a mean Tumor Recall of 99.31 ± 0.30%. Separately, ensemble inference on the independent held-out test set of 1,904 images achieved 99.84% accuracy, with AUC = 1.000 for all classes and only three misclassifications. The Grad-CAM++ results show that the model attention as anatomically relevant, which further indicates that KidneyCNN is a clinically safe and deployable kidney disease screening tool.
Full text article
References
Abid, M. H., Ashraf, R., Mahmood, T., & Faisal, C. M. N. (2023). Multi-modal medical image classification using deep residual network and genetic algorithm. PLoS ONE, 18(6 June). https://doi.org/10.1371/journal.pone.0287786
Aljohani, M. (2025). Clinical-Oriented Hierarchical Machine Learning Framework for Early Kidney Tumor Detection and Malignant Subtype Classification. Tomography, 11(11). https://doi.org/10.3390/tomography11110122
Alzu’Bi, D., Abdullah, M., Hmeidi, I., Alazab, R., Gharaibeh, M., El-Heis, M., Almotairi, K. H., Forestiero, A., Hussein, A. M., & Abualigah, L. (2022). Kidney Tumor Detection and Classification Based on Deep Learning Approaches: A New Dataset in CT Scans. Journal of Healthcare Engineering, 2022. https://doi.org/10.1155/2022/3861161
Bhandari, M., Yogarajah, P., Kavitha, M. S., & Condell, J. (2023). Exploring the Capabilities of a Lightweight CNN Model in Accurately Identifying Renal Abnormalities: Cysts, Stones, and Tumors, Using LIME and SHAP. Applied Sciences (Switzerland), 13(5). https://doi.org/10.3390/app13053125
Badawy, M., Almars, A. M., Magdy Balaha, H., Shehata, M., Qaraad, M., & Elhosseini, M. (2023). A two-stage renal disease classification based on transfer learning with hyperparameters optimization. Frontiers in Medicine, 10, 1106717. https://doi.org/10.3389/fmed.2023.1106717
He, K., Zhang, X., Ren, S., & Sun, J. (2015). Deep Residual Learning for Image Recognition. http://arxiv.org/abs/1512.03385
Hossain, M. N., Bhuiyan, E., Miah, M. B. A., Sifat, T. A., Muhammad, Z., & Masud, M. F. Al. (2025). Detection and Classification of Kidney Disease from CT Images: An Automated Deep Learning Approach. Technologies, 13(11). https://doi.org/10.3390/technologies13110508
Hossain, T., Sayed, F., & Islam, S. (2024). Adaptive Local Binary Pattern: A Novel Feature Descriptor for Enhanced Analysis of Kidney Abnormalities in CT Scan Images using ensemble based Machine Learning Approach. http://arxiv.org/abs/2404.14560
Islam, M. N., Hasan, M., Hossain, M. K., Alam, M. G. R., Uddin, M. Z., & Soylu, A. (2022). Vision transformer and explainable transfer learning models for auto detection of kidney cyst, stone and tumor from CT-radiography. Scientific Reports, 12(1). https://doi.org/10.1038/s41598-022-15634-4
Jahan, F., Reza, A. S., Morol, M. K., Nandi, D., Hossen, M. J., & Rahman, M. (2026). Explainable deep learning for early diagnosis of chronic kidney disease from CT images in Bangladeshi patients. Scientific Reports, 16(1). https://doi.org/10.1038/s41598-026-42654-1
Khan, Z. A., Waqar, M., Ullah Khan, H., Chaudhary, N. I., Tma Khan, A., Ishtiaq, I., Khan, A., Asif, M., & Raja, Z. (2025). Fine-tuned deep transfer learning: An effective strategy for the accurate chronic kidney disease classification. PeerJ Computer Science, 11, e2800. https://doi.org/10.7717/peerj-cs.2800
Maçin, G., Genç, F., Taşcı, B., Dogan, S., & Tuncer, T. (2025). KidneyNeXt: A Lightweight Convolutional Neural Network for Multi-Class Renal Tumor Classification in Computed Tomography Imaging. Journal of Clinical Medicine, 14(14). https://doi.org/10.3390/jcm14144929
Maqsood, F., Zhenfei, W., Ali, M. M., Qiu, B., Rehman, N. U., Sabah, F., Mahmood, T., Din, I., & Sarwar, R. (2024). Artificial Intelligence-Based Classification of CT Images Using a Hybrid SpinalZFNet. Interdisciplinary Sciences - Computational Life Sciences, 16(4), 907–925. https://doi.org/10.1007/s12539-024-00649-4
Pimpalkar, A., Saini, D. K. J. B., Shelke, N., Balodi, A., Rapate, G., & Tolani, M. (2025). Fine-tuned deep learning models for early detection and classification of kidney conditions in CT imaging. Scientific Reports, 15(1). https://doi.org/10.1038/s41598-025-94905-2
Rao, N. J., Girish, H., Gowrishankar, M. C., Kumar, S., & Kumar, N. (2025). A two-stage deep learning framework for kidney stone detection and clinical severity grading in CT imaging. Informatics in Medicine Unlocked, 59. https://doi.org/10.1016/j.imu.2025.101704
Saber, A., Hassan, E., Elbedwehy, S., Awad, W. A., & Emara, T. Z. (2025). Leveraging ensemble convolutional neural networks and metaheuristic strategies for advanced kidney disease screening and classification. Scientific Reports, 15(1). https://doi.org/10.1038/s41598-025-93950-1
Sharma, K., Uddin, Z., Wadal, A., & Gupta, D. (2025). Hybrid deep learning framework for classification of kidney CT images: Diagnosis of stones, cysts, and tumors. arXiv. https://doi.org/10.48550/arXiv.2502.04367
Sistaninejhad, B., Rasi, H., & Nayeri, P. (2023). A Review Paper about Deep Learning for Medical Image Analysis. In Computational and Mathematical Methods in Medicine (Vol. 2023). Hindawi Limited. https://doi.org/10.1155/2023/7091301
Tekin, H., Kılıç, Ş., & Doğan, Y. (2025). DiagNeXt: A Two-Stage Attention-Guided ConvNeXt Framework for Kidney Pathology Segmentation and Classification. Journal of Imaging, 11(12). https://doi.org/10.3390/jimaging11120433
Authors
Copyright (c) 2026 Kazheen Sileman Muhammad, and Araz Rajab Abrahim
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License [CC BY-NC-SA 4.0] that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work, with an acknowledgment of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online.