SSP: self-supervised pertaining technique for classification of shoulder implants in x-ray medical images: a broad experimental study

被引:7
作者
Alzubaidi, Laith [1 ,2 ,3 ]
Fadhel, Mohammed A. [3 ]
Hollman, Freek [2 ,4 ]
Salhi, Asma [2 ,3 ]
Santamaria, Jose [5 ]
Duan, Ye [6 ]
Gupta, Ashish [2 ,3 ,4 ]
Cutbush, Kenneth [2 ,7 ]
Abbosh, Amin [8 ]
Gu, Yuantong [1 ,2 ]
机构
[1] Queensland Univ Technol, Sch Mech Med & Proc Engn, Brisbane, Qld, Australia
[2] Queensland Univ Technol, QUASR ARC Ind Transformat Training Ctr Joint Biome, Brisbane, Qld, Australia
[3] Akunah Co, Res & Dev Dept, Brisbane, Qld, Australia
[4] Greenslopes Private Hosp, Brisbane, Qld, Australia
[5] Univ Jaen, Dept Comp Sci, Jaen, Spain
[6] Clemson Univ, Sch Comp, Clemson, SC USA
[7] Univ Queensland, Sch Med, Brisbane, Qld, Australia
[8] Univ Queensland, Sch Informat Technol & Elect Engn, Brisbane, Qld, Australia
关键词
Deep learning; Medical imaging; Shoulder implants; Transfer learning; Self-supervised pertaining; NEURAL-NETWORK;
D O I
10.1007/s10462-024-10878-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Multiple pathologic conditions can lead to a diseased and symptomatic glenohumeral joint for which total shoulder arthroplasty (TSA) replacement may be indicated. The long-term survival of implants is limited. With the increasing incidence of joint replacement surgery, it can be anticipated that joint replacement revision surgery will become more common. It can be challenging at times to retrieve the manufacturer of the in situ implant. Therefore, certain systems facilitated by AI techniques such as deep learning (DL) can help correctly identify the implanted prosthesis. Correct identification of implants in revision surgery can help reduce perioperative complications and complications. DL was used in this study to categorise different implants based on X-ray images into four classes (as a first case study of the small dataset): Cofield, Depuy, Tornier, and Zimmer. Imbalanced and small public datasets for shoulder implants can lead to poor performance of DL model training. Most of the methods in the literature have adopted the idea of transfer learning (TL) from ImageNet models. This type of TL has been proven ineffective due to some concerns regarding the contrast between features learnt from natural images (ImageNet: colour images) and shoulder implants in X-ray images (greyscale images). To address that, a new TL approach (self-supervised pertaining (SSP)) is proposed to resolve the issue of small datasets. The SSP approach is based on training the DL models (ImageNet models) on a large number of unlabelled greyscale medical images in the domain to update the features. The models are then trained on a small labelled data set of X-ray images of shoulder implants. The SSP shows excellent results in five ImageNet models, including MobilNetV2, DarkNet19, Xception, InceptionResNetV2, and EfficientNet with precision of 96.69%, 95.45%, 98.76%, 98.35%, and 96.6%, respectively. Furthermore, it has been shown that different domains of TL (such as ImageNet) do not significantly affect the performance of shoulder implants in X-ray images. A lightweight model trained from scratch achieves 96.6% accuracy, which is similar to using standard ImageNet models. The features extracted by the DL models are used to train several ML classifiers that show outstanding performance by obtaining an accuracy of 99.20% with Xception+SVM. Finally, extended experimentation has been carried out to elucidate our approach's real effectiveness in dealing with different medical imaging scenarios. Specifically, five different datasets are trained and tested with and without the proposed SSP, including the shoulder X-ray with an accuracy of 99.47% and CT brain stroke with an accuracy of 98.60%.
引用
收藏
页数:45
相关论文
共 89 条
[51]   Deep neural network improves fracture detection by clinicians [J].
Lindsey, Robert ;
Daluiski, Aaron ;
Chopra, Sumit ;
Lachapelle, Alexander ;
Mozer, Michael ;
Sicular, Serge ;
Hanel, Douglas ;
Gardner, Michael ;
Gupta, Anurag ;
Hotchkiss, Robert ;
Potter, Hollis .
PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 2018, 115 (45) :11591-11596
[52]   Preoperative depression is a risk factor for complication and increased health care utilization following total shoulder arthroplasty [J].
Lunati, Matthew P. ;
Wilson, Jacob M. ;
Farley, Kevin X. ;
Gottschalk, Michael B. ;
Wagner, Eric R. .
JOURNAL OF SHOULDER AND ELBOW SURGERY, 2021, 30 (01) :89-96
[53]   From YouTube to the brain: Transfer learning can improve brain-imaging predictions with deep learning [J].
Malik, Nahiyan ;
Bzdok, Danilo .
NEURAL NETWORKS, 2022, 153 :325-338
[54]   International evaluation of an AI system for breast cancer screening [J].
McKinney, Scott Mayer ;
Sieniek, Marcin ;
Godbole, Varun ;
Godwin, Jonathan ;
Antropova, Natasha ;
Ashrafian, Hutan ;
Back, Trevor ;
Chesus, Mary ;
Corrado, Greg C. ;
Darzi, Ara ;
Etemadi, Mozziyar ;
Garcia-Vicente, Florencia ;
Gilbert, Fiona J. ;
Halling-Brown, Mark ;
Hassabis, Demis ;
Jansen, Sunny ;
Karthikesalingam, Alan ;
Kelly, Christopher J. ;
King, Dominic ;
Ledsam, Joseph R. ;
Melnick, David ;
Mostofi, Hormuz ;
Peng, Lily ;
Reicher, Joshua Jay ;
Romera-Paredes, Bernardino ;
Sidebottom, Richard ;
Suleyman, Mustafa ;
Tse, Daniel ;
Young, Kenneth C. ;
De Fauw, Jeffrey ;
Shetty, Shravya .
NATURE, 2020, 577 (7788) :89-+
[55]   A robust framework for shoulder implant X-ray image classification [J].
Minh Thanh Vo ;
Vo, Anh H. ;
Tuong Le .
DATA TECHNOLOGIES AND APPLICATIONS, 2022, 56 (03) :447-460
[56]   Classification of Retinal Diseases in Optical Coherence Tomography Images Using Artificial Intelligence and Firefly Algorithm [J].
Ozdas, Mehmet Batuhan ;
Uysal, Fatih ;
Hardalac, Firat .
DIAGNOSTICS, 2023, 13 (03)
[57]   Automated detection of COVID-19 cases using deep neural networks with X-ray images [J].
Ozturk, Tulin ;
Talo, Muhammed ;
Yildirim, Eylul Azra ;
Baloglu, Ulas Baran ;
Yildirim, Ozal ;
Acharya, U. Rajendra .
COMPUTERS IN BIOLOGY AND MEDICINE, 2020, 121
[58]   Deep Transfer Learning Based Classification Model for COVID-19 Disease [J].
Pathak, Y. ;
Tiwari, A. ;
Stalin, S. ;
Singh, S. ;
Shukla, P. K. .
IRBM, 2022, 43 (02) :87-92
[59]  
Raghu M, 2019, Arxiv, DOI arXiv:1902.07208
[60]  
Rajpurkar P, 2018, Arxiv, DOI arXiv:1712.06957