Rethinking Benign Overfitting in Two-Layer Neural Networks

TL;DR

This paper refines the feature-noise data model to include class-dependent noise, revealing that neural networks can utilize data noise to improve classification in long-tailed distributions, supported by theoretical analysis and experiments.

Contribution

It introduces a refined heterogenous noise model and analyzes neural network training dynamics, showing how data noise can enhance generalization in long-tailed data.

Findings

Neural networks leverage data noise to improve classification accuracy.

Test loss bounds are established for the refined noise model.

Experimental results validate the theoretical insights.

Abstract

Recent theoretical studies (Kou et al., 2023; Cao et al., 2022) have revealed a sharp phase transition from benign to harmful overfitting when the noise-to-feature ratio exceeds a threshold-a situation common in long-tailed data distributions where atypical data is prevalent. However, harmful overfitting rarely happens in overparameterized neural networks. Further experimental results suggested that memorization is necessary for achieving near-optimal generalization error in long-tailed data distributions (Feldman & Zhang, 2020). We argue that this discrepancy between theoretical predictions and empirical observations arises because previous feature-noise data models overlook the heterogeneous nature of noise across different data classes. In this paper, we refine the feature-noise data model by incorporating class-dependent heterogeneous noise and re-examine the overfitting phenomenon in neural networks. Through a comprehensive analysis of the training dynamics, we establish test loss bounds for the refined model. Our findings reveal that neural networks can leverage "data noise" to learn implicit features that improve the classification accuracy for long-tailed data. Our analysis also provides a training-free metric for evaluating data influence on test performance. Experimental validation on both synthetic and real-world datasets supports our theoretical results.