Adaptive Regularization for Random Features: A Neighboring Early-Stopping Rule with Oracle-Rate Guarantees
Authors: Caixing Wang, Zhibo Chen, Yue Wang
Organizations: School of Statistics and Data Science, Southeast University · Faculty of Business for Science and Technology, School of Management, University of Science and Technology of China
Random feature methods provide a scalable approximation to kernel ridge regression (KRR), but the regularization parameter that yields the oracle learning rate depends on unknown smoothness and capacity parameters. In this work, we propose a neighboring early-stopping rule for adaptive regularization in KRR with random features (KRR-RF). The method uses a grid that is uniform in inverse regularization and compares only adjacent estimators, reducing the number of discrepancy comparisons relative to standard all-pairs Lepskii-type procedures. Both the neighboring discrepancy and its empirical complexity term can be computed directly in the random feature space, without constructing the exact kernel Gram matrix. We establish a high-probability comparison bound for neighboring KRR-RF estimators and show that, under standard source and capacity conditions together with suitable grid and random feature budget conditions, the selected estimator attains the oracle polynomial learning rate up to logarithmic factors. The result allows the regularization parameter to be selected without prior knowledge of the source and capacity exponents and covers both well-specified and partially misspecified regimes. Our analysis is based on an empirical random feature effective dimension that connects the observable stopping threshold with the population complexity of the random feature model. Simulation and real-data experiments illustrate the prediction performance and computational behavior of the proposed method in comparison with standard tuning procedures.
Figures & tables
Figure 1: Left: The results for KRR-RF over 16 candidate values of λ . The blue curve is the test RMSE on an independent test set after training on the full sample and its minimizer is the oracle choice. The red dashed curve is the 5-fold cross-validation score, obtained by refitting the model on training folds and evaluating it on validation folds. Cross-validation selects a smaller λ and incurs a larger test error than the oracle choice. Right: averaged over 12 trials, the oracle path benchmark requires 16 solves, while 5-fold cross-validation requires 80 solves. Despite this 5× higher cost, cross-validation still gives a worse average test RMSE (0.694 ± 0.013 vs. 0.677 ± 0.005).
Kmax=⌊4κ2h∣D∣⌋,δD=5Kmaxδ,λk=hk1.
Algorithm 1 Neighboring Early-Stopping Rule for KRR-RF (NESR-KRR-RF)
Method
Main computational cost
Memory cost
LP
O(K∣D∣3+K2∣D∣2)
O(∣D∣2)
ASUS
O(K∣D∣3+K∣D∣2)
O(∣D∣2)
LP-RF
O(∣D∣M2+KM3+K2M2)
O(∣D∣M+M2)
NESR-KRR-RF
O(∣D∣M2+KM3+KM2)
O(∣D∣M+M2)
Table 1: Computational complexity of different adaptive regularization methods.
q
δ
CLP
CES
h
KES
M
∣D∣
∣D′∣
Fig. 2 :(a)
0.5
0.01
0.1
0.005
50
20
100
5000
1000
Fig. 2 :(b)
0.5
0.01
0.05
0.003
50
20
100
5000
1000
Fig. 2 :(c)
0.5
0.01
0.0007
0.01
50
20
100
5000
1000
Fig. 2 :(d)
0.5
0.01
0.002
0.002
200
100
100
5000
1000
Fig. 3 :(a)
0.5
0.01
0.1
0.005
50
20
100
-
1000
Fig. 3 :(b)
0.5
0.01
0.05
0.003
50
20
100
-
1000
Table 2: Parameter settings for NESR and LP.
Figure 2: Curve fitting performance with different r and γ . In each panel, the solid black line represents the true function f(x) , and the blue scatter points show the noisy observations f(x)+ε . Among the four compared estimators, the purple dotted line indicates the Oracle benchmark, and the orange two-dash line corresponds to Cross-Validation (CV). Additionally, the red dot-dash line denotes Lepskii Principle (LP), while the green dashed line illustrates our proposed NESR.
Figure 3: RMSE vs number of training data ∣D∣ for four approaches with different r and γ .
Figure 4: Run time vs number of training data ∣D∣ for NESR and LP with different r and γ .
Figure 5: RMSE vs number of random features M for four approaches with different r and γ .
Figure 6: Run time vs number of random features M for NESR and LP with different r and γ .
Figure 7: RMSE vs subdivision factor h for NESR with different r and γ .
Figure 8: log MSE vs log number of training data ∣D∣ for NESR with different r and γ .
datasets
d
∣D′∣
∣D∣
τ
M
CES
CLP
Fig. 9 :(a)
KIN8NM
8
1192
-
2
1000
0.0250
0.0001
Fig. 9 :(b)
SPCD
81
3263
-
8
1000
0.0400
0.0003
Fig. 9 :(c)
PPOPTS
9
5730
-
1
1000
0.0800
0.0010
Fig. 9 :(d)
SUSY
18
2000
-
22.5
1000
0.0750
0.0001
Fig. 9 :(e)
HTRU2
8
3808
-
2
1000
0.0200
0.0010
Fig. 9 :(f)
MGT
10
3020
-
21.5
1000
0.0500
0.0030
Table 3: Parameter settings and dataset characteristics.
Figure 9: Prediction accuracy and runtime vs the training sample size ∣D∣ on real datasets. Top panel: Regression tasks, where the line plots represent the MSE and the bar plots represent the average runtime over 10 independent trials. Bottom panel: Classification tasks, where the line plots represent the classification error rate and the bar plots represent the average runtime over 10 independent trials.
Figure 10: Prediction accuracy and runtime vs the number of random features M on real datasets. Top panel: Regression tasks, where the line plots represent the MSE and the bar plots represent the average runtime over 10 independent trials. Bottom panel: Classification tasks, where the line plots represent the classification error rate and the bar plots represent the average runtime over 10 independent trials.
Random features provide a standard approach to scaling kernel methods to large datasets by approximating the kernel with a finite-dimensional feature map. Existing works of random feature methods primarily focus on well-specified models and light-tailed noise conditions, leaving a gap in understanding their performance under more general conditions, such as misspecified models and heavy-tailed noise. In this paper, we provide a comprehensive and improved analysis of kernel ridge regression with random features (KRR-RF) under general conditions. In the light-tailed noise setting, we improve upon the worst-case uniform bounds of the feature map used in the previous works and lead to the minimax optimal learning rates with 2r+γ>α0, where (r,γ,α0) denote the regularity of the target function, the effective dimension of the kernel, and the embedding index of the RKHS, respectively. In the heavy-tailed noise setting, we leverage a recently developed Hilbert-valued Fuk-Nagaev inequality and extend the same optimal results under finite weak moment conditions. The additional cost to achieve robustness is a mildly stronger explicit sample-size requirement than in the light-tailed case. For both settings, our results recover the conditions random features budgets (up to log factors) required in the previous works. In the most benign case when α0=γ, KRR-RF are minimax optimal for all r∈(0,1]. To the best of our knowledge, this is the first work to establish optimal learning rates for KRR-RF under weak moment conditions in the misspecified setting. Numerical experiments and a real data analysis validate the derived learning rates and demonstrate their practical relevance in the presence of mis-specification and heavy-tailed noise.
Caixing Wang
School of Statistics and Data Science, Southeast University, Nanjing, China
We study data-driven early stopping for spectral regularisation methods in the classical non-parametric regression setting. Building on the discrepancy principle, we propose a multi-scale stopping rule that applies to general kernel estimators and show that, unlike previous approaches, it achieves full adaptivity over all smoothness levels in the well-specified case. A key contribution of our work is an extension based on random feature approximations, which reduces computational cost on large datasets while preserving minimax-optimal statistical guarantees. Our procedure not only selects an optimal stopping time but also provides a fully data-driven choice of the number of random features needed to achieve optimal rates. Through the established connection between random features and neural networks in the neural tangent kernel regime, our method further yields a principled, data-driven recommendation for the network width. We prove that the resulting simultaneously chosen width and stopping time allow neural networks to attain minimax-optimal learning rates without prior knowledge of smoothness or capacity parameters.
We study fixed-design random feature ridge regression beyond the mean-kernel approximation. The expectation is taken over the frozen-feature ensemble, conditional on the training sample. Because the predictor is a nonlinear function of the empirical kernel, its mean training error, test error, and conditional generalization gap depend on centered kernel covariances as well as on mean kernel objects. We derive the covariance-level, or one-loop, corrections from a finite resolvent identity. This avoids an almost-sure Neumann-series assumption and yields an explicit remainder bound for the training error. The test correction additionally requires mixed train--test covariance tensors and therefore cannot, in general, be recovered from the train-restricted kernel law alone. Numerical experiments show that the covariance term removes the leading inverse-width discrepancy in a controlled regularization regime, while an ablation that discards the mixed train--test covariance fails for the test error even though the training correction is unchanged. They also identify a width--regularization boundary, measured by a resolvent-fluctuation diagnostic, beyond which the second-order truncation loses accuracy. All inverse-width statements are for fixed sample size, depth, design, and positive kernel-level regularization. The analysis concerns frozen features and does not model feature learning or neural-tangent-kernel evolution.
Taeyoung Kim
Center for AI and Natural Sciences, Korea Institute for Advanced Study, Seoul 02455, South Korea