Communication Dans Un Congrès Année : 2023

Binary Classification vs. Anomaly Detection on Imbalanced Tabular Medical Datasets

Résumé

When working on imbalanced medical datasets for a classification task, the chosen method depends mainly on label availability. We consider two classes, e.g., normal and abnormal. If the dataset is labeled, we can choose Binary Classification; otherwise, we should opt for Anomaly Detection. We aim to determine the trade-off between using labeled data with Binary Classification and unlabeled data with Anomaly Detection while being aware of the labeling effort. The comparison is made on BabyGarches, an original dataset of newborn motricity, and on two publicly available imbalanced tabular medical datasets, Thyroid and Cardioctography. We observe that Binary Classification outperforms Anomaly Detection on all datasets, achieving the highest Area Under the Curve ROC scores. Nevertheless, the performance of Deep Anomaly Detection methods such as Autoencoders and GANs is close to Binary Classification models' performance. Moreover, they reach the best Recall scores, which physicians appreciate. They may thus be recommended when data labeling is costly. Since labeling our use case dataset BabyGarches is expensive, Anomaly Detection is relevant to its classification.

Fichier non déposé

Dates et versions

hal-04567598 , version 1 (03-05-2024)

Identifiants

Citer

Philippe Rambaud, Adel Taleb, Raphael Fauches, Arpad Rimmel, Joanna Tomasik, et al.. Binary Classification vs. Anomaly Detection on Imbalanced Tabular Medical Datasets. 2023 Congress in Computer Science, Computer Engineering, & Applied Computing (CSCE), Jul 2023, Las Vegas, France. pp.01-05, ⟨10.1109/CSCE60160.2023.00220⟩. ⟨hal-04567598⟩
267 Consultations
0 Téléchargements

Altmetric

Partager

  • More