Comparative Evaluation of Single CNN Models, Test-Time Augmentation, and Weighted Ensembling for Knee Osteoarthritis Severity Classification
By: Huseyin Coskun and Zekeriya Katılmış
References
- Y. Badshah et al., “Genetic markers of osteoarthritis: Early diagnosis in susceptible Pakistani population,” J. Orthop. Surg. Res., vol. 16, Feb. 2021, Art. no. 124. https://doi.org/10.1186/s13018-021-02230-x
- S. K. Das and A. Farooqi, “Osteoarthritis,” Best Pract. Res. Clin. Rheumatol., vol. 22, no. 4, pp. 657–675, Aug. 2008. https://doi.org/10.1016/j.berh.2008.07.002
- M. N. Iqbal, F. R. Haidri, B. Motiani, and A. Mannan, “Frequency of factors associated with knee osteoarthritis,” J. Pak. Med. Assoc., vol. 61, no. 8, pp. 786–789, 2011.
- K. Maiese, “Picking a bone with WISP1 (CCN4): New strategies against degenerative joint disease,” J. Transl. Sci., vol. 2, no. 1, pp. 83–85, 2016. https://doi.org/10.15761/JTS.1000120
- S. S. Gornale, P. U. Patravali, and R. R. Manza, “Detection of osteoarthritis using knee X-ray image analyses: A machine vision based approach,” Int. J. Comput. Appl., vol. 145, no. 1, pp. 20–26, July 2016. https://doi.org/10.5120/ijca2016910544
- T. Neogi, “The epidemiology and impact of pain in osteoarthritis,” Osteoarthritis Cartilage, vol. 21, no. 9, pp. 1145–1153, Sep. 2013. https://doi.org/10.1016/j.joca.2013.03.018
- J. H. Kellgren and J. S. Lawrence, “Radiological assessment of osteoarthrosis,” Ann. Rheum. Dis., vol. 16, no. 4, pp. 494–502, Dec. 1957. https://doi.org/10.1136/ard.16.4.494
- D. Schiphof, M. Boers, and S. M. A. Bierma-Zeinstra, “Differences in descriptions of Kellgren and Lawrence grades of knee osteoarthritis,” Ann. Rheum. Dis., vol. 67, no. 7, pp. 1034–1036, July 2008. https://doi.org/10.1136/ard.2007.079020
- D. Saini, T. Chand, D. K. Chouhan, and M. Prakash, “A comparative analysis of automatic classification and grading methods for knee osteoarthritis focussing on X-ray images,” Biocybern. Biomed. Eng., vol. 41, no. 2, pp. 419–444, Apr.–June 2021. https://doi.org/10.1016/j.bbe.2021.03.002
- S. Roy, T. Meena, and S.-J. Lim, “Demystifying supervised learning in healthcare 4.0: A new reality of transforming diagnostic medicine,” Diagnostics, vol. 12, no. 10, Oct. 2022, Art. no. 2549. https://doi.org/10.3390/diagnostics12102549
- P. Chen. (2018). Knee Osteoarthritis Severity Grading Dataset. [Online}. Available: https://data.mendeley.com/datasets/56rmx5bjcr/1
- S. M. Ahmed and R. J. Mstafa, “Identifying severity grading of knee osteoarthritis from X-ray images using an efficient mixture of deep learning and machine learning models,” Diagnostics, vol. 12, no. 12, Nov. 2022, Art. no. 2939. https://doi.org/10.3390/diagnostics12122939
- A. S. Mohammed, A. A. Hasanaath, G. Latif, and A. Bashar, “Knee osteoarthritis detection and severity classification using residual neural networks on preprocessed X-ray images,” Diagnostics, vol. 13, no. 8, Apr. 2023, Art. no. 1380. https://doi.org/10.3390/diagnostics13081380
- T. Momenpour and A. A. Mallouh, “Optimizing CNN-based diagnosis of knee osteoarthritis: Enhancing model accuracy with CleanLab relabeling,” Diagnostics, vol. 15, no. 11, pp. 1332–1349, May 2025. https://doi.org/10.3390/diagnostics15111332
- K. Zuiderveld, “Contrast limited adaptive histogram equalization,” in Graphics Gems, P. S. Heckbert, Ed., Academic Press, 1994, pp. 474–485. https://doi.org/10.1016/B978-0-12-336156-1.50061-6
- C. Shorten and T. M. Khoshgoftaar, “A survey on image data augmentation for deep learning,” J. Big Data, vol. 6, no. 60, July 2019, Art. no. 60. https://doi.org/10.1186/s40537-019-0197-0
- T. He, Z. Zhang, H. Zhang, Z. Zhang, J. Xie, and M. Li, “Bag of tricks for image classification with convolutional neural networks,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, June 2019, pp. 558–567. https://doi.org/10.1109/CVPR.2019.00065
- J. M. Johnson and T. M. Khoshgoftaar, “Survey on deep learning with class imbalance,” J. Big Data, vol. 6, Mar. 2019, Art. no. 27. https://doi.org/10.1186/s40537-019-0192-5
- M. Tan and Q. V Le, “EfficientNet: Rethinking model scaling for convolutional neural networks,” in Proceedings of the 36th International Conference on Machine Learning, 2019, pp. 6105–6114.
- Z. Liu, H. Mao, C.-Y. Wu, C. Feichtenhofer, T. Darrell, and S. Xie, “A ConvNet for the 2020s,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA, June 2022, pp. 11976–11986. https://doi.org/10.1109/CVPR52688.2022.01167
- I. Loshchilov and F. Hutter, “Decoupled weight decay regularization,” in 7th International Conference on Learning Representations, ICLR 2019, New Orleans: International Conference on Learning Representations, ICLR, May 2017. Accessed: Oct. 04, 2026. [Online]. Available: https://arxiv.org/abs/1711.05101v3
- I. Loshchilov and F. Hutter, “SGDR: Stochastic Gradient Descent with Warm Restarts,” in 5th International Conference on Learning Representations, ICLR 2017 - Conference Track Proceedings, International Conference on Learning Representations, ICLR, Aug. 2016. Accessed: Oct. 04, 2026. [Online]. Available: https://arxiv.org/abs/1608.03983v5
- T.-Y. Lin, P. Goyal, R. Girshick, K. He, and P. Dollár, “Focal loss for dense object detection,” in Proceedings of the IEEE international conference on computer vision, Venice, Italy, Oct. 2017, pp. 2980–2988. https://doi.org/10.1109/ICCV.2017.324
- T. G. Dietterich, “Ensemble methods in machine learning,” in Multiple Classifier Systems, vol. 1857, J. Kittler and F. Roli, Eds., in Lecture Notes in Computer Science, vol. 1857. Springer, 2000, pp. 1–15. https://doi.org/10.1007/3-540-45014-9_1
- Z.-H. Zhou, Ensemble Methods: Foundations and Algorithms. CRC Press, June 2012. https://doi.org/10.1201/b12207
- I. Kandel and M. Castelli, “Improving convolutional neural networks performance for image classification using test time augmentation: A case study using MURA dataset,” J. Med. Syst., vol. 45, no. 7, July2021, Art. no. 84. https://doi.org/10.1007/s10916-021-01757-0
- R. R. Selvaraju, M. Cogswell, A. Das, R. Vedantam, D. Parikh, and D. Batra, “Grad-cam: Visual explanations from deep networks via gradient-based localization,” in Proceedings of the IEEE international conference on computer vision, Venice, Italy, Oct. 2017, pp. 618–626. https://doi.org/10.1109/ICCV.2017.74
DOI: https://doi.org/10.2478/acss-2026-0016 | Journal eISSN: 2255-8691 | Journal ISSN: 2255-8683 (formerly 2255-8691)
Language: English
Page range: 187 - 205
Submitted on: Aug 20, 2026
Accepted on: Oct 12, 2026
Published online: Oct 28, 2026
Published in print: Jan 1, 2026
Published by: Riga Technical University
In partnership with: Paradigm Publishing Services
Publication frequency: Volume open
Keywords:
Related subjects:
© 2026 Huseyin Coskun, Zekeriya Katılmış, published by Riga Technical University
This work is licensed under the Creative Commons Attribution 4.0 License.