Papers
arxiv:2608.15062

Gated Recurrent Transformers: Expressive Depth through Recurrent Modulation

Published on Aug 26
· Submitted by
Amr Hegazy
on Aug 27
Authors:
,

Abstract

A gated recurrent transformer reuses a shared core across depth with adaptive update gates, achieving comparable or better quality than deeper models with far fewer parameters and lower memory.

Scaling transformer language models creates an inherent tension between expressivity and memory efficiency. While unique weights across layers preserve functional specialization---from input-grounding to abstract refinement---they incur a substantial memory footprint. Conversely, standard depth-sharing enforces uniform transformations that collapse representational diversity and degrade modeling quality. We introduce Gated Recurrent Transformer, a recurrent depth transformer where fixed-depth prelude and coda blocks bracket a single shared core iterated R times. Inspired by gated recurrent neural networks, we employ a lightweight projection and an elementwise update gate---conditioned on the hidden state, the fixed prelude output, and noise resampled at every step---to modulate the recurrent update. This allows the model to specialize the input to the same few layers across recurrences, rather than requiring many unique layers to achieve functional diversity. Under an isoFLOPS constraint, a 3-layer Gated Recurrent Transformer matches the accuracy of a 12-layer GPT-2 Small baseline with similar training and inference FLOPs, and leads MoR and heavy-tail depth sampling in all nine scale-by-budget cells; at medium and large scale it approaches dense quality at the standard token budget and overtakes it at medium scale once that budget is doubled. Under an isoPARAMS constraint, deeper recurrence achieves a 2.76 validation loss versus 2.84 for a non-recurrent counterpart at matched parameter and data budget. Our results demonstrate that adaptive depth reuse is a principled strategy for trading parameters for quality: at large scale, 63% fewer parameters and 59% less peak decoding memory for a 10% increase in compiled generation latency.

Community

Hi, I really liked this paper and all the ablation studies you did, congrats!
Do you plan to share the repo you used to train this model and the checkpoints? I wanted to check how does EBT (Energy based transformer) compared with the RecurrentGPT.

·
Paper author

Hi, Thank you for your interest in our paper! We actually just released the training code: https://github.com/Amr-Hegazy1/gated-recurrent-transformer

Paper author Paper submitter
This comment has been hidden (marked as Off-Topic)
Paper author Paper submitter
edited about 19 hours ago

Can a transformer gain expressive depth without adding unique layers?

We introduce Gated Recurrent Transformers, which repeatedly apply a small shared core while using a lightweight elementwise update gate to modulate how representations evolve across recurrences. This lets the same parameters specialize across computation steps rather than performing an identical transformation each time.

Under isoFLOPs, a 3-layer model matches a 12-layer GPT-2 Small baseline, while at larger scale we retain competitive quality with substantially fewer parameters and lower decoding memory. We also find that the recurrent design continues to benefit strongly from additional training data.

data_scaling-1

Because the model is trained across different recurrence depths, it also supports a natural early-exit tradeoff at inference time where fewer recurrent steps can be used when lower compute is preferred, while additional steps recover more of the model's full performance.

isoflops (2)-1

This is an automated message from the Librarian Bot. I found the following papers similar to this paper.

The following papers were recommended by the Semantic Scholar API

Please give a thumbs up to this comment if you found it helpful!

If you want recommendations for any Paper on Hugging Face checkout this Space

You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2608.15062
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2608.15062 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2608.15062 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2608.15062 in a Space README.md to link it from this page.

Collections including this paper 3