Generalization bounds for regression and classification on adaptive covering input domains
Wen-Liang Hwang
TL;DR
The paper develops generalization bounds for regression and classification on adaptively covering input domains using a local geometry parameter $\gamma_s$. By linking $\gamma_s$ to network capacity through a refinement oracle, it shows bounds scale with $(K_f+K_{\mathcal{M}})\gamma_s$ for regression and $ (\gamma_s)^{d-1}\mathrm{vol}B_{d-1}(\mathbf{0},1)|\partial f|$ for classification, and proves that $\gamma_s$ decreases polynomially with the number of network parameters. It establishes concentration-based sample complexities for both tasks under fixed-radius deterministic and random ball coverings, yielding $m_0$ bounds and explicit dependence on dimension $d$ and boundary size. An oracle implementation via deep nets links the covering radius to parameter count and demonstrates how to halve the radius through hierarchical partitions, supporting benign overfitting in over-parameterized models. The results highlight that classification can require fewer samples than regression under these bounds and provide insights into inductive biases, network design, and potential extensions to generative modeling.
Abstract
Our main focus is on the generalization bound, which serves as an upper limit for the generalization error. Our analysis delves into regression and classification tasks separately to ensure a thorough examination. We assume the target function is real-valued and Lipschitz continuous for regression tasks. We use the 2-norm and a root-mean-square-error (RMSE) variant to measure the disparities between predictions and actual values. In the case of classification tasks, we treat the target function as a one-hot classifier, representing a piece-wise constant function, and employ 0/1 loss for error measurement. Our analysis underscores the differing sample complexity required to achieve a concentration inequality of generalization bounds, highlighting the variation in learning efficiency for regression and classification tasks. Furthermore, we demonstrate that the generalization bounds for regression and classification functions are inversely proportional to a polynomial of the number of parameters in a network, with the degree depending on the hypothesis class and the network architecture. These findings emphasize the advantages of over-parameterized networks and elucidate the conditions for benign overfitting in such systems.
