A Self-ensembling Framework for Semi-supervised Knee Cartilage Defects
Assessment with Dual-Consistency
- URL: http://arxiv.org/abs/2005.09212v2
- Date: Mon, 12 Oct 2020 11:08:17 GMT
- Title: A Self-ensembling Framework for Semi-supervised Knee Cartilage Defects
Assessment with Dual-Consistency
- Authors: Jiayu Huo, Liping Si, Xi Ouyang, Kai Xuan, Weiwu Yao, Zhong Xue, Qian
Wang, Dinggang Shen, Lichi Zhang
- Abstract summary: We propose a novel approach for knee cartilage defects assessment, including severity classification and lesion localization.
A self-ensembling framework is composed of a student network and a teacher network with the same structure.
Experiments show that the proposed method can significantly improve the self-ensembling performance in both knee cartilage defects classification and localization.
- Score: 40.67137486295487
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Knee osteoarthritis (OA) is one of the most common musculoskeletal disorders
and requires early-stage diagnosis. Nowadays, the deep convolutional neural
networks have achieved greatly in the computer-aided diagnosis field. However,
the construction of the deep learning models usually requires great amounts of
annotated data, which is generally high-cost. In this paper, we propose a novel
approach for knee cartilage defects assessment, including severity
classification and lesion localization. This can be treated as a subtask of
knee OA diagnosis. Particularly, we design a self-ensembling framework, which
is composed of a student network and a teacher network with the same structure.
The student network learns from both labeled data and unlabeled data and the
teacher network averages the student model weights through the training course.
A novel attention loss function is developed to obtain accurate attention
masks. With dual-consistency checking of the attention in the lesion
classification and localization, the two networks can gradually optimize the
attention distribution and improve the performance of each other, whereas the
training relies on partially labeled data only and follows the semi-supervised
manner. Experiments show that the proposed method can significantly improve the
self-ensembling performance in both knee cartilage defects classification and
localization, and also greatly reduce the needs of annotated data.
Related papers
- An AI System for Continuous Knee Osteoarthritis Severity Grading Using Self-Supervised Anomaly Detection with Limited Data [0.30723404270319693]
This work proposes a three stage approach for automated continuous grading of knee OA.
It learns a robust representation of healthy knee X-rays and grading disease severity based on its distance to the centre of normality.
The proposed methodology outperforms existing techniques by margins of up to 24% in terms of OA detection and the disease severity scores correlate with the Kellgren-Lawrence grading system at the same level as human expert performance.
arXiv Detail & Related papers (2024-07-16T08:37:33Z) - Class Attention to Regions of Lesion for Imbalanced Medical Image
Recognition [59.28732531600606]
We propose a framework named textbfClass textbfAttention to textbfREgions of the lesion (CARE) to handle data imbalance issues.
The CARE framework needs bounding boxes to represent the lesion regions of rare diseases.
Results show that the CARE variants with automated bounding box generation are comparable to the original CARE framework.
arXiv Detail & Related papers (2023-07-19T15:19:02Z) - Automatic diagnosis of knee osteoarthritis severity using Swin
transformer [55.01037422579516]
Knee osteoarthritis (KOA) is a widespread condition that can cause chronic pain and stiffness in the knee joint.
We propose an automated approach that employs the Swin Transformer to predict the severity of KOA.
arXiv Detail & Related papers (2023-07-10T09:49:30Z) - An End-to-End Framework For Universal Lesion Detection With Missing
Annotations [24.902835211573628]
We present a novel end-to-end framework for mining unlabeled lesions while simultaneously training the detector.
Our framework follows the teacher-student paradigm. High-confidence predictions are combined with partially-labeled ground truth for training the student model.
arXiv Detail & Related papers (2023-03-27T09:16:10Z) - Joint localization and classification of breast tumors on ultrasound
images using a novel auxiliary attention-based framework [7.6620616780444974]
We propose a novel joint localization and classification model based on the attention mechanism and disentangled semi-supervised learning strategy.
The proposed modularized framework allows flexible network replacement to be generalized for various applications.
arXiv Detail & Related papers (2022-10-11T20:14:13Z) - A Two-stream Convolutional Network for Musculoskeletal and Neurological
Disorders Prediction [14.003588854239544]
Musculoskeletal and neurological disorders are the most common causes of walking problems among older people.
Recent deep learning-based methods have shown promising results for automated analysis.
arXiv Detail & Related papers (2022-08-18T14:32:16Z) - Coherence Learning using Keypoint-based Pooling Network for Accurately
Assessing Radiographic Knee Osteoarthritis [18.47511520060851]
Knee osteoarthritis (OA) is a common degenerate joint disorder that affects a large population of elderly people worldwide.
Current clinically-adopted knee OA grading systems are observer subjective and suffer from inter-rater disagreements.
We propose a computer-aided diagnosis approach to provide more accurate and consistent assessments of both composite and fine-grained OA grades simultaneously.
arXiv Detail & Related papers (2021-12-16T19:59:13Z) - Cross-Site Severity Assessment of COVID-19 from CT Images via Domain
Adaptation [64.59521853145368]
Early and accurate severity assessment of Coronavirus disease 2019 (COVID-19) based on computed tomography (CT) images offers a great help to the estimation of intensive care unit event.
To augment the labeled data and improve the generalization ability of the classification model, it is necessary to aggregate data from multiple sites.
This task faces several challenges including class imbalance between mild and severe infections, domain distribution discrepancy between sites, and presence of heterogeneous features.
arXiv Detail & Related papers (2021-09-08T07:56:51Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - A Teacher-Student Framework for Semi-supervised Medical Image
Segmentation From Mixed Supervision [62.4773770041279]
We develop a semi-supervised learning framework based on a teacher-student fashion for organ and lesion segmentation.
We show our model is robust to the quality of bounding box and achieves comparable performance compared with full-supervised learning methods.
arXiv Detail & Related papers (2020-10-23T07:58:20Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.