AI Paper Atlas

Research map / Optimization / Implicit bias and generalization

In-context learning as gradient descent

Shows transformers implement gradient-descent-like algorithms for in-context linear regression.

41 accepted papers at ICML, NeurIPS, ICLR, CVPR and AAAI (2017–2026) · an approach in Implicit bias and generalization, keyword cluster “relu · width · kernel”

State History Tried Preprints Who Export

Where it stands

Not enough benchmark data Fewer than four papers report the same benchmark, so there is no record to follow.

17 papers in 2023–24 and 16 in 2025–26 (at ICML, NeurIPS, ICLR, CVPR and AAAI); its share of the topic shrank, from 7.2% to 6.1%.

How this is decided

A benchmark is tracked when at least four papers of this approach report their own method on the same dataset and metric in at least two different years. Its record is the best number reported up to 2024 compared with the best reported in 2025–26: still improving means at least 1% better, flat means at most 0.2% better with at least two papers from 2025–26 trying. Numbers on a different scale from the rest (a fraction among percentages) and far outliers are left out. Results come from different papers with different setups, so they are signals, not head-to-head comparisons. The share of papers is counted at ICML, NeurIPS, ICLR, CVPR and AAAI.

How it got here

2016: 0162017: 1172018: 1182019: 2192020: 3202021: 0212022: 1222023: 5232024: 12242025: 7252026: 926

Most cited paper of each year

Earliest papers here

What has been tried

28 papers here report 216 measured changes (a component added, removed or swapped, compared with a baseline): helped 60%, no clear effect 15%, hurt 20%, mixed 5%. Papers mostly publish changes that helped, so "helped" is over-represented.

Ablations in recent papers

What was changedMeasured onResultEffectPaper
remove L2 regularizer from shared-layer transformersLinear regression (LR) · test query MSEPrefixLM-SL: 8.6e-3 -> 6.7e-3; CausalLM-SL: 1.9e-1 -> 5.0e-2MixedCausalLM is not optimal for in-context learning (ICLR 2024 · 2 citations)
Use finite rather than infinite samples per training taskNew-task test tasks, mtest=20, 1-GD · excess risk1.78 ± 0.02 -> 0.81 ± 0.01HelpedFirst-order ANIL provably learns representations despite overparametrisation (ICLR 2024 · 1 citation)
Remove gate regularization and the gate–weight timescale differenceAlternating two-task curriculum · task-switch adaptation speedHurtFlexible task abstractions emerge in linear networks with fast and bounded units (NeurIPS 2024 · 1 citation)
Increase hidden size to 32 or 64In-context linear regression · ability to mimic OLSHelpedTransformers Learn to Achieve Second-Order Convergence Rates for In-Context Linear Regression (NeurIPS 2024 · 1 citation)
add the MLP componentLinear regression, β* = (0,0,...,0)^⊤ · loss0.024 -> 0.003HelpedIn-Context Learning of a Linear Transformer Block: Benefits of the MLP Component and One-Step GD Initialization (NeurIPS 2024 · 1 citation)
Use rank 2 for separate key and query weightsIn-context linear regression, D=8 · plateau patternMixedTraining Dynamics of In-Context Learning in Linear Attention (ICML 2025 · 0 citations)
add the input-dependent selection mechanism to S4In-context linear regression · ICL capabilityHelpedTrained Mamba Emulates Online Gradient Descent in In-Context Linear Regression (NeurIPS 2025 · 0 citations)
Remove the loss term corresponding to R_*Modular addition · hidden nodes per frequency required for a global solution6 to 3HelpedComposing Global Solutions to Reasoning Tasks via Algebraic Objects in Neural Nets (NeurIPS 2025 · 0 citations)
Increase context length to L = 10Synthetic data, context length L = 10 · attention concentration on noise patchesNo clear effectOn the Optimization and Generalization of Two-layer Transformers with Sign Gradient Descent (ICLR 2025 · 0 citations)
Restrict sparse-vector support to the first 10 entriesIn-context sparse recovery, support restricted to first 10 entries · label prediction lossHelpedOn the Learn-to-Optimize Capabilities of Transformers in In-Context Sparse Recovery (ICLR 2025 · 0 citations)
Mask non-diagonal gradients of W^KQ_32 and W^PV_12 during trainingAR process, fixed all-ones initial token · vanilla gradient-descent mesa-optimizationã b̃ = 1/(1 + (d-1)/(T-2) ∑ (t=2)^{T-1} 1/(t-1))HelpedOn Mesa-Optimization in Autoregressively Trained Transformers: Emergence and Capability (NeurIPS 2024 · 0 citations)
Constrain attention matrices to be diagonalUniform σ_τ∼U(0,5) · adjusted evaluation loss0.047 -> 0.047No clear effectLinear Transformers are Versatile In-Context Learners (NeurIPS 2024 · 0 citations)

Recent preprints

Loading recent arXiv preprints…

    Who publishes on it

    Institutions on the most papers of this approach since 2023 (of 33 papers; from OpenAlex and the papers' affiliations).

    Use it

    Download BibTeX of 30 key papersOpen the topic in the interactive map

    Get alerts. Describe your idea in scime atlas and save it (free scime account): you get a weekly email when new preprints land close to it.

    Same keyword cluster

    Elsewhere in the topic

    Benchmark numbers, ablations and baseline numbers are extracted by AI from each paper's full text and can be wrong; the paper is the authority. Data: results index of 2026-11-02, citation counts from OpenAlex (2026-10-09).

    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