atlas

Research map / Optimization / Implicit bias and generalization

Improving Adaptivity via Over-Parameterization in Sequence Models

NeurIPS 2024 · Optimization › Implicit bias and generalization · approach: Low-rank and single-index recovery · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that eigenvalue ordering matters for kernel regression and proposes over-parameterized (learnable-eigenvalue) gradient flow in a Gaussian sequence model, proving adaptive near-optimal rates with a universal stopping time and benefits of depth.

Problem

Kernel regression with a fixed kernel performs poorly when the ordering of kernel eigenvalues is misaligned with the coefficients of the target function in the eigenbasis, so the best kernel depends on unknown signal structure.

Approach

Replace the fixed square-root eigenvalues in a diagonalized kernel gradient flow with trainable per-component parameters (theta_j = a_j b_j^D beta_j), initialized at the eigenvalues, so that gradient descent adapts the effective eigenvalues to the signal; deeper factors further refine this adaptation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
learn eigenvalues with over-parameterized gradient descent instead of using fixed eigenvaluesSequence model · Sequence model (p=1, q=2)convergence rate: 0.33 -> 0.5Helped
add one extra parameterization layer (D=1) (ablation)OpGD · Sequence model (p=1, q=2, gamma=1.1)convergence rate: 0.48 -> 0.49Helped
add one extra parameterization layer (D=1) (ablation)OpGD · Sequence model (p=1, q=2, gamma=2)convergence rate: 0.50 -> 0.50No clear effect
add one extra parameterization layer (D=1) (ablation)OpGD · Sequence model (p=1, q=2, gamma=3)convergence rate: 0.50 -> 0.44Hurt
add two extra parameterization layers (D=2) (ablation)OpGD · Sequence model (p=1, q=2, gamma=1.1)convergence rate: 0.48 -> 0.48No clear effect
add two extra parameterization layers (D=2) (ablation)OpGD · Sequence model (p=1, q=2, gamma=2)convergence rate: 0.50 -> 0.49Hurt
add two extra parameterization layers (D=2) (ablation)OpGD · Sequence model (p=1, q=2, gamma=3)convergence rate: 0.50 -> 0.50No clear effect
increase parameterization depth (ablation)Over-parameterized gradient descent · Sequence model, gamma=1.1convergence rate: 0.38 -> 0.34Mixed

Limitations

Experimental setup

Datasets: California Housing Concrete Compressive Strength

Benchmarks: California Housing Concrete Compressive Strength

Metrics: generalization error convergence rate exponent

Baselines: vanilla gradient flow fixed-eigenvalue kernel gradient descent

Builds on: neural tangent kernel diagonal linear networks implicit regularization Mercer decomposition

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy