Robust Loss Functions for Training Decision Trees with Noisy Labels
This work addresses the problem of noisy labels in decision tree training for machine learning practitioners, offering incremental improvements through theoretical insights and a new loss function.
The paper tackles training decision trees with noisy labels by analyzing robust loss functions and introduces a new adaptive loss, the negative exponential loss, which shows effectiveness in experiments across multiple datasets and noise settings.
We consider training decision trees using noisily labeled data, focusing on loss functions that can lead to robust learning algorithms. Our contributions are threefold. First, we offer novel theoretical insights on the robustness of many existing loss functions in the context of decision tree learning. We show that some of the losses belong to a class of what we call conservative losses, and the conservative losses lead to an early stopping behavior during training and noise-tolerant predictions during testing. Second, we introduce a framework for constructing robust loss functions, called distribution losses. These losses apply percentile-based penalties based on an assumed margin distribution, and they naturally allow adapting to different noise rates via a robustness parameter. In particular, we introduce a new loss called the negative exponential loss, which leads to an efficient greedy impurity-reduction learning algorithm. Lastly, our experiments on multiple datasets and noise settings validate our theoretical insight and the effectiveness of our adaptive negative exponential loss.