ScoreStop: Gradient-based early stopping using functional score tests
About
Gradient boosted decision trees require a stopping rule to avoid overfitting. The standard rule monitors a validation loss and stops if the loss fails to improve for a fixed patience period. However, the patience parameter has no interpretable scale and validation losses can be noisy or implicitly defined by a user-specified gradient. We propose ScoreStop, a gradient-based early-stopping rule that casts the stopping decision at each iteration as a test of the null hypothesis that the current predictor is the population risk minimizer. We use a functional score test, computed on validation data, with a statistic that is scale-invariant in the update direction, with a known asymptotic distribution under the null. Because our test uses gradients rather than loss values, the same construction applies to implicit losses such as LambdaRank, and data-dependent losses such as Cox regression via influence functions. In synthetic experiments and real-data benchmarks, we show that ScoreStop is competitive with loss-based methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Offline Learning to Rank | MSLR-WEB10K (test) | -- | 44 | |
| Ranking | Synthetic | Median Excess Test Loss0.0152 | 21 | |
| Classification | Synthetic | Median Excess Test Loss0.0625 | 21 | |
| Regression | Synthetic | Median Excess Test Loss0.0477 | 21 | |
| Survival | Synthetic | Median Excess Test Loss0.0632 | 21 | |
| Regression | Abalone (test) | L2 Risk0.0049 | 19 | |
| Survival Analysis | FLCHAIN (test) | -- | 18 | |
| Classification | Phoneme (test) | Median Excess Test Loss0.0046 | 5 | |
| Poisson Regression | Bike Sharing (test) | Median Excess Test Loss0.2621 | 5 | |
| Survival Analysis | SUPPORT (test) | Median Excess Test Loss0.0064 | 5 |