Linear-Core Surrogates: Smooth Loss Functions with Linear Rates for Classification and Structured Prediction
Abstract
Lay Summary
When training machine learning models, the choice of loss function, which measures how wrong predictions are penalized, involves a fundamental trade-off. Smooth loss functions (whose curves have no sharp corners) let optimizers converge quickly, but provide weak guarantees on how well surrogate performance translates to actual accuracy. Non-smooth losses (with sharp corners) give much stronger guarantees, but make optimization harder and slower. We introduce Linear-Core Surrogates, a new family of loss functions that resolve this trade-off. The key idea is to stitch a straight-line segment in the critical region near the decision boundary to smooth curved tails farther away. This gives the best of both worlds: the optimization speed of smooth losses and the strong accuracy guarantees of non-smooth ones. To our knowledge, these are the first explicitly defined loss functions to achieve both properties simultaneously. We prove these guarantees for binary classification, multi-class classification, and structured prediction (e.g., sequence tagging). In practice, our loss functions improve robustness to mislabeled data, outperforming the standard Cross-Entropy loss by 2.6% on corrupted image classification, and enable a 23x training speedup on large-vocabulary sequence tagging by replacing expensive exact inference with simple random sampling.