Deep Learning meets Nonparametric Regression: Are Weight-Decayed DNNs Locally Adaptive?
Kaiqi Zhang, Yu-Xiang Wang
TL;DR
The paper tackles adaptive function estimation in nonparametric regression under heterogeneous smoothness by analyzing parallel ReLU neural networks trained with standard weight decay. It proves that weight decay induces an effective $\ell_{2/L}$ sparsity on the learned dictionary, making deep parallel networks behave like sparse regression with representation learning. The main result shows that for targets in Besov or BV classes, the estimator can achieve near-minimax rates, with the error approaching the optimal rate as depth increases. This provides a theoretical explanation for the empirical depth advantage of DNNs and suggests a practical pathway to adaptivity across diverse smoothness regimes without tuning architecture to specific function spaces. The work bridges neural network theory with classical nonparametric estimation, highlighting both the synthesis (basis construction) and analysis (regularization) perspectives in a unified framework.
Abstract
We study the theory of neural network (NN) from the lens of classical nonparametric regression problems with a focus on NN's ability to adaptively estimate functions with heterogeneous smoothness -- a property of functions in Besov or Bounded Variation (BV) classes. Existing work on this problem requires tuning the NN architecture based on the function spaces and sample size. We consider a "Parallel NN" variant of deep ReLU networks and show that the standard $\ell_2$ regularization is equivalent to promoting the $\ell_p$-sparsity ($0<p<1$) in the coefficient vector of an end-to-end learned function bases, i.e., a dictionary. Using this equivalence, we further establish that by tuning only the regularization factor, such parallel NN achieves an estimation error arbitrarily close to the minimax rates for both the Besov and BV classes. Notably, it gets exponentially closer to minimax optimal as the NN gets deeper. Our research sheds new lights on why depth matters and how NNs are more powerful than kernel methods.
