Self-Ensembling Semi-Supervised Model for Bone X-ray Images Landmark Detection.

Tian Bai,Shenyao Liu,Yuzhao Wang,Yu Wang,Dong Dong
DOI: https://doi.org/10.1109/BIBM52615.2021.9669420
2021-01-01
Abstract:Bone image landmark detection plays a vital role in orthopedic diseases diagnosis and analysis. As a promising approach, the data-driven convolutional neural network has been widely applied in landmarks detection. However, the model’s performance relies heavily on numerous high-quality labeled data, which are difficult to access. A practical solution for this issue in bone images landmark detection has not been provided, so we propose a framework of semi-supervised bone images landmark detection encouraged by the recent success of the self-ensembling method. The student and teacher networks share the same structure in our framework, where unlabeled data are utilized by enforcing the prediction consistency of two networks under different perturbations to enhance the network’s performance. The two networks use a mutual learning approach to improve the efficiency of using unlabeled data. In addition, we also propose a transformative perturbation scheme to prevent model overfitting during training and enhance the generalization capability of the model on unseen data. Experimental results on an in-house hip dataset collected from the actual process of clinical diagnosis and a publicly available dataset demonstrate that our framework has excellent performance compared with other semi-supervised methods.
What problem does this paper attempt to address?