2024 ICML ICML 2024

A Theory of Non-Linear Feature Learning with One Gradient Step in Two-Layer Neural Networks