Temporal Output Discrepancy for Loss Estimation-Based Active Learning

被引:0
|
作者
Huang, Siyu [1 ]
Wang, Tianyang [2 ]
Xiong, Haoyi [3 ]
Wen, Bihan [4 ]
Huan, Jun [5 ]
Dou, Dejing [3 ]
机构
[1] Harvard Univ, Harvard A John Paulson Sch Engn & Appl Sci, Cambridge, MA 02134 USA
[2] Austin Peay State Univ, Dept Comp Sci & Informat Technol, Clarksville, TN 37044 USA
[3] Baidu Res, Big Data Lab, Beijing 100193, Peoples R China
[4] Nanyang Technol Univ, Sch Elect & Elect Engn, Singapore 639798, Singapore
[5] Amazon, AWS AI Lab, Seattle, WA 98109 USA
关键词
Active learning; loss estimation; model selection; semisupervised learning; temporal consistency regularization;
D O I
10.1109/TNNLS.2022.3186855
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
While deep learning succeeds in a wide range of tasks, it highly depends on the massive collection of annotated data which is expensive and time-consuming. To lower the cost of data annotation, active learning has been proposed to interactively query an oracle to annotate a small proportion of informative samples in an unlabeled dataset. Inspired by the fact that the samples with higher loss are usually more informative to the model than the samples with lower loss, in this article we present a novel deep active learning approach that queries the oracle for data annotation when the unlabeled sample is believed to incorporate high loss. The core of our approach is a measurement temporal output discrepancy (TOD) that estimates the sample loss by evaluating the discrepancy of outputs given by models at different optimization steps. Our theoretical investigation shows that TOD lower-bounds the accumulated sample loss thus it can be used to select informative unlabeled samples. On basis of TOD, we further develop an effective unlabeled data sampling strategy as well as an unsupervised learning criterion for active learning. Due to the simplicity of TOD, our methods are efficient, flexible, and task-agnostic. Extensive experimental results demonstrate that our approach achieves superior performances than the state-of-the-art active learning methods on image classification and semantic segmentation tasks. In addition, we show that TOD can be utilized to select the best model of potentially the highest testing accuracy from a pool of candidate models.
引用
收藏
页码:2109 / 2123
页数:15
相关论文
共 50 条
  • [1] MCDAL: Maximum Classifier Discrepancy for Active Learning
    Cho, Jae Won
    Kim, Dong-Jin
    Jung, Yunjae
    Kweon, In So
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (11) : 8753 - 8763
  • [2] Inconsistency Based Active Learning for temporal object detection
    Li, Xinjie
    Zhang, Lijun
    Zhou, Fangyu
    ACM SYMPOSIUM ON SPATIAL USER INTERACTION, SUI 2023, 2023,
  • [3] Active Learning of Pattern Classification Based on PEDCC-Loss
    Zhu, Qiuyu
    Luan, Jianbing
    Li, Tiantian
    Zu, Xuewen
    IEEE ACCESS, 2021, 9 : 147626 - 147633
  • [4] Nuclear discrepancy for single-shot batch active learning
    Tom J. Viering
    Jesse H. Krijthe
    Marco Loog
    Machine Learning, 2019, 108 : 1561 - 1599
  • [5] Nuclear discrepancy for single-shot batch active learning
    Viering, Tom J.
    Krijthe, Jesse H.
    Loog, Marco
    MACHINE LEARNING, 2019, 108 (8-9) : 1561 - 1599
  • [6] Discrepancy-Based Active Learning for Weakly Supervised Bleeding Segmentation in Wireless Capsule Endoscopy Images
    Bai, Fan
    Xing, Xiaohan
    Shen, Yutian
    Ma, Han
    Meng, Max Q. -H.
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT VIII, 2022, 13438 : 24 - 34
  • [7] Active label distribution learning via kernel maximum mean discrepancy
    Dong, Xinyue
    Luo, Tingjin
    Fan, Ruidong
    Zhuge, Wenzhang
    Hou, Chenping
    FRONTIERS OF COMPUTER SCIENCE, 2023, 17 (04)
  • [8] Unsupervised active learning with loss prediction
    Chuanbing Wan
    Fusheng Jin
    Zhuang Qiao
    Weiwei Zhang
    Ye Yuan
    Neural Computing and Applications, 2023, 35 : 3587 - 3595
  • [9] Active label distribution learning via kernel maximum mean discrepancy
    DONG Xinyue
    LUO Tingjin
    FAN Ruidong
    ZHUGE Wenzhang
    HOU Chenping
    Frontiers of Computer Science, 2023, 17 (04)
  • [10] EdgeAL: An Edge Estimation Based Active Learning Approach for OCT Segmentation
    Kadir, Md Abdul
    Alam, Hasan Md Tusfiqur
    Sonntag, Daniel
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT II, 2023, 14221 : 79 - 89