BOLT: Universal Classifier Training via a Bayes-Error Upper Bound
Organizations: University of Michigan · University of Toronto · Stanford University
Abstract
This work introduces a novel approach to compute a universal upper bound on the Bayes error rate (BER) from classifier-output samples, which serves as a Bayes-error-related performance criterion for the underlying classification task. We show that the bound is tight in the binary case and characterize the max-sum gap that appears in the aggregate multiclass formulation. Based on this bound, we define a new loss function, called Bayes optimal learning threshold (BOLT), for training parameterized classifiers. For each data sample, the BOLT loss computes a term whose expectation upper-bounds the Bayes error, so that during training the model parameters can be updated toward minimizing this bound. Numerical experiments with both synthetic and real-world datasets show that, in the reported settings, BOLT training often attains lower held-out classification error than cross-entropy training.
Figures & tables
| Method | CIFAR-10 | MNIST | F-MNIST | IMDb | SST-2 |
|---|---|---|---|---|---|
| LDA | 42.0 | 87.9 | 80.4 | 86.8 | – |
| RF | 48.0 | 95.6 | 88.0 | 80.3 | – |
| CE | 91.9 | 99.3 | 91.6 | 91.8 | 89.9 |
| BOLT | 93.3 | 99.5 | 92.6 | 92.8 | 91.3 |
| Dataset | Method | FLOPs/image (G) | Time/epoch (s) | Executed epochs | Total time (s) |
|---|---|---|---|---|---|
| CIFAR-10 | ResNet-18 + CE | 0.041 | 12.0 | 100 | |
| CIFAR-10 | ResNet-18 + BOLT | 0.041 | 13.0 | 100 | |
| MNIST | CNN + CE | – | 3.6 | 20 | |
| MNIST | CNN + BOLT | – | 4.0 | 20 | |
| MNIST | -ball | – | – | one shot | 210 |
| Checkpoint | Sparsity (%) | BOLT (%) | CE (%) |
|---|---|---|---|
| Epoch 10 | 20.0 | 85.23 | 91.29 |
| Epoch 50 | 48.8 | 92.74 | 92.47 |
Multiclass Classification
Multiclass classification assigns inputs to one of several labels while addressing uncertainty, class imbalance, and errors across classes.
See all 95 papers