Gérer les ensembles de données déséquilibrés

Détection de la fraude en R

Bart Baesens

Professor Data Science at KU Leuven

Ensembles de données déséquilibrés

Défi clé : étiqueter les événements comme fraude ou non
- Défi majeur pour la classification et la détection d'anomalies
Le classifieur favorise la classe majoritaire (= non-fraude)
- forte erreur de classification sur les fraudes
Les classifieurs apprennent mieux avec une distribution équilibrée

meule de foin et aiguille

Ensembles de données déséquilibrés

Défi clé : étiqueter les événements comme fraude ou non
- Défi majeur pour la classification et la détection d'anomalies
Le classifieur favorise la classe majoritaire (= non-fraude)
- forte erreur de classification sur les fraudes
Les classifieurs apprennent mieux avec une distribution équilibrée
Solution possible : modifier la distribution des classes par échantillonnage

meule de foin et aiguille

Déséquilibre initial

Suréchantillonner la classe minoritaire…

class_barplot_over

… ou sous-échantillonner la classe majoritaire…

class_barplot_under

… ou les deux !

class_barplot_both

Résultat après échantillonnage…

class_barplot_result2

… ou comme ceci

class_barplot_result1

Suréchantillonnage aléatoire (ROS)

orig_data_v0

orig_data_train_test

random_oversampling_v0

random_oversampling_v1

Suréchantillonnage aléatoire en pratique

Jeu de données Credit Card Fraud Detection sur Kaggle
- $\sim$ 300 k virements par carte de crédit, anonymisés, étiquetés fraude ou légitime
À propos des données…
- Variables numériques (anonymisées) : V1, V2, … , V28
- Time = secondes écoulées entre chaque virement et le premier du jeu de données
- Amount = montant de la transaction
- Class = variable cible : valeur 1 en cas de fraude, sinon 0

creditcard_V2vsV1

Vérifier le déséquilibre

head(creditcard)

  Time         V1         V2  ...            V27         V28 Amount Class
1    0  1.1918571  0.2661507  ...  -0.0089830991  0.01472417   2.69     0
2   10  0.3849782  0.6161095  ...   0.0424724419 -0.05433739   9.99     0
3   12 -0.7524170  0.3454854  ...  -0.1809975001  0.12939406  15.99     0
4   17  0.9624961  0.3284610  ...   0.0163706433 -0.01460533  34.09     0
5   34  0.2016859  0.4974832  ...   0.1427572469  0.21923761   9.99     0

prop.table(table(creditcard$Class))

   0    1 
0.98 0.02

ovun.sample du paquet ROSE

n_legit <- 24108
new_frac_legit <- 0.50
new_n_total <- n_legit / new_frac_legit ## = 24108 / 0.50 = 48216

library(ROSE)
oversampling_result <- ovun.sample(formula = Class ~ ., data = creditcard,
                                   method = "over",
                                   N = new_n_total,
                                   seed = 2018)

oversampled_credit <- oversampling_result$data
prop.table(table(oversampled_credit$Class))

  0   1 
0.5 0.5

creditcard_V2vsV1_oversampled

Passons à la pratique !

Détection de la fraude en R