Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Oliver J. Hines, Christian L. Hines• 2026

Related benchmarks

TaskDatasetResultRank
Offline Learning to RankMSLR-WEB10K (test)--
44
RankingSynthetic
Median Excess Test Loss0.0152
21
ClassificationSynthetic
Median Excess Test Loss0.0625
21
RegressionSynthetic
Median Excess Test Loss0.0477
21
SurvivalSynthetic
Median Excess Test Loss0.0632
21
RegressionAbalone (test)
L2 Risk0.0049
19
Survival AnalysisFLCHAIN (test)--
18
ClassificationPhoneme (test)
Median Excess Test Loss0.0046
5
Poisson RegressionBike Sharing (test)
Median Excess Test Loss0.2621
5
Survival AnalysisSUPPORT (test)
Median Excess Test Loss0.0064
5
Showing 10 of 15 rows

Other info

Follow for update