2018

Small ReLU networks are powerful memorizers: a tight analysis of memorization capacity

Yun, Chulhee, Sra, Suvrit, Jadbabaie, Ali

Understand

We study finite sample expressivity, i.e., memorization power of ReLU networks.

  • Recent results require $N$ hidden nodes to memorize/interpolate arbitrary $N$ data points.
  • In contrast, by exploiting depth, we show that 3-layer ReLU networks with $\Omega(\sqrt{N})$ hidden nodes can perfectly memorize most datasets with $N$ points.
  • We also prove that width $\Theta(\sqrt{N})$ is necessary and sufficient for memorizing $N$ data points, proving tight bounds on memorization capacity.

Reading the bibliography…