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:
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…
▽ More
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.
△ Less
Submitted 29 April, 2021;
originally announced April 2021.
