On Non-Random Missing Labels in Semi-Supervised Learning
On Non-Random Missing Labels in Semi-Supervised Learning
复制标题
关于半监督学习中的非随机缺失标签
DOI:
--
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
Hanwang Zhang
中科院分区:
文献类型:
--
作者:
Xinting Hu;Yulei Niu;C. Miao;Xiansheng Hua;Hanwang Zhang
Semi-Supervised Learning (SSL) is fundamentally a missing label problem, in which the label Missing Not At Random (MNAR) problem is more realistic and challenging, compared to the widely-adopted yet naive Missing Completely At Random assumption where both labeled and unlabeled data share the same class distribution. Different from existing SSL solutions that overlook the role of"class"in causing the non-randomness, e.g., users are more likely to label popular classes, we explicitly incorporate"class"into SSL. Our method is three-fold: 1) We propose Class-Aware Propensity (CAP) that exploits the unlabeled data to train an improved classifier using the biased labeled data. 2) To encourage rare class training, whose model is low-recall but high-precision that discards too many pseudo-labeled data, we propose Class-Aware Imputation (CAI) that dynamically decreases (or increases) the pseudo-label assignment threshold for rare (or frequent) classes. 3) Overall, we integrate CAP and CAI into a Class-Aware Doubly Robust (CADR) estimator for training an unbiased SSL model. Under various MNAR settings and ablations, our method not only significantly outperforms existing baselines but also surpasses other label bias removal SSL methods. Please check our code at: https://github.com/JoyHuYY1412/CADR-FixMatch.
影响因子:
5
作者:
Funk, Michele Jonsson;Westreich, Daniel;Davidian, Marie
通讯作者:
Davidian, Marie