arXiv · 2501.07754
Universal Training of Neural Networks to Achieve Bayes Optimal Classification Accuracy
Abstract
This work invokes the notion of $f$-divergence to introduce a novel upper bound on the Bayes error rate of a general classification task. We show that the proposed bound can be computed by sampling from the output of a parameterized model. Using this practical interpretation, we introduce the Bayes optimal learning threshold (BOLT) loss whose minimization enforces a classification model to achieve the Bayes error rate. We validate the proposed loss for image and text classification tasks, considering MNIST, Fashion-MNIST, CIFAR-10, and IMDb datasets. Numerical experiments demonstrate that models trained with BOLT achieve performance on par with or exceeding that of cross-entropy, particularly on challenging datasets. This highlights the potential of BOLT in improving generalization.
Explore related subjects
Keep this discovery
Mohammadreza Tavasoli Naeini, Ali Bereyhi, Morteza Noshad, Ben Liang, Alfred O. Hero III. 2025-01-13. Universal Training of Neural Networks to Achieve Bayes Optimal Classification Accuracy. https://arxiv.org/abs/2501.07754
Cite the original work for its findings. Save a collection to share your selection of sources.