Instance-Dependent Positive and Unlabeled Learning With Labeling Bias Estimation
Chen Gong,Qizhou Wang,Tongliang Liu,Bo Han,Jane You,Jian Yang,Dacheng Tao
DOI: https://doi.org/10.1109/TPAMI.2021.3061456
IF: 23.6
2022-01-01
IEEE Transactions on Pattern Analysis and Machine Intelligence
Abstract:This paper studies instance-dependent Positive and Unlabeled (PU) classification, where whether a positive example will be labeled (indicated by s) is not only related to the class label y, but also depends on the observation x. Therefore, the labeling probability on positive examples is not uniform as previous works assumed, but is biased to some simple or critical data points. To depict the above dependency relationship, a graphical model is built in this paper which further leads to a maximization problem on the induced likelihood function regarding P(s, y vertical bar x). By utilizing the well-known EM and Adam optimization techniques, the labeling probability of any positive example P(s = 1 vertical bar y = 1, x) as well as the classifier induced by P(y vertical bar x) can be acquired. Theoretically, we prove that the critical solution always exists, and is locally unique for linear model if some sufficient conditions are met. Moreover, we upper bound the generalization error for both linear logistic and non-linear network instantiations of our algorithm, with the convergence rate of expected risk to empirical risk as O(1/root k+1/root n-k+1/root n) (k and n are the sizes of positive set and the entire training set, respectively). Empirically, we compare our method with state-of-the-art instance-independent and instance-dependent PU algorithms on a wide range of synthetic, benchmark and real-world datasets, and the experimental results firmly demonstrate the advantage of the proposed method over the existing PU approaches.