Grasping Detection Network with Uncertainty Estimation for Confidence-Driven Semi-Supervised Domain Adaptation

08/20/2020
by   Haiyue Zhu, et al.
18

Data-efficient domain adaptation with only a few labelled data is desired for many robotic applications, e.g., in grasping detection, the inference skill learned from a grasping dataset is not universal enough to directly apply on various other daily/industrial applications. This paper presents an approach enabling the easy domain adaptation through a novel grasping detection network with confidence-driven semi-supervised learning, where these two components deeply interact with each other. The proposed grasping detection network specially provides a prediction uncertainty estimation mechanism by leveraging on Feature Pyramid Network (FPN), and the mean-teacher semi-supervised learning utilizes such uncertainty information to emphasizing the consistency loss only for those unlabelled data with high confidence, which we referred it as the confidence-driven mean teacher. This approach largely prevents the student model to learn the incorrect/harmful information from the consistency loss, which speeds up the learning progress and improves the model accuracy. Our results show that the proposed network can achieve high success rate on the Cornell grasping dataset, and for domain adaptation with very limited data, the confidence-driven mean teacher outperforms the original mean teacher and direct training by more than 10 overfitting and model diverging.

READ FULL TEXT
POST COMMENT

Comments

There are no comments yet.

Authors

page 1

page 2

page 3

page 4

page 5

page 6

07/13/2020

Dual-Teacher: Integrating Intra-domain and Inter-domain Teachers for Annotation-efficient Cardiac Segmentation

Medical image annotations are prohibitively time-consuming and expensive...
09/29/2021

Uncertainty-aware Mean Teacher for Source-free Unsupervised Domain Adaptive 3D Object Detection

Pseudo-label based self training approaches are a popular method for sou...
01/17/2019

Certainty-Driven Consistency Loss for Semi-supervised Learning

The recently proposed semi-supervised learning methods exploit consisten...
06/16/2017

Self-ensembling for visual domain adaptation

This paper explores the use of self-ensembling for visual domain adaptat...
07/12/2018

Learning-based Regularization for Cardiac Strain Analysis with Ability for Domain Adaptation

Reliable motion estimation and strain analysis using 3D+time echocardiog...
04/25/2019

Exploring Object Relation in Mean Teacher for Cross-Domain Detection

Rendering synthetic data (e.g., 3D CAD-rendered images) to generate anno...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.