Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2026
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2604.12426 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911592034074624 |
|---|---|
| author | Curth, Alicia Lawrence, Rachel Karmalkar, Sushrut Prasad, Niranjani |
| author_facet | Curth, Alicia Lawrence, Rachel Karmalkar, Sushrut Prasad, Niranjani |
| contents | We investigate whether transformers use their depth adaptively across tasks of increasing difficulty. Using a controlled multi-hop relational reasoning task based on family stories, where difficulty is determined by the number of relationship hops that must be composed, we monitor (i) how predictions evolve across layers via early readouts (the logit lens) and (ii) how task-relevant information is integrated across tokens via causal patching. For pretrained models, we find some limited evidence for adaptive depth use: some larger models need fewer layers to arrive at plausible answers for easier tasks, and models generally use more layers to integrate information across tokens as chain length increases. For models finetuned on the task, we find clearer and more consistent evidence of adaptive depth use, with the effect being stronger for less constrained finetuning regimes that do not preserve general language modeling abilities. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2604_12426 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Do Transformers Use their Depth Adaptively? Evidence from a Relational Reasoning Task Curth, Alicia Lawrence, Rachel Karmalkar, Sushrut Prasad, Niranjani Machine Learning Computation and Language We investigate whether transformers use their depth adaptively across tasks of increasing difficulty. Using a controlled multi-hop relational reasoning task based on family stories, where difficulty is determined by the number of relationship hops that must be composed, we monitor (i) how predictions evolve across layers via early readouts (the logit lens) and (ii) how task-relevant information is integrated across tokens via causal patching. For pretrained models, we find some limited evidence for adaptive depth use: some larger models need fewer layers to arrive at plausible answers for easier tasks, and models generally use more layers to integrate information across tokens as chain length increases. For models finetuned on the task, we find clearer and more consistent evidence of adaptive depth use, with the effect being stronger for less constrained finetuning regimes that do not preserve general language modeling abilities. |
| title | Do Transformers Use their Depth Adaptively? Evidence from a Relational Reasoning Task |
| topic | Machine Learning Computation and Language |
| url | https://arxiv.org/abs/2604.12426 |