Salvato in:
Dettagli Bibliografici
Autori principali: Ponti, Moacir Antonelli, Oliveira, Lucas de Angelis, Esteban, Mathias, Garcia, Valentina, Román, Juan Martín, Argerich, Luis
Natura: Preprint
Pubblicazione: 2022
Soggetti:
Accesso online:https://arxiv.org/abs/2210.11327
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911782260441088
author Ponti, Moacir Antonelli
Oliveira, Lucas de Angelis
Esteban, Mathias
Garcia, Valentina
Román, Juan Martín
Argerich, Luis
author_facet Ponti, Moacir Antonelli
Oliveira, Lucas de Angelis
Esteban, Mathias
Garcia, Valentina
Román, Juan Martín
Argerich, Luis
contents Real world datasets contain incorrectly labeled instances that hamper the performance of the model and, in particular, the ability to generalize out of distribution. Also, each example might have different contribution towards learning. This motivates studies to better understanding of the role of data instances with respect to their contribution in good metrics in models. In this paper we propose a method based on metrics computed from training dynamics of Gradient Boosting Decision Trees (GBDTs) to assess the behavior of each training example. We focus on datasets containing mostly tabular or structured data, for which the use of Decision Trees ensembles are still the state-of-the-art in terms of performance. Our methods achieved the best results overall when compared with confident learning, direct heuristics and a robust boosting algorithm. We show results on detecting noisy labels in order clean datasets, improving models' metrics in synthetic and real public datasets, as well as on a industry case in which we deployed a model based on the proposed solution.
format Preprint
id arxiv_https___arxiv_org_abs_2210_11327
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Improving Data Quality with Training Dynamics of Gradient Boosting Decision Trees
Ponti, Moacir Antonelli
Oliveira, Lucas de Angelis
Esteban, Mathias
Garcia, Valentina
Román, Juan Martín
Argerich, Luis
Machine Learning
Real world datasets contain incorrectly labeled instances that hamper the performance of the model and, in particular, the ability to generalize out of distribution. Also, each example might have different contribution towards learning. This motivates studies to better understanding of the role of data instances with respect to their contribution in good metrics in models. In this paper we propose a method based on metrics computed from training dynamics of Gradient Boosting Decision Trees (GBDTs) to assess the behavior of each training example. We focus on datasets containing mostly tabular or structured data, for which the use of Decision Trees ensembles are still the state-of-the-art in terms of performance. Our methods achieved the best results overall when compared with confident learning, direct heuristics and a robust boosting algorithm. We show results on detecting noisy labels in order clean datasets, improving models' metrics in synthetic and real public datasets, as well as on a industry case in which we deployed a model based on the proposed solution.
title Improving Data Quality with Training Dynamics of Gradient Boosting Decision Trees
topic Machine Learning
url https://arxiv.org/abs/2210.11327