A Simple Generalisation of the Implicit Dynamics of In-Context Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Innocenti, Francesco, Achour, El Mehdi
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909957619712000
author Innocenti, Francesco
Achour, El Mehdi
author_facet Innocenti, Francesco
Achour, El Mehdi
contents In-context learning (ICL) refers to the ability of a model to learn new tasks from examples in its input without any parameter updates. In contrast to previous theories of ICL relying on toy models and data settings, recently it has been shown that an abstraction of a transformer block can be seen as implicitly updating the weights of its feedforward network according to the context (Dherin et al., 2025). Here, we provide a simple generalisation of this result for (i) all sequence positions beyond the last, (ii) any transformer block beyond the first, and (iii) more realistic residual blocks including layer normalisation. We empirically verify our theory on simple in-context linear regression tasks and investigate the relationship between the implicit updates related to different tokens within and between blocks. These results help to bring the theory of Dherin et al. (2025) even closer to practice, with potential for validation on large-scale models.
format Preprint
id arxiv_https___arxiv_org_abs_2512_11255
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Simple Generalisation of the Implicit Dynamics of In-Context Learning
Innocenti, Francesco
Achour, El Mehdi
Machine Learning
Artificial Intelligence
I.2.6
In-context learning (ICL) refers to the ability of a model to learn new tasks from examples in its input without any parameter updates. In contrast to previous theories of ICL relying on toy models and data settings, recently it has been shown that an abstraction of a transformer block can be seen as implicitly updating the weights of its feedforward network according to the context (Dherin et al., 2025). Here, we provide a simple generalisation of this result for (i) all sequence positions beyond the last, (ii) any transformer block beyond the first, and (iii) more realistic residual blocks including layer normalisation. We empirically verify our theory on simple in-context linear regression tasks and investigate the relationship between the implicit updates related to different tokens within and between blocks. These results help to bring the theory of Dherin et al. (2025) even closer to practice, with potential for validation on large-scale models.
title A Simple Generalisation of the Implicit Dynamics of In-Context Learning
topic Machine Learning
Artificial Intelligence
I.2.6
url https://arxiv.org/abs/2512.11255