Scalable Semi-supervised Landmark Localization for X-ray Images using
Few-shot Deep Adaptive Graph
- URL: http://arxiv.org/abs/2104.14629v1
- Date: Thu, 29 Apr 2021 19:46:18 GMT
- Title: Scalable Semi-supervised Landmark Localization for X-ray Images using
Few-shot Deep Adaptive Graph
- Authors: Xiao-Yun Zhou, Bolin Lai, Weijian Li, Yirui Wang, Kang Zheng, Fakai
Wang, Chihung Lin, Le Lu, Lingyun Huang, Mei Han, Guotong Xie, Jing Xiao, Kuo
Chang-Fu, Adam Harrison, Shun Miao
- Abstract summary: Based on a fully-supervised graph-based method, DAG, we proposed a semi-supervised extension of it, termed few-shot DAG.
It first trains a DAG model on the labeled data and then fine-tunes the pre-trained model on the unlabeled data with a teacher-student SSL mechanism.
We extensively evaluated our method on pelvis, hand and chest landmark detection tasks.
- Score: 19.588348005574165
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Landmark localization plays an important role in medical image analysis.
Learning based methods, including CNN and GCN, have demonstrated the
state-of-the-art performance. However, most of these methods are
fully-supervised and heavily rely on manual labeling of a large training
dataset. In this paper, based on a fully-supervised graph-based method, DAG, we
proposed a semi-supervised extension of it, termed few-shot DAG, \ie five-shot
DAG. It first trains a DAG model on the labeled data and then fine-tunes the
pre-trained model on the unlabeled data with a teacher-student SSL mechanism.
In addition to the semi-supervised loss, we propose another loss using JS
divergence to regulate the consistency of the intermediate feature maps. We
extensively evaluated our method on pelvis, hand and chest landmark detection
tasks. Our experiment results demonstrate consistent and significant
improvements over previous methods.
Related papers
- GeoGuide: Geometric guidance of diffusion models [8.34616719984217]
GeoGuide is a guidance model based on tracing the distance of the diffusion model's trajectory from the data manifold.
It surpasses the probabilistic approach ADM-G with respect to both the FID scores and the quality of the generated images.
arXiv Detail & Related papers (2024-07-17T07:56:27Z) - GOODAT: Towards Test-time Graph Out-of-Distribution Detection [103.40396427724667]
Graph neural networks (GNNs) have found widespread application in modeling graph data across diverse domains.
Recent studies have explored graph OOD detection, often focusing on training a specific model or modifying the data on top of a well-trained GNN.
This paper introduces a data-centric, unsupervised, and plug-and-play solution that operates independently of training data and modifications of GNN architecture.
arXiv Detail & Related papers (2024-01-10T08:37:39Z) - From Cluster Assumption to Graph Convolution: Graph-based Semi-Supervised Learning Revisited [51.24526202984846]
Graph-based semi-supervised learning (GSSL) has long been a hot research topic.
graph convolutional networks (GCNs) have become the predominant techniques for their promising performance.
arXiv Detail & Related papers (2023-09-24T10:10:21Z) - On-the-Fly Guidance Training for Medical Image Registration [14.309599960641242]
This study introduces a novel On-the-Fly Guidance (OFG) training framework for enhancing existing learning-based image registration models.
Our method proposes a supervised fashion for training registration models, without the need for any labeled data.
Our method is tested across several benchmark datasets and leading models, it significantly enhanced performance.
arXiv Detail & Related papers (2023-08-29T11:12:53Z) - Graph Out-of-Distribution Generalization with Controllable Data
Augmentation [51.17476258673232]
Graph Neural Network (GNN) has demonstrated extraordinary performance in classifying graph properties.
Due to the selection bias of training and testing data, distribution deviation is widespread.
We propose OOD calibration to measure the distribution deviation of virtual samples.
arXiv Detail & Related papers (2023-08-16T13:10:27Z) - Cluster-level pseudo-labelling for source-free cross-domain facial
expression recognition [94.56304526014875]
We propose the first Source-Free Unsupervised Domain Adaptation (SFUDA) method for Facial Expression Recognition (FER)
Our method exploits self-supervised pretraining to learn good feature representations from the target data.
We validate the effectiveness of our method in four adaptation setups, proving that it consistently outperforms existing SFUDA methods when applied to FER.
arXiv Detail & Related papers (2022-10-11T08:24:50Z) - A semi-supervised Teacher-Student framework for surgical tool detection
and localization [2.41710192205034]
We introduce a semi-supervised learning (SSL) framework in surgical tool detection paradigm.
In the proposed work, we train a model with labeled data which initialises the Teacher-Student joint learning.
Our results on m2cai16-tool-locations dataset indicate the superiority of our approach on different supervised data settings.
arXiv Detail & Related papers (2022-08-21T17:21:31Z) - PCA: Semi-supervised Segmentation with Patch Confidence Adversarial
Training [52.895952593202054]
We propose a new semi-supervised adversarial method called Patch Confidence Adrial Training (PCA) for medical image segmentation.
PCA learns the pixel structure and context information in each patch to get enough gradient feedback, which aids the discriminator in convergent to an optimal state.
Our method outperforms the state-of-the-art semi-supervised methods, which demonstrates its effectiveness for medical image segmentation.
arXiv Detail & Related papers (2022-07-24T07:45:47Z) - Adapting the Mean Teacher for keypoint-based lung registration under
geometric domain shifts [75.51482952586773]
deep neural networks generally require plenty of labeled training data and are vulnerable to domain shifts between training and test data.
We present a novel approach to geometric domain adaptation for image registration, adapting a model from a labeled source to an unlabeled target domain.
Our method consistently improves on the baseline model by 50%/47% while even matching the accuracy of models trained on target data.
arXiv Detail & Related papers (2022-07-01T12:16:42Z) - A self-training framework for glaucoma grading in OCT B-scans [6.382852973055393]
We present a self-training-based framework for glaucoma grading using OCT B-scans under the presence of domain shift.
A two-step learning methodology resorts to pseudo-labels generated during the first step to augment the training dataset on the target domain.
We propose a novel glaucoma-specific backbone which introduces residual and attention modules via skip-connections to refine the embedding features of the latent space.
arXiv Detail & Related papers (2021-11-23T10:33:55Z) - Combining Deep Learning with Geometric Features for Image based
Localization in the Gastrointestinal Tract [8.510792628268824]
We propose a novel approach to combine Deep Learning method with traditional feature based approach to achieve better localization with small training data.
Our method fully exploits the best of both worlds by introducing a Siamese network structure to perform few-shot classification to the closest zone in the segmented training image set.
The accuracy is improved by 28.94% (Position) and 10.97% (Orientation) with respect to state-of-art method.
arXiv Detail & Related papers (2020-05-11T23:04:00Z)
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.