MACRO: Markov Chain Routing of Transformer Layers¶
复现级别:核心机制复现。 论文的中心算子在本地真实执行;生产私有数据、大模型权重或专用服务未复刻,论文结果与本地结果严格分开。
论文信息¶
| 项目 | 内容 |
|---|---|
| 论文链接 | arXiv 2608.05872 |
| 公司/机构 | Heinrich Heine University Düsseldorf |
| 首次公开日期 | 2026-08-06(arXiv v1) |
| 原文开源代码 | 是:官方/作者代码 |
| Adapter | macro |
| 本地复现代码 | src/auto_research/reproductions/macro/ |
原始论文总结¶
背景与主要改动¶
主题:动态层路由。 固定顺序执行所有 Transformer 层并非总是最优。MACRO 用上下文条件 Markov policy 表示 skip、repeat、residual-add 等操作,再用反馈更新路由分布和 top-k Viterbi 解码候选程序,不修改底座权重。
主要架构¶
flowchart LR
N0["任务上下文"] --> N1
N1["Markov 路由策略"] --> N2
N2["skip / repeat / add"] --> N3
N3["top-k Viterbi"] --> N4
N4["冻结 LLM"]
原论文关键图¶
原论文 Figure 2(关键图):展示原论文的整体流程、关键阶段及其数据流向。图片来自原论文,版权归原作者所有;点击图片可查看来源。
核心公式¶
\(p(\rho\mid x)=\prod_t\pi(a_t\mid \ell_t,b_t,d_t,x)\)
论文离线与线上效果¶
平均准确率相对顺序执行 +5.0%,相对 Dr. LLM +7.2 个点;搜索时间由 14.8 小时降至 1.6 小时(9.4×)。
本地复现¶
在确定性分类 mini-suite 搜索 81 条 skip/repeat/residual 路由;同时作为 micro-LLM evolve 的可选结构。
运行:
稳定指标保存在 metrics/public-seed42.json,不提交 checkpoint。
本地对照口径:基线为去掉论文特有机制、其余数据切分与预算相同的 matched control;实验组为
macro核心机制;相对变化见public-seed42.json;跨论文百分比不适用。
复现边界¶
- 本地结果用于验证机制能执行和比较方向,不等价于原论文规模复现。
- 私有特征、线上流量和生产 serving 不可获得;原文线上数值只作为引用。
- 可接入 evolve 的结构已注册为候选;只影响 serving 的系统方法保留为独立可执行 adapter,不冒充可训练 genome。
