Lune

ICLR2021Top-tier venue

Why Are Convolutional Nets More Sample-Efficient than Fully-Connected Nets?

Zhiyuan Li, Yi Zhang, Sanjeev Arora

2021Year
22Citations
15Top-tier citations

Abstract

Convolutional neural networks often dominate fully-connected counterparts in generalization performance, especially on image classification tasks. This is often explained in terms of 'better inductive bias'. However, this has not been made mathematically rigorous, and the hurdle is that the fully connected net can always simulate the convolutional net (for a fixed task). Thus the training algorithm plays a role. The current work describes a natural task on which a provable sample complexity gap can be shown, for standard training algorithms. We construct a single natural distribution on Rd×{±1}\mathbb{R}^d\times\{\pm 1\} on which any orthogonal-invariant algorithm (i.e. fully-connected networks trained with most gradient-based methods from gaussian initialization) requires Ω(d2)Ω(d^2) samples to generalize while O(1)O(1) samples suffice for convolutional architectures. Furthermore, we demonstrate a single target function, learning which on all possible distributions leads to an O(1)O(1) vs Ω(d2/ε)Ω(d^2/\varepsilon) gap. The proof relies on the fact that SGD on fully-connected network is orthogonal equivariant. Similar results are achieved for ℓ2\ell_2 regression and adaptive training algorithms, e.g. Adam and AdaGrad, which are only permutation equivariant.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext fd514b2f-623e-402d-9d0d-fcfbe220a9a8

Cited by top-tier papers15

Ask how each one uses it

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines