Many resources on machine learning (ML) classification problems recommend that if one's dataset has unbalanced class sizes, one should modify the data to have equal class counts. Yet it is shown here that this is both unnecessary and often harmful. Alternatives are presented.
Why is imbalance a problem? Our ML algorithm may never predict the small class, making the algorithm rather useless.
A popular solution is to artificially balance the data.
This changes the class probabilities, with undesirable consequences.
Related problem: the class probabilities may be wrong to begin with, either due to the original sampling scheme or because the probabilities may change.
Two problems will be addressed in this document:
What can be done instead of artificially balancing the data?
What can we do to adjust data with the wrong class probabilities?
Solutions:
Regarding imbalance, we consider but reject ROC, and instead argue that informal use of conditional probabilities is the best solution.
Regarding incorrect class probabilities, we develop an adjustment formula.
Illustrations of the (perceived) problems and offered remedies appear in numerous parts of the ML literature, ranging from Web tutorials to the research literature. Major packages, such as caret, parsnip, and mlr3, also offer remedies.
All of these sources recommend that you artificially equalize the class counts in your data, via various resampling methods. Say for instance we are in the two-class case, and have class sizes of 20000 and 80000 for class 1 and class 0, respectively. Here are some ways to rebalance.
Randomly discard 60000 data points from class 0.
Resample 80000 data points from the 20000, with replacement.
Using distribution approximation methods such as SMOTE, resample 80000 from the 20000.
Upon closer inspection, though, one sees that this is generally inadvisable, indeed harmful, for several reasons:
Undersampling is clearly problematic: Why throw away data? Discarding data weakens our ability to predict new cases.
The data may be unbalanced for a reason. Thus the imbalance itself is useful information, again resulting in reduced predictive power if it is ignored.
There are principled alternatives to resampling, including an adjustment formula to be presented here.
In other words:
Resampling methods are both harmful and unnecessay.
This is a Kaggle dataset. Quoting from the Kaggle site,
The datasets contains transactions made by credit cards in September 2013 by european cardholders. This dataset presents transactions that occurred in two days, where we have 492 frauds out of 284,807 transactions. The dataset is highly unbalanced, the positive class (frauds) account for 0.172% of all transactions.
Note the phrase, "highly unbalanced."
Due to privacy concerns, PCA has been used to replace most of the features, though two original features have been retained.
This is a Kaggle dataset on whether patients keep their medical appointments. The imbalance here is more mild, with about 20% of the patients being no-shows.
This is a well-known UCI Machine Learning Repository dataset. Again quoting from the site:
The objective is to identify each of a large number of black-and-white rectangular pixel displays as one of the 26 capital letters in the English alphabet. The character images were based on 20 different fonts and each letter within these 20 fonts was randomly distorted to produce a file of 20,000 unique stimuli. Each stimulus was converted into 16 primitive numerical attributes (statistical moments and edge counts) which were then scaled to fit into a range of integer values from 0 through 15.
This dataset is close to balanced, with each letter appearing about 775 times.
The dataset is conveniently available in the mlbench
package.
In an X-ray classification tudy from Mount Sinai Hospital the classification method worked well on the original hospital data, but not in prediction of new cases at other locations. The study's authors found that an important factor underlying the discrepancy was that the pi vary from one hospital to another. Here the class probabilities really do change, not artificially, but the issues are the same, and again an adjustment procedure would be desirable.
In a study by Fawcett and Provost, the authors found that
The level of fraud changes dramat-ically month-to-month because of modifications to work practices (both the carrier’s andthe bandits’).
We refer to the class probabilities for given feature values as conditional class probabilities, because they are probabilities subject to conditions. If say we wish to classify a patient as to whether she has a certain disease or not, and we know her blood test value on some measure is 2.39, the latter is the condition. Among all patients with that test value, what proportion of them have the disease?
The overall class probabilities, e.g. the 0.000172 value above, are unconditional class probabilities.
The conditional and unconditional class probabilities are often referred to as the posterior and prior probabilities. This sounds Bayesian, and indeed we do use Bayes' Rule, but there is no subjectivity involved. These are real probabilities. E.g. in a disease classification application, there is a certain proportion of people in the population who have the disease. This is a prior probability, which we will estimate from our training data.
2-class case (Y = 0,1): p: P(Y = 1) (probability of class 1) ri: estimated q1(Xi)
The examples above illustrate two important cases:
The fraud data is imbalanced, but naturally so. Assuming the two-day data collection period was typical, the population class probability for fraud will be about what we see in the data, 0.172%.
The LetterRecognition data is balanced, but only artificially so. The curator of the dataset wanted the data to have about the same number of instances of each letter. But in general English usage, letters occur with quite different frequencies:
E 12.02% T 9.10% A 8.12% O 7.68% I 7.31% N 6.95% ... ... Q 0.11 J 0.10 Z 0.07
(source). One can obtain these numbers in regtools
:
> data(ltrfreqs)
> lf <- ltrfreqs
> lf <- ltrfreqs[,2] / 100
> names(lf) <- ltrfreqs[,1]
# example
> lf['A']
A
0.0812
Consider the 2-class setting, with Y = 0,1. As above, denote the (conditional) probability that Y = 1 for a particular new case to classify by ri. We might have an estimate of ri from fitting a logistic model, for instance.
The rule that minimizes the overall probability of misclassification is:
Ypred = 1 if ri >= 0.5
Ypred = 0 if ri < 0.5
But note the criterion here, minimizing the overall probability of misclassification. Other criteria are possible. It may be, for instance, that we wish to place different weights on false positives and false negatives. In the credit card fraud case, a false negative may mean a major loss, much worse than what we lose with a false positive (wasted investigation time, etc.). We may define our loss in those terms, and seek to minimize expected loss.
Specifically, let l01 be the loss we incur if we guess Y = 0 but Y = 1. In the opposite case -- guess 1 but Y = 0 -- say our loss is 1.0. (All that matters is the ratio of the two losses.) To minimize expected loss, the optimal rule can be shown to be (see Appendix A below)
Ypred = 1 if ri > 1/(1+l10)
Ypred = 0 if ri <= 1/(1+l10)
The loss value l01 may be hard to quantify, and in any case, we argue below that any mechanical rule is overly constraining anyway.
ML algorithms take your data literally. Say you have a two-class setting, for Classes 0 and 1. If about 1/2 your data is Class 1, then the algorithm, whether directly or indirectly, operates under the assumption that the true population class probabilities are each about 0.5.
In the LetterRecognition data, since the sampling was actually designed to have about the same number of instances for each letter, the algorithm you use will then assume the true probabilities of the letters are about 1/26 each. We know that is false, as the table shown earlier illustrates.
So, if your sampling scheme artificially creates balanced data, as in the LetterRecognition data, or if you do resampling to make your data balanced, as is commonly recommended, you are fooling your ML algorithm.
In fooling your algorithm, it will generate the wrong conditional class probabilities ri in our notation above. And whether we wish to minimize the overall probability of misclassification, or expected loss, or any other criterion, the algorithm will (again, directly or indirectly) rely on the values of ri.
Consider the fraud example, in which the data are highly imbalanced but in which wrongly guessing the large class carries heavy penalties for us. Recall our notation l01 from above. As noted, this may be difficult to quantity, but for the moment, let's suppose we can do so. What are the implications in terms of artificially balancing the data?
Actually, if we are going to be doing any adjustment of class sizes in our data, we should make the fraud class larger than the non-fraud class, not of equal size. How much larger will depend on the value of l01, but in any case, balancing the data will be wrong.
Frank Harrell says it well:
For this reason the odd practice of subsampling the controls is used in an attempt to balance the frequencies and get some variation that will lead to sensible looking classifiers (users of regression models would never exclude good data to get an answer). Then they have to, in some ill-defined way, construct the classifier to make up for biasing the sample. It is simply the case that a classifier trained to a 1⁄2 [q = 1/2] prevalence situation will not be applicable to a population with a 1⁄1000 [p = 1/1000] prevalence.
Clearly, one's course of action should center around the conditional class probabilities ri. (Note the plural; each case will have its own value of ri.) So, specifically, how should we use them? We will discuss two approaches:
Use of the ROC curve, which is derived from the ri values.
Informal, nonmechanical consideration of the r values.
Our recommendation will be Approach 2 above.
Here one considers various threshhold values h, where we guess class 1 if r > h, 0 otherwise. The value one chooses for h will determine the True Positive Rate and False Positive Rate:
TPR(h) = P(Ypred = 1 | Y = 1) = ∫t > h f1(t) dt
FPR(h) = P(Ypred = 1 | Y = 0) = ∫t > h f0(t) dt
The ROC curve is then a graph of TPR vs. FPR. As we vary h, it traces out the ROC curve. Some R packages, such as ROCR, colorize the curve, with the color showing the value of h.
The idea here is that even if we cannot quantity l01, we can at least explore various values of h to produce a decision rule that roughly reflects the relative values we place on true and false positives.
Often a 45-degree line is superimposed on the ROC graph for comparison. To see why, say we are classifying patients as having a certain disease or not, say on the basis of a blood test. For a value of h associated with a point on the line, diseased and disease-free patients would have the same probability of testing positive, indicating that the test has no discriminatory value.
A note on AUC: AUC is the total area under the ROC curve. As such, it is a measure of the general predictive ability of your algorithm on this data. This may seem attractive at first, but it is probably irrelevant in most applications, as it places equal weight on all possible TPR/FPR scenarios; usually we are far more interested in some settings than others. Note too that AUC values for original data vs. the artificially balanced data are not comparable.
The key points are:
ROC involves macro-level quantities, TPR and FPR. In highly unbalanced settings, we are concerned with the micro-level, rare events where the much smaller class occurs.
Mechanical rules are too constraining for many applications.
This latter point is especially relevant. Often the required decision is too critical to be left up to a machine. For instance, relevant to our fraud example, in Fraud: a Guide to Its Prevention, Detection and Investigation by Price Waterhouse Coopers, it is pointed out that
... every fraud incident is different, and reactive responses will vary depending on the facts that are unique to each case.
Use of a machine learning algorithm to make a mechanical decision would obviate this needed flexibility.
Instead, a practical, effective approach would be to simply look at the r values for the new cases directly. In other words, we have the algorithm only take the first step, after which we work "by hand" rather than by computer. We would still set a threshold h, yes, but would use it only as first-stage screening, with the second stage being done by humans.
At that point, the (human) auditor would take into account both that estimated probability -- now worrying not only that it is larger than h but also how much larger -- as well as such factors as the amount of the charge, special characteristics not measured in the available features, and so on. The auditor may not give priority, for instance, to a case for which the probability is above h but the monetary value of the transaction is small. On the other hand, if the probability is far above h, the auditor may give this transaction a closer look even if the monetary value is small.
A related issue is that of ranking. In the credit card fraud example, for instance, we may have sufficient audit funding to investigate 100 transactions per month. Thus we would rank the ri, and cull out the top 100 cases.
Some ML algorithms naturally provide the ri, while others do not.
Below is an example for the logistic model, for a data frame ccf containing the credit card fraud data, with class variable Class. Say we take h = 0.25. We'll cull out the cases with at least that probability of fraud, for preparation of investigation "by hand."
> glmout <- glm(Class ~ .,data=ccf,family=binomial)
> condProbs <- predict(glmout,ccf,type='response')
> toCheck <- which(condProbs > 0.25)
> names(toCheck) <- NULL
> head(toCheck)
[1] 542 6109 6332 6335 6337 6339
So we'd check cases 542, 6109 and so on by hand.
On the other hand, the SVM method does not produce the ri. In addition, even the ri produced by, e.g. glm() may have biases on the edges of the data. Thus an external method is needed.
Many implementations of SVM use a method known as Platt scaling. This
assumes a logistic model from the regression function of Y (2-class
case) against the SVM scores. In regtools
, we instead use an method
we developed, which performs a k-Nearest Neighbors regression of Y
against the scores. (In non-SVM cases, the scores can be anything used
by the ML method at hand to predict Y.) Our method is implemented in
the regtools
function scoresToProbs()
.
> ma <- read.csv('KaggleV2-May-2016.csv',header=T,stringsAsFactors=T)
> ma <- ma[,-c(1,2,4,5,7)]
> svmout <- qeSVM(ma,'No.show',holdout=NULL)
Let's predict a hypothetical patient with the characteristics of patient 8 in our training data, but who is male:
> newx <- ma[8,-9]
> newx$Gender <- 'M'
> predict(svmout,newx)
$predClasses
8
No
Levels: No Yes
We predict him to show up for the appointment. But what is the probability of that?
> predict(svmout,newx,k=75)
$predClasses
8
No
Levels: No Yes
$probs
No Yes
[1,] 0.8133333 0.1866667
What happened here? Our function predict.qeSVM
includes this code:
if (!is.null(k)) {
y <- object$y
trnScores <- object$decision.values
newScores <- getDValsE1071(object, newx)
probs <- scoresToProbs(y, trnScores, newScores, k)
res$probs <- probs
}
Armed with this probability estimate for the given patient, we can make an informal decision as to whether to invest time and effort to make sure he does show up, say by phone call reminders, additional text messages etc.
And what about ranking? We would call predict() on our current set of new cases, then rank the resulting probabilities. Just as an example, let's take our training set as the "new cases."
> preds <- predict(svmout,svmout$x,k=75)
> probs <- preds$probs
> probsOrder <- order(probs)
> probsSorted <- probs[probsOrder]
> head(probsSorted,100)
[1] 0.01333333 0.01333333 0.01333333 0.01333333 0.01333333 0.01333333
[7] 0.01333333 0.01333333 0.01333333 0.01333333 0.01333333 0.01333333
[13] 0.01333333 0.01333333 0.01333333 0.01333333 0.01333333 0.01333333
[19] 0.01333333 0.01333333 0.02666667 0.02666667 0.02666667 0.02666667
[25] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[31] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[37] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[43] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[49] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[55] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[61] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[67] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[73] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[79] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[85] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[91] 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667 0.02666667
[97] 0.02666667 0.02666667 0.02666667 0.02666667
> head(probsOrder,100)
[1] 111857 116081 116105 117394 121485 132575 139922 141078 156201 158322
[11] 166633 169521 172965 174563 191263 191534 193848 194199 200048 211445
[21] 111474 111529 113535 113572 113577 113579 113582 113584 113585 113590
[31] 113593 113598 113617 113634 113636 113637 113640 114420 115955 116040
[41] 117506 117538 117540 117542 117543 117546 117548 117551 117555 117562
[51] 117565 117628 117634 117638 117639 117640 117641 117648 117651 117663
[61] 117693 117715 117718 117773 117842 117843 117845 117847 117849 117853
[71] 117856 117956 117962 117966 117969 117970 117971 117979 117982 118010
[81] 118013 118015 118018 118021 118024 118026 118027 118037 118038 118041
[91] 118043 118048 118112 118125 118127 118130 118131 118133 118138 118139
So we might investigate case numbers 111857, 116081 and so on.
As noted in the LetterRecognition data example, in some cases the data are artificially balanced to begin with, due to the sampling design. Or we may have a situation like that of the Mt. Sinai Hospital radiology data, in which the class probabilities may change from one site to another, or may change over time as in the cell phone fraud example. Thus, our estimated values of the pi will be wrong. Can we fix that?
For simplicity, we'll assume the two-class setting here, with Class 0
and Class 1. This is the code for adjustment (the function is part of
the regtools
package):
classadjust <- function (econdprobs, wrongprob1, trueprob1)
{
wrongratio <- (1 - wrongprob1)/wrongprob1
fratios <- (1/econdprobs - 1) * (1/wrongratio)
trueratios <- (1 - trueprob1)/trueprob1
1/(1 + trueratios * fratios)
}
}
where
condprobs
is the vector of conditional class probabilities for
the dataset at hand, reported by the software applied to that data using
incorrect pi
wrongratio
is the ratio of the numbers of Class 0 to Class 1
datapoints in our dataset
trueratio
is the actual such ratio
The return value is the set of adjusted conditional class probabilities.
For instance, suppose we are in a setting in which there are equal
numbers of the two classes in our dataset, yet we know the true
(unconditional) class probabilities are 0.2 and 0.8 for Classes 0 and 1.
Then wrongratio
would be 0.5/0.5 = 1.0, and trueratio
would be
0.2/0.8 = 0.25.
What if the data are balanced but the true unconditional class probabilities are unknown? Other than creating various scenarios involving the true values and exploring the results, there is not much that we can do. In essence, the result is a maximum-likelihood kind of situation: Our predicted class will be the one whose (nominal) conditional probability makes our feature data most likely given the class (see Appendix B below), which is very different from the question we want to ask, Which class is most likely given the feature set? We'll get answers ("Hey, the computer said such and such!"), but those answers will be of questionable value unless the predictors have very strong predictive power.
Note that setting utility values for "false positive," "false negative" and so on does not solve the problem. One still needs to factor in the class probabilities in order to maximize expected utility.
Once again, balancing the data will not help.
There's really no need to artificially balance your data, and doing so could be harmful.
Please don't throw away data.
Conversely, creating extra data by up-sampling distorts the analysis and may undermine our predictive power.
If your data is "naturally" unbalanced, meaning it reflects population structure as in the credit card fraud example, don't artificially force balance. Instead, choose a threshhold for conditional probabilities, and flag new cases that exceed it.
If your data is unrealistically balanced, as in the LettersRecognition example, and the true unconditional class probabilities are known, use the adjustment formula to convert the reported unconditional probabilities to realistic ones, and classify using them.
If your data is unrealistically balanced but the true unconditional class probabilities are unknown, recognize that your ML analysis may have only a very restricted interpretation and value.
Say the random variable W takes on the values 0,1, with P(W = 1) = r. (In the above, W plays the role of Y, conditioned on X.) Let's compute the expected loss under two strategies:
Then our loss is 0 P(W = 0) + l01 P(W = 1) = l01 r.
Then our loss is 1 P(W = 0) + 0 P(W = 1) = 1-r.
So our optimal stragegy is to guess W = 1 if and only if 1-r < l01 r. In other words,
Guess W = 1 if and only if r > 1/(1+l01).
(For ease of notation etc., no distinction will be made here between sample and population quantities.)
Say there are two classes, labeled 1 and 0. Let Y denote the label and X denote the features, and say we have a new case with X = t. Then the true equation is
P(Y = 1 | X = t) = p f1(t) / [p f1t) + (1-p) f0(t)] (Eqn. 1)
where p is P(Y = 1), the true unconditional probability of Class 1, and fi(t) is the conditional density of X within Class i. (If X is a discete random variable, substitute a probability for f.)
Rewrite the above as
P(Y = 1 | X = t) = 1 / [1 + {(1-p)/p} f0(t) / f1(t)] (Eqn. 2)
Now suppose the analyst artificially changes the class counts in the data (or, as in the LetterRecognition example, the data is artificially sampled by design), with proportions q and 1-q for the two classes. In the case of artificially equalizing the class proportions, we have q = 0.5. Then the above becomes, in the eyes of your ML algorithm,
P(Y = 1 | X = t) = 1 / [1 + {(1-q)/q} f0(t) / f1(t)] (Eqn. 3)
As noted earlier, what the ML algorithm is computing, directly or
indirectly, is P(Y = 1 | X = t). Moreover, as also noted earlier, even
caret
and mlr3
do make these quantities available for the various t,
so we can solve for f0(t) / f1(t):
f0(t) / f1(t) = [g(t)-1] q/(1-q) (Eqn. 4)
where
g(t) = 1 / P(Y = 1 | X = t)
Keep in mind that the g values are the ones output by our ML algorithm. Setting t to the matrix of our feature values, we can plug t into Eqn. 4 to get the vector of f0/f1 values, and plug the result into Eqn. 2 to obtain the vector of correct conditional probabilities.
We can now substitute in (Eqn. 2) from (Eqn. 4) to get the proper conditional probability.
The general m-class case. classes 0,1,...,m-1, actually reduces to the 2-class case, because
P(Y = i | X = t)
can be viewed as the conditional probability of class i vs. all other classes.
For each j = 1,...,c, denote by qj and pj the probabilities assumed by our ML algorithm, and the actual probabilities, respecitvely. For example, if the data are artificially balanced, either by resampling or by the original sampling design, we have qj = 1/c for all j.
To guide our intuition, note first that Eqn. 2 is now
P(Y = j | X = Xi) = 1 / [1 + {(1-pj)/pj} [Σm ≠ j fm(Xi)] / fj(Xi) Eqn. 5
Now to handle the multiclass case, for each j, j = 1,...,c and each i = 1,...,n, do:
gij = 1 / P(Y = j | X = Xi),
[Σm ≠ j fm(Xi)] / fj(Xi) = [gij - 1] qj / (1 - qj) Eqn. 6
Say we have balanced data, so the qi = 1/m for each i. Then in predicting the class of a new case having X = t, Equation 1 becomes
P(Y = i | X = t) = fi(t) / [ Σj fjt) ]
for i = 0,1,...,m-1, since all the 1/m factors cancel.
This shows that our guessed class is
j = arg maxi fi(t)
In other words, you are in effect asking, "Within which class j would our data X = t be most likely?" That is, we are maximizing (say in the discrete case)
P(X = t | Y = j)
over j. That's completely different from the question we really are interested in, "Which class j is most likely, given X = t?", i.e. maximizing
P(Y = j | X = t )
But it does show that if we artificially equalize the class sizes, we are finding the Maximum Likelihood Estimate of j, if the pi are unknown.
If we really don't know the true class probabilities pi, and artificially equalize the class sizes, we are at least getting a kind of MLE. However, what then is the practical meaning? Unless you are a subjective Bayesian (I am not), setting a flat prior, there is not much here.
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.