English

Hierarchical vs. Flat Iteration in Shared-Weight Transformers

Computation and Language 2026-04-17 v1 Artificial Intelligence

Abstract

We present an empirical study of whether hierarchically structured, shared-weight recurrence can match the representational quality of independent-layer stacking in a Transformer-based language model. HRM-LM replaces L independent Transformer layers with a two-speed recurrent pair: a Fast module operating at every step for local refinement, and a Slow module operating every T steps for global compression. This recurrent hierarchy is unrolled for M = N x T steps with shared parameters. The central and most robust finding, supported by a parameter-matched Universal Transformer ablation (UniTF, 1.2B) across five independent runs, is a sharp empirical gap between the two approaches.

Keywords

Cite

@article{arxiv.2604.14442,
  title  = {Hierarchical vs. Flat Iteration in Shared-Weight Transformers},
  author = {Sang-Il Han},
  journal= {arXiv preprint arXiv:2604.14442},
  year   = {2026}
}
R2 v1 2026-07-01T12:11:43.549Z