AUL is a better optimization metric  in PU learning

1 INTRODUCTION . Classic binary classification tasks in machine learning usually assume that all data are fully labeled as positive or negative ( PN learning ) . However , in real-world applications , dataset is usually nonideal and only a small fraction of positive data are labeled . Training a model from such partially labeled positive data is called positive-unlabeled ( PU ) learning . Take financial fraud detection as an example . Some fraudulent manners are found and can be labeled as positive , but we can not simply regard the remaining data as negative , because in most cases only a subset of fraud manners are detected and the remaining data may also contain undetected positive data . As a result , the remaining data can only be regarded as unlabeled . Other typical PU learning applications include text classification , drug discovery , outlier detection , malicious URL detection , online advertise , etc ( Yu et al . ( 2002 ) , Li & Liu ( 2003 ) , Li et al . ( 2009 ) , Blanchard et al . ( 2010 ) , Zhang et al . ( 2017 ) , Wu et al . ( 2018 ) ) . A naive way for PU learning is treating unlabeled data as negative and using traditional PN learning algorithms . But the model trained in this way is biased and its prediction results are not reliable ( Elkan & Noto ( 2008 ) ) . Some early works try to recover labels for unlabeled data by heuristic algorithms , such as S-EM ( Liu et al . ( 2002 ) ) , 1-DNF ( Yu et al . ( 2002 ) ) , Rocchio ( Li & Liu ( 2003 ) ) , k-means ( Chaudhari & Shevade ( 2012 ) ) . But the performance of heuristic algorithms , which is critical to these works , is not guaranteed . Some other kind of methods introduce an unbiased risk estimator to eliminate the bias ( Du Plessis et al . ( 2014 ) , Du Plessis et al . ( 2015 ) , Kiryo et al . ( 2017 ) ) . However , these methods rely on the knowledge of the proportion of positive samples in unlabeled samples , which is also unknown in practice . Another annoying problem of PU learning is how to accurately evaluate the model ’ s performance . Model performance is usually evaluated by some metrics , such as accuracy , precision , recall , Fscore , AUC ( Area Under ROC Curve ) , etc . During the life cycle of a model , its performance is usually monitored to ensure that the model is keeping a desired level of performance , with the variance and growth of data . In PU learning , the metrics above are also biased due to the lack of proportion of positive samples . Although Menon et al . ( 2015 ) proves that the ground-truth AUC ( AUC ) and the AUC estimated from PU data ( AUCPU ) is linearly correlated , which indicates that AUCPU can be used to compare the performances between two models , it ’ s still not possible to evaluate the true performance of a single model . Consider a situation when a model is evaluated on two different PU datasets generated from the same PN dataset but with different positive sample proportions . The ground-truth AUC which indicates the true performance of the model on two datasets are the same , but the AUCPU on the two datasets are different . Hence , AUCPU can not be used to directly evaluate the model ’ s performance . Jain et al . ( 2017 ) and Ramola et al . ( 2019 ) show that they can correct AUCPU , accuracyPU , balanced accuracyPU , F-scorePU and Matthews correlation coefficient , with the knowledge of proportion of positive samples . However , this proportion is difficult to obtain in practice . Recently many works focus on estimating the proportion of positive samples Du Plessis & Sugiyama ( 2014 ) , Christoffel et al . ( 2016 ) , Ramaswamy et al . ( 2016 ) , Jain et al . ( 2016 ) , Bekker & Davis ( 2018 ) , Zeiberg et al . ( 2020 ) , which are called mixture proportion estimation ( MPE ) algorithms . Yet according to our experiments on 9 datasets , the estimation methods still introduce some errors and thus make the corrected metrics inaccurate . Besides , the MPE algorithms may also introduce non-trivial computational overhead ( by up to 2,000 seconds per proportion estimation in our experiments ) , which slows down the evaluation process . In this work , we find that Area Under Lift chart ( AUL ) ( Vuk & Curk ( 2006 ) , Tufféry ( 2011 ) ) is a discriminating , unbiased and computation-friendly metric for PU learning . We make the following contributions . a ) . We theoretically prove that AUL estimation is unbiased to the ground-truth AUL and calculate a theoretical bound of the estimation error . b ) . We carry out experimental evaluation on 9 datasets and the results show that the average absolute error of AUL estimation is only 1/6 of AUC estimation , which means AUL estimation is more accurate and more stable than AUC estimation . c ) . By experiments we also find that , compared with state-of-the-art AUC-optimization algorithm , AUL-optimization algorithm can not only significantly save the computational cost , but also improve the model performance by up to 10 % . The remaining of this paper is organized as follows . Section 2 describes the background knowledge . Section 3 theoretically proves the unbiased feature of AUL estimation in PU learning . Section 4 evaluates the performance of AUL estimation by experiments on 9 datasets . Section 5 experimentally shows the performance of AUL-optimization algorithm by applying AUL in PU learning . Section 6 concludes the whole paper . 2 BACKGROUND . Binary Classification Problem : Let D = { < xi , yi > , i = 1 , ... n } be a positive and negative ( PN ) dataset which has n instances . Each tuple < xi , yi > is a record , in which xi ∈ Rd is the feature vector and yi ∈ { 1 , 0 } is the corresponding ground-truth label . Let XP , XN be the feature vectors set of positive , negative samples respectively , and nP , nN be the number of samples in these sets respectively . XP = { xi|yi = 1 , i = 1 , ... nP } XN = { xi|yi = 0 , i = 1 , ... nN } In PU learning , we use α = n P nP+nN = n P n to indicate the proportion of positive samples in all samples . Confusion Matrix : A confusion matrix is used to discriminate the model performance of different binary classification algorithms . In confusion matrix , true positive ( TP ) ( actual label and predicted label are both positive ) , true negative ( TN ) ( actual label and predicted label are both negative ) , false positive ( FP ) ( actually negative but predicted as positive ) , and false negative ( FN ) ( actually positive but predicted as negative ) are counted according to model ’ s outputs . Obviously , nTP + nFN = nP , nTN + nFP = nN . ROC : Since the numbers of TP , TN , FP and FN in a confusion matrix are highly related to the classification threshold ( θ ) , Receiver Operating Characteristic ( ROC ) curve ( Fawcett & Tom , 2003 ) is proposed to plot ( x , y ) = ( fpr ( θ ) , tpr ( θ ) ) over all possible classification thresholds θ . In some literature , tpr is also known as sensitivity and the value of 1− fpr is called specificity . true positive rate ( tpr ) = nTP nP , false positive rate ( fpr ) = nFN nN AUC : As a curve , ROC is not convenient enough to describe the model performance . Consequently , the Area Under ROC Curve ( AUC ) , which is a single value , is proposed and widely used as a metric to evaluate a binary classification algorithm . AUC provides a summary of model performance under all possible classification thresholds . It also provides an elegant probabilistic interpretation that AUC is the probability of correct ranking between a random positive sample and a random negative sample ( Hanley & McNeil , 1982 ) , which is a kind of ranking capability . According to Vuk & Curk ( 2006 ) , for a model g : Rd → R , AUC can be computed as follows , AUC = 1 nPnN ∑ xi∈XP ∑ xj∈XN S ( g ( xi ) , g ( xj ) ) ( 1 ) where S ( a , b ) = 1 a > b 1 2 a = b 0 a < b It is worth noting that , there are other ways to calculate AUC , but they are essentially the same . AUL : Lift curve , which is popular in econometrics to decide a suitable marketing strategy ( Tufféry ( 2011 ) , Vuk & Curk ( 2006 ) ) , has not been well studied in machine learning field . Lift curve can be seen as a variant of ROC and it illustrates ( x , y ) = ( Yrate ( θ ) , tpr ( θ ) ) over all possible classification thresholds θ. Yrate represents the proportion of samples predicted as positive . Yrate = nTP + nFP n In the curve figure , Lift curve has the same y-axis as ROC curve , but a different x-axis . Area Under Lift chart ( AUL ) ( Vuk & Curk ( 2006 ) , Tufféry ( 2011 ) ) , can also be used as a metric to evaluate the model performance . One way to compute AUL is AUL = 1 nPn ∑ xi∈XP ∑ xj∈XP∪XN S ( g ( xi ) , g ( xj ) ) ( 2 ) Essentially , AUL can be regarded as the probability of correct ranking between a random positive sample and a random sample . AUL and AUC is linearly related ( Tufféry , 2011 ) , i.e . AUL = 0.5α+ ( 1− α ) AUC which shows that AUL has the same discriminating power with AUC . 3 UNBIASEDNESS OF AUL ESTIMATION IN PU LEARNING : THEORETICAL PROOF . A PU dataset D′ = { < xi , yi , si > , si ∈ { 1 , 0 } , i = 1 , ... n } is generated by sampling a subset of positive data as labeled and leaving remain as unlabeled from D. In D′ , si is the observed label and yi is the ground-truth label which may be unknown . If si = 1 , we can confirm yi = 1 ( positive ) . If si = 0 , yi would be 1 or 0 . In this paper , we assume that the labeled data is Select Completely At Random ( SCAR ) ( Bekker & Davis , 2018 ) from positive data . Therefore the distribution of labeled samples in D′ are the same as the distribution of positive samples in D. Let XL , XU be the feature vectors set of labeled and unlabeled samples respectively , and nL , nU be the number of samples in these sets respectively . XL = { xi|si = 1 , i = 1 , ... nL } XU = { xi|si = 0 , i = 1 , ... nU } We use β = n L nP to indicate the proportion of labeled samples in positive samples . AUC Estimation is Biased To calculate AUC with PU dataset ( AUCPU ) , unlabeled data is regarded as negative , thus we have AUCPU = 1 nLnU ∑ xi∈XL ∑ xj∈XU S ( g ( xi ) , g ( xj ) ) ( 3 ) where function S is the same as in Eq.1 . The expectation of AUCPU over the distribution of D′ is E [ AUCPU ] = 1− α 1− αβ ( AUC − 0.5 ) + 0.5 This formula is slightly different from the one in Menon et al . ( 2015 ) . Here we define AUC on a specific dataset but not on a distribution . This formula indicates thatAUCPU is an biased estimation of AUC . We demonstrate the bias on an example dataset ( 1a ) , which contains 20 samples sorted by prediction score . Figure 1b illustrates two ROC curves on this dataset . curve-ROC is ploted with ground-truth label y and curve-ROCPU is ploted with observed label s. We can see that curve-ROC is almost above curve-ROCPU . The corresponding AUC is AUC = 0.740 and AUCPU = 0.653 respectively . There is a big difference ( 0.087 ) among the two measurements . As we discussed in section 1 , ( Jain et al. , 2017 ) tries to recover AUC from AUCPU from the estimation of 1−α1−αβ . To estimate 1−α 1−αβ , some works ( Elkan & Noto ( 2008 ) , Du Plessis & Sugiyama ( 2014 ) , Sanderson & Scott ( 2014 ) , Jain et al . ( 2016 ) , Ramaswamy et al . ( 2016 ) , Christoffel et al . ( 2016 ) , Bekker & Davis ( 2018 ) , Zeiberg et al . ( 2020 ) ) develop their mixture proportion estimation ( MPE ) algorithms . But according to our experiment on 9 datasats , these algorithms are neither accurate enough nor time saving . AUL Estimation is Unbiased Similar to AUCPU , AUL with PU dataset ( AULPU ) can be calculated as AULPU = 1 nLn ∑ xi∈XL ∑ xj∈XL∪XU S ( g ( xi ) , g ( xj ) ) ( 4 ) Unlike AUCPU , AULPU is unbiased estimation of AUL . In contrast to Figure 1b , Figure 1c illustrates two lift curves which are very close to each other . curve-lift is ploted with ground-truth label y and curve-liftPU is ploted with observed label s. The corresponding AUL , AUL = 0.620 and AULPU = 0.615 , are very close . We then prove the unbiasedness . Theorem 1 For a given classifier g : Rd → R , a PN dataset D with the proportion of labeled samples in positive samples β = n L nP , a PU dataset D′ can be generated following SCAR , the expectation and variance of AULPU over the distribution of D′ are as follows , E [ AULPU ] = AUL Var [ AULPU ] = nP − nL nP − 1 σ2 nL where σ2 is the variance of { 1 n ∑ xj∈XP∪XN S ( g ( xi ) , g ( xj ) ) , i = 1 , ... n P } . Proof Let txi = 1 n ∑ xj∈XP∪XN S ( g ( xi ) , g ( xj ) ) = 1 n ∑ xj∈XL∪XU S ( g ( xi ) , g ( xj ) ) then , AUL = 1 nP ∑ xi∈XP txi AULPU = 1 nL ∑ xi∈XL txi XL is generated by random sampling without replacement from XP , hence AULPU is the estimation of the mean of { txi , xi ∈ XP } which isAUL . According the theory of simple random sampling without replacement ( Lohr ( 2009 ) ) , the estimated population mean AULPU is an unbiased estimator of the population mean AUL , i.e . E [ AULPU ] = AUL , and variance of AULPU is Var [ AULPU ] = ( 1− n L nP ) 1 nL ( ∑ xi∈XP ( txi − t ) 2 nP − 1 ) = nP − nL nP − 1 1 nL ( ∑ xi∈XP ( txi − t ) 2 nP ) = nP − nL nP − 1 σ2 nL According to Theorem 1 , applying Chebyshev ’ s inequality , we have P ( |AUL−AULPU | ≥ ) ≤ Var = σ2 nL ( 1− β ) n P nP − 1 where Var is the variance of AULPU , note that 0 < txi < 1 , hence σ2 = E [ ( t− t ) 2 ] ≤ E [ ( t− t ) 2 − ( 0− t ) ( t− 1 ) ] = E [ −2tt+ t2 + t ] = t− t2 ≤ 1 4 then P ( |AUL−AULPU | ≥ ) ≤ 1− β 4nL nP nP − 1 ≈ 1− β 4nL This equation gives a theoretical bound for the error between AUL and AULPU . Hundreds of labeled samples nL can reduce the error to an acceptable level . score 0.92 0.82 0.73 0.66 0.6 0.58 0.54 0.5 0.45 0.43 y 1 1 0 1 1 1 1 0 0 0 s 1 0 0 1 0 0 1 0 0 0 score 0.41 0.39 0.38 0.36 0.35 0.3 0.25 0.2 0.15 0.1 y 1 0 1 1 0 0 0 1 0 0 s 1 0 0 0 0 0 0 1 0 0 ( a )

Summary: In this paper, the author proposed to use Area Under Lift chart (AUL) as a new optimization metric for positive unlabeled (PU) learning. The proposed AUL can be estimated unbiasedly from PU data, without the need to estimate the mixture proportions. Experiments on several datasets show that the proposed method outperforms AUC optimization algorithms.