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.