Data Science

Which evaluation metrics are most reliable for imbalanced datasets in predictive classification?

RO Asked by Robert Wilson · 22-01-2024
0 upvotes 8,896 views 0 comments
The question

I am building a predictive model for fraud detection where the positive class is very rare. My accuracy is 99%, but it's failing to catch actual fraud. Between Precision-Recall curves, F1-Score, and ROC-AUC, which should I prioritize to ensure my model is actually performing well on the minority class? I need a metric that reflects real-world business impact rather than just statistical probability. 

3 answers

0
EL
Answered on 15-03-2024

In fraud detection, "Accuracy" is a trap because a model can be 99% accurate by simply guessing "No Fraud" every time. You should prioritize the Precision-Recall (PR) Curve over ROC-AUC. ROC-AUC can be overly optimistic when dealing with heavily imbalanced classes. The F1-Score is a great middle ground as it provides the harmonic mean of precision and recall. However, if the cost of missing a fraud case (False Negative) is much higher than a false alarm, you should specifically optimize for Recall while maintaining a tolerable level of Precision for your team. 

0
JA
Answered on 10-04-2024

Have you considered using a Cost-Benefit matrix alongside your metrics? It allows you to assign a dollar value to False Positives versus False Negatives. What is the actual cost of a missed fraud? 

CH 15-04-2024

That is a great point, James. For our bank, a missed fraud costs about $500 on average, while a false flag costs $10 in manual review time. By weighting the loss function in your XGBoost or Random Forest model to reflect this 50:1 ratio, the model will naturally prioritize the minority class. This shifts the focus from pure probability to actual business utility and ROI.

0
BA
Answered on 02-05-2024

I always use the AUPRC (Area Under the Precision-Recall Curve). It’s far more sensitive to the performance of the positive class than the standard ROC curve. 

RO 05-05-2024

Exactly, Barbara. AUPRC is the gold standard for rare-event prediction. I used it for a churn project last year and it changed our entire perspective on model success.

Share your thoughts

Your email address will not be published. Required fields are marked (*)

Professional Counselling Session

Still have questions?
Schedule a free counselling session

Our experts are ready to help you with any questions about courses, admissions, or career paths. Get personalized guidance from industry professionals.

Request a Call Back

Search Online

We Accept

We Accept

Follow Us

"PMI®", "PMBOK®", "PMP®", "CAPM®" and "PMI-ACP®" are registered marks of the Project Management Institute, Inc. | "CSM", "CST" are Registered Trade Marks of The Scrum Alliance, USA. | COBIT® is a trademark of ISACA® registered in the United States and other countries.

Book Free Session