PulseAugur
实时 12:14:24

Two gradient steps enhance feature learning in linear-width networks

This paper investigates feature learning in two-layer neural networks with a linear width, examining the impact of two gradient descent steps compared to one. The research provides a detailed spectral characterization of updated weights, revealing they form a spiked random matrix with multiple learned directions. It highlights that reusing batches allows for capturing directions beyond a single information exponent, a benefit that extends to high-dimensional limits. AI

影响 Provides a mathematical framework for understanding optimization and feature learning in overparameterized networks.

排序理由 Academic paper published on arXiv detailing theoretical advancements in neural network feature learning.

在 arXiv stat.ML 阅读 →

AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →

Two gradient steps enhance feature learning in linear-width networks

报道来源 [2]

  1. arXiv stat.ML TIER_1 English(EN) · Behrad Moniri, Hamed Hassani ·

    Feature Learning in Linear-Width Two-Layer Networks: Two vs. One Step of Gradient Descent

    arXiv:2605.17767v1 Announce Type: new Abstract: We study feature learning in two-layer neural networks within the linear-width regime, where the number of hidden neurons, sample size, and input dimension scale proportionally. While recent work has analyzed feature learning via a …

  2. arXiv stat.ML TIER_1 English(EN) · Hamed Hassani ·

    Feature Learning in Linear-Width Two-Layer Networks: Two vs. One Step of Gradient Descent

    We study feature learning in two-layer neural networks within the linear-width regime, where the number of hidden neurons, sample size, and input dimension scale proportionally. While recent work has analyzed feature learning via a single step of gradient descent, such updates ar…