Yeoh, Pauline Shan Qing and Lai, Khin Wee and Goh, Siew Li and Hasikin, Khairunnisa and Wu, Xiang and Li, Pei (2023) Transfer learning-assisted 3D deep learning models for knee osteoarthritis detection: Data from the osteoarthritis initiative. Frontiers in Bioengineering And Biotechnology, 11. ISSN 2296-4185, DOI https://doi.org/10.3389/fbioe.2023.1164655.
Full text not available from this repository.Abstract
Knee osteoarthritis is one of the most common musculoskeletal diseases and is usually diagnosed with medical imaging techniques. Conventionally, case identification using plain radiography is practiced. However, we acknowledge that knee osteoarthritis is a 3D complexity; hence, magnetic resonance imaging will be the ideal modality to reveal the hidden osteoarthritis features from a three-dimensional view. In this work, the feasibility of well-known convolutional neural network (CNN) structures (ResNet, DenseNet, VGG, and AlexNet) to distinguish knees with and without osteoarthritis (OA) is investigated. Using 3D convolutional layers, we demonstrated the potential of 3D convolutional neural networks of 13 different architectures in knee osteoarthritis diagnosis. We used transfer learning by transforming 2D pre-trained weights into 3D as initial weights for the training of the 3D models. The performance of the models was compared and evaluated based on the performance metrics balanced accuracy, precision, F1 score, and area under receiver operating characteristic (AUC) curve]. This study suggested that transfer learning indeed enhanced the performance of the models, especially for ResNet and DenseNet models. Transfer learning-based models presented promising results, with ResNet34 achieving the best overall accuracy of 0.875 and an F1 score of 0.871. The results also showed that shallow networks yielded better performance than deeper neural networks, demonstrated by ResNet18, DenseNet121, and VGG11 with AUC values of 0.945, 0.914, and 0.928, respectively. This encourages the application of clinical diagnostic aid for knee osteoarthritis using 3DCNN even in limited hardware conditions.
| Item Type: | Article |
|---|---|
| Funders: | UNSPECIFIED |
| Uncontrolled Keywords: | Convolutional neural network; Deep learning; Disease classification; Knee osteoarthritis; Magnetic resonance imaging |
| Subjects: | R Medicine T Technology > T Technology (General) T Technology > TA Engineering (General). Civil engineering (General) |
| Divisions: | Faculty of Engineering Faculty of Engineering > Biomedical Engineering Department Faculty of Medicine |
| Depositing User: | Ms Zaharah Ramly |
| Date Deposited: | 31 Jan 2024 01:37 |
| Last Modified: | 31 Jan 2024 01:37 |
| URI: | http://eprints.um.edu.my/id/eprint/38291 |
Actions (login required)
 |
View Item |
