arXiv:2604.21254cs.LGcs.CL2026-04被引 8

用循环变压器结构减少参数量,提升模型内存效率。

Hyperloop Transformers

论文配图:Hyperloop Transformers
图 1 · 摘自论文原文
  • 采用循环Transformer层复用机制,仅中间块重复使用。
  • 相比标准Transformer,参数减少约50%仍保持良好性能。
  • 适合边缘设备等内存受限场景的高效语言建模应用。

大语言模型研究通常在固定计算/延迟预算下追求模型质量最大化。然而,许多实际应用如边缘和设备端部署还受制于模型内存占用,因而需要参数高效的架构。本文提出一种简单架构,通过循环Transformer作为核心单元,复用深层中的Transformer层,从而比普通深度匹配的Transformer更具参数效率。该架构分为三个模块——起始、中间和结束块,其中仅中间块沿深度方向循环使用。我们在中间块引入超连接(hyper-connections),将残差流扩展为矩阵形式,且仅在每轮循环后应用,新增参数和计算开销极小。在多种模型规模下,我们的超连接循环变压器(Hyperloop Transformer)在参数量约为基准模型50%的情况下,表现优于深度匹配Transformer和mHC Transformer基线,且在权重量化后性能依然稳定,展现出优异的内存效率,适用于内存受限的语言建模任务。

原文摘要 · Abstract (English)

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 perform well compared to depth-matched Transformer and mHC Transformer baselines despite using approximately 50% fewer parameters. This performance persists through post-training weight quantization, thus positioning Hyperloop Transformers as an attractive architecture for memory-efficient language modeling.

参数效率循环结构内存优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。