Journal of Bionic Engineering ›› 2024, Vol. 21 ›› Issue (2): 892-912.doi: 10.1007/s42235-023-00477-0

• • 上一篇    下一篇

Transfer Learning‑Based Class Imbalance‑Aware Shoulder Implant Classifcation from X‑Ray Images

Marut Jindal1; Birmohan Singh1   

  1. 1 Department of Computer Science and Engineering, Sant Longowal Institute of Engineering and Technology, Sangrur 148106, Punjab, India
  • 出版日期:2024-01-30 发布日期:2024-04-08
  • 通讯作者: Birmohan Singh E-mail:birmohans@gmail.com
  • 作者简介:Marut Jindal1; Birmohan Singh1

Transfer Learning‑Based Class Imbalance‑Aware Shoulder Implant Classifcation from X‑Ray Images

Marut Jindal1; Birmohan Singh1   

  1. 1 Department of Computer Science and Engineering, Sant Longowal Institute of Engineering and Technology, Sangrur 148106, Punjab, India
  • Online:2024-01-30 Published:2024-04-08
  • Contact: Birmohan Singh E-mail:birmohans@gmail.com
  • About author:Marut Jindal1; Birmohan Singh1

摘要: Total shoulder arthroplasty is a standard restorative procedure practiced by orthopedists to diagnose shoulder arthritis in which a prosthesis replaces the whole joint or a part of the joint. It is often challenging for doctors to identify the exact model and manufacturer of the prosthesis when it is unknown. This paper proposes a transfer learning-based class imbalance-aware prosthesis detection method to detect the implant’s manufacturer automatically from shoulder X-ray images. The framework of the method proposes a novel training approach and a new set of batch-normalization, dropout, and fully convolutional layers in the head network. It employs cyclical learning rates and weighting-based loss calculation mechanism. These modifcations aid in faster convergence, avoid local-minima stagnation, and remove the training bias caused by imbalanced dataset. The proposed method is evaluated using seven well-known pre-trained models of VGGNet, ResNet, and DenseNet families. Experimentation is performed on a shoulder implant benchmark dataset consisting of 597 shoulder X-ray images. The proposed method improves the classifcation performance of all pre-trained models by 10–12%. The DenseNet-201-based variant has achieved the highest classifcation accuracy of 89.5%, which is 10% higher than existing methods. Further, to validate and generalize the proposed method, the existing baseline dataset is supplemented to six classes, including samples of two more implant manufacturers. Experimental results have shown average accuracy of 86.7% for the extended dataset and show the preeminence of the proposed method.

关键词: Biomedical engineering , · Artifcial intelligence , · Total shoulder arthroplasty , · Prosthesis identifcation , · Biomedical image classifcation

Abstract: Total shoulder arthroplasty is a standard restorative procedure practiced by orthopedists to diagnose shoulder arthritis in which a prosthesis replaces the whole joint or a part of the joint. It is often challenging for doctors to identify the exact model and manufacturer of the prosthesis when it is unknown. This paper proposes a transfer learning-based class imbalance-aware prosthesis detection method to detect the implant’s manufacturer automatically from shoulder X-ray images. The framework of the method proposes a novel training approach and a new set of batch-normalization, dropout, and fully convolutional layers in the head network. It employs cyclical learning rates and weighting-based loss calculation mechanism. These modifcations aid in faster convergence, avoid local-minima stagnation, and remove the training bias caused by imbalanced dataset. The proposed method is evaluated using seven well-known pre-trained models of VGGNet, ResNet, and DenseNet families. Experimentation is performed on a shoulder implant benchmark dataset consisting of 597 shoulder X-ray images. The proposed method improves the classifcation performance of all pre-trained models by 10–12%. The DenseNet-201-based variant has achieved the highest classifcation accuracy of 89.5%, which is 10% higher than existing methods. Further, to validate and generalize the proposed method, the existing baseline dataset is supplemented to six classes, including samples of two more implant manufacturers. Experimental results have shown average accuracy of 86.7% for the extended dataset and show the preeminence of the proposed method.

Key words: Biomedical engineering , · Artifcial intelligence , · Total shoulder arthroplasty , · Prosthesis identifcation , · Biomedical image classifcation