arXiv:2405.17573v3 Announce Type: replace-cross
Abstract: We study Leaky ResNets, which interpolate between ResNets and Fully-Connected nets depending on an ‘effective depth’ hyper-parameter $tildeL$. In the infinite depth limit, we study ‘representation geodesics’ $A_p$: continuous paths in representation space (similar to NeuralODEs) from input $p=0$ to output $p=1$ that minimize the parameter norm of the network. We give a Lagrangian and Hamiltonian reformulation, which highlight the importance of two terms: a kinetic energy which favors small layer derivatives $partial_pA_p$ and a potential energy that favors low-dimensional representations, as measured by the ‘Cost of Identity’. The balance between these two forces offers an intuitive understanding of feature learning in ResNets. We leverage this intuition to explain the emergence of a bottleneck structure, as observed in previous work: for large $tildeL$ the potential energy dominates and leads to a separation of timescales, where the representation jumps rapidly from the high dimensional inputs to a low-dimensional representation, move slowly inside the space of low-dimensional representations, before jumping back to the potentially high-dimensional outputs. Inspired by this phenomenon, we train with an adaptive layer step-size to adapt to the separation of timescales.
A woman’s uterus has been kept alive outside the body for the first time
“Think of this as a human body,” says Javier González. In front of me is essentially a metal box on wheels. Standing at around a