Hyperloop Transformers

LLM PTSAD MQIA
大语言模型(LLM)架构研究通常旨在固定计算量或延迟预算的前提下,最大化模型性能。然而,许多实际应用场景——例如边缘设备与终端设备上的部署——还额外受到模型内存占用的严格限制,因此亟需参数更高效的语言建模架构。本文提出一种简洁的架构设计,显著提升了大语言模型的参数利用效率。该架构以“循环式Transformer”(looped Transformer)为核心基本单元:它通过在深度方向上复用同一组Transformer层,从而相比同等深度的传统Transformer大幅减少参数量。我们将循环式Transformer划分为三个模块——起始模块(begin block)、中间模块(middle block)和终止模块(end block),其中每个模块本身均由多个Transformer层构成,而仅中间模块在深度方向上被循环应用。此外,我们为中间循环模块引入了“超连接”(hyper-connections,Xie等,2026),将残差流从标量或向量形式扩展为矩阵形式的残差流。超连接仅在每次循环迭代结束后施加,因此新增的参数量与计算开销均极小。在多种模型规模下,我们的“超连接循环式Transformer”(Hyperloop Transformer)均显著优于参数量相当(depth-matched)的传统Transformer基线及mHC Transformer基线,同时参数总量减少约50%。值得注意的是,这一性能优势在模型后训练权重量化(post-training weight quantization)之后依然保持稳定,因而Hyperloop Transformer成为面向内存受限场景的语言建模任务极具吸引力的架构选择。
LLM architecture research generally aims to maximize model quality subject to fixed compute/latency budgets. However, many applications of interest such as edge and on-device deployment are further constrained by the model's memory footprint, thus motivating parameter-efficient architectures for language modeling. This paper describes a simple architecture that improves the parameter-efficiency of LLMs. Our architecture makes use of looped Transformers as a core primitive, which reuse Transformer layers across depth and are thus more parameter-efficient than ordinary (depth-matched) Transformers. We organize the looped Transformer into three blocks--begin, middle, and end blocks--where each block itself consists of multiple Transformer layers, and only the middle block is applied recurrently across depth. We augment the looped middle block with hyper-connections (Xie et al., 2026), which expand the residual stream into matrix-valued residual streams. Hyper-connections are applied only after each loop, and therefore add minimal new parameters and compute cost. Across various model scales, we find that our Hyper-Connected Looped Transformer (Hyperloop Transformer) is able to outperform depth-matched Transformer and mHC Transformer baselines despite using approximately 50% fewer parameters. The outperformance persists through post-training weight quantization, thus positioning Hyperloop Transformers as an attractive architecture for memory-efficient language modeling.
许愿