Poster
Matching the Statistical Query Lower Bound for -Sparse Parity Problems with Sign Stochastic Gradient Descent
Yiwen Kou · Zixiang Chen · Quanquan Gu · Sham Kakade
West Ballroom A-D #7107
Abstract:
The -sparse parity problem is a classical problem in computational complexity and algorithmic theory, serving as a key benchmark for understanding computational classes. In this paper, we solve the -sparse parity problem with sign stochastic gradient descent, a variant of stochastic gradient descent (SGD) on two-layer fully-connected neural networks. We demonstrate that this approach can efficiently solve the -sparse parity problem on a -dimensional hypercube () with a sample complexity of using neurons, matching the established lower bounds of Statistical Query (SQ) models. Our theoretical analysis begins by constructing a good neural network capable of correctly solving the -parity problem. We then demonstrate how a trained neural network with sign SGD can effectively approximate this good network, solving the -parity problem with small statistical errors. To the best of our knowledge, this is the first result that matches the SQ lower bound for solving -sparse parity problem using gradient-based methods.
Chat is not available.