Deep Learning Model Ensemble for the Accuracy of Classification Degenerative Arthritis

被引:2
作者
Lee, Sang-min [1 ]
Kim, Namgi [1 ]
机构
[1] Kyonggi Univ, Dept Comp Sci, Suwon, South Korea
来源
CMC-COMPUTERS MATERIALS & CONTINUA | 2023年 / 75卷 / 01期
关键词
Knee osteoarthritis; deep learning; convolutional neural network; Kellgren-Lawrence grade; classification; knee X-ray; NETWORK;
D O I
10.32604/cmc.2023.035245
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Artificial intelligence technologies are being studied to provide scientific evidence in the medical field and developed for use as diagnos-tic tools. This study focused on deep learning models to classify degen-erative arthritis into Kellgren-Lawrence grades. Specifically, degenerative arthritis was assessed by X-ray radiographic images and classified into five classes. Subsequently, the use of various deep learning models was investigated for automating the degenerative arthritis classification process. Although research on the classification of osteoarthritis using deep learning has been conducted in previous studies, only local models have been used, and an ensemble of deep learning models has never been applied to obtain more accurate results. To address this issue, this study compared the classification performance of deep learning models, including VGGNet, DenseNet, ResNet, TinyNet, EfficientNet, MobileNet, Xception, and ViT, on a dataset com-monly used for osteoarthritis classification tasks. Our experimental results verified that even without applying a separate methodology, the performance of the ensemble was comparable to that of existing studies that only used the latest deep learning model and changed the learning method. From the trained models, two ensembles were created and evaluated: weight and specialist. The weight ensemble showed an improvement in accuracy of 1%, and the proposed specialist ensemble improved accuracy, precision, recall, and F1 score by 5%, 6%, 6%, and 6%, respectively, compared with the results of prior studies.
引用
收藏
页码:1981 / 1994
页数:14
相关论文
共 24 条
[1]  
Antony Joseph, 2017, Machine Learning and Data Mining in Pattern Recognition. 13th International Conference, MLDM 2017. Proceedings: LNAI 10358, P376, DOI 10.1007/978-3-319-62416-7_27
[2]   A Parallel Deep Convolutional Neural Network for Alzheimer's disease classification on PET/CT brain images [J].
Baydargil, Husnu Baris ;
Park, Jangsik ;
Kang, Do-Young ;
Kang, Hyun ;
Cho, Kook .
KSII TRANSACTIONS ON INTERNET AND INFORMATION SYSTEMS, 2020, 14 (09) :3583-3597
[3]   A Lightweight CNN and Joint Shape-Joint Space (JS']JS2) Descriptor for Radiological Osteoarthritis Detection [J].
Bayramoglu, Neslihan ;
Nieminen, Miika T. ;
Saarakkala, Simo .
MEDICAL IMAGE UNDERSTANDING AND ANALYSIS, 2020, 1248 :331-345
[4]  
Chen Pingjun, 2018, Mendeley Data, V1, DOI 10.17632/56RMX5BJCR.1
[5]   Fully automatic knee osteoarthritis severity grading using deep neural networks with a novel ordinal loss [J].
Chen, Pingjun ;
Gao, Linlin ;
Shi, Xiaoshuang ;
Allen, Kyle ;
Yang, Lin .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2019, 75 :84-92
[6]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[7]  
Foran JRH., 2020, Total Knee Replacement - AAOS
[8]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861
[9]  
Han K, 2020, Arxiv, DOI arXiv:2010.14819
[10]  
He KM, 2015, Arxiv, DOI [arXiv:1512.03385, DOI 10.48550/ARXIV.1512.03385]