Aug, 2020

基于不确定性估计的抓握检测网络用于基于置信度驱动的半监督域适应

TL;DR该论文提出了一种以置信度驱动的半监督学习网络,结合了位于物体抓取领域的域适应技术,使用特征金字塔网络 (FPN) 提供一种预测不确定性估计机制,并利用该信息加强关注置信度较高的未标记数据的一致性损失,避免了学生模型从一致性损失中学习到错误 / 有害信息,从而提高了学习进度和模型准确性。经过实验,该方法在 Cornell grasping 数据集上取得了高的成功率,并在使用有限数据进行领域适应时比原始平均教师 (mean teacher) 和直接训练的评估损失性能提高了不少于 10% 的性能,特别是在避免过拟合和模型发散时。