Library of Layers (LoL):推理时动态排序 Transformer 层
摘要
论文提出了 Library of Layers (LoL),一种在推理时动态重排其层的 Transformer 架构,统一了稠密(dense)、深度递归(depth-recurrent)与 MoE 等基线在参数复用与稀疏性之间的权衡,并在推理、多语言和安全任务上实现了可解释的路由。
查看缓存全文
缓存时间: 2026/11/03 07:34
论文页面 - Library of Layers (LoL):运行时自适应排序 Transformer 层
来源:https://huggingface.co/papers/2611.03526
摘要
Library of Layers 在推理过程中动态排列 Transformer 块的顺序,使计算能够根据任务自适应调整,从而实现参数复用、稀疏化以及可解释的路由,适用于推理、多语言和安全等多种场景。
计算机程序员将相同的可复用计算原语(例如条件语句或循环)以不同顺序链接,从而实现不同的算法。相比之下,现有的语言模型架构始终以相同顺序执行其各层。我们提出 Library of Layers(https://huggingface.co/papers?q=Library%20of%20Layers)(简称 LoL),一种在推理过程中动态排列层顺序以解决当前任务的架构。LoL 仅有一个超参数,即其全局 transformer 块(https://huggingface.co/papers?q=transformer%20blocks)库的大小,这使它能够在参数复用(https://huggingface.co/papers?q=parameter%20reuse)与稀疏化(https://huggingface.co/papers?q=sparsity)之间无缝过渡,将稠密(dense)、深度循环(depth-recurrent,https://huggingface.co/papers?q=depth-recurrent)以及混合专家(mixture of experts,https://huggingface.co/papers?q=mixture%20of%20experts)架构在参数效率与 FLOPs 效率之间的权衡统一到一个通用框架中。我们表明,在文本数据上预训练后,LoL 能够与上述三种基线模型相媲美。在合成数独和迷宫数据上,我们观察到模型采用不同的层顺序来实现不同的求解策略,并且我们可以干预路由(https://huggingface.co/papers?q=routing),强制模型对特定问题实例应用特定的求解策略。在多语言自然语言场景中,我们发现 LoL 会根据语言和文字系统选择不同的层顺序模式。类似地,我们在安全数据上训练 LoL 并对其路由器输出(https://huggingface.co/papers?q=router%20outputs)拟合探针,发现这些探针能够以很高的 AUROC 区分安全与不安全的文本。LoL 为模型动态组合自身计算图的未来提供了路径,从而实现更高效、更可控的推理。
查看 arXiv 页面 (https://arxiv.org/abs/2611.03526)|查看 PDF (https://arxiv.org/pdf/2611.03526)|GitHub (https://github.com/arahmani1/library-of-layers)|加入收藏 (https://huggingface.co/login?next=%2Fpapers%2F2611.03526)
在你的 agent 中获取这篇论文:
hf papers read 2611.03526
没有最新的 CLI?curl -LsSf https://hf.co/cli/install.sh | bash
引用该论文的模型 0
没有模型关联此论文。
在模型 README.md 中引用 arxiv.org/abs/2611.03526 即可从本页面链接。
引用该论文的数据集 0
没有数据集关联此论文。
在数据集 README.md 中引用 arxiv.org/abs/2611.03526 即可从本页面链接。
引用该论文的 Spaces 0
没有 Space 关联此论文。
在 Space README.md 中引用 arxiv.org/abs/2611.03526 即可从本页面链接。
收录该论文的合集 0
没有合集收录此论文。
将此论文加入合集(https://huggingface.co/new-collection)即可从本页面链接。
相似文章
LayerRoute: 一种通过LoRA保持质量的自适应层跳过方法,用于高效LLM推理
LayerRoute引入了一种参数高效的自适应transformer层跳过方法,结合了逐层路由与联合LoRA微调,在LLM推理中实现了经过验证的速度提升和质量改进。
跳层还是循环?学习LLM中的Program-of-Layers
本文介绍了PoLar,一个学习针对冻结transformer层的输入特定执行程序的框架,允许层被跳过、保留或重复。与固定深度方法相比,它提高了准确性并减少了推理开销。
LayerRoute:基于输入条件的自适应层跳跃方法——通过LoRA微调实现代理语言模型优化
LayerRoute是一种轻量级适配器,能够根据输入类型在推理过程中选择性跳过Transformer块,通过门控路由和LoRA自适应实现计算节省,同时保持或提升模型质量。在代理语言模型上,它实现了12.91%的跳跃差异。
内存高效型循环Transformer:循环语言模型中的计算与内存解耦
提出内存高效型循环Transformer(MELT),这是一种新型循环大语言模型架构,通过跨循环共享单一KV缓存,并结合插值过渡与注意力对齐蒸馏的分块训练方法,实现了推理深度与内存消耗的解耦。
Aletheia:基于梯度引导的层选择方法,实现跨架构的高效LoRA微调
Aletheia 提出了一种基于梯度引导的层选择方法,用于高效的 LoRA 微调。该方法通过轻量级梯度探针识别与任务相关的 Transformer 层,并选择性地应用适配器,在 14 个模型上实现了 15%-28% 的训练加速,同时保持了在 MMLU、GSM8K 和 HumanEval 基准测试中的下游性能。