Variable-Width Transformers
该工作质疑 Transformer 各层等宽 (constant width) 均摊参数与算力的惯例,提出在深度上做非均匀容量分配的沙漏形架构:两端层更宽、中间层更窄,用无参数的 residual resizing 衔接不同宽度。在 200M–2B dense 与 3B MoE 上,相比参数匹配的等宽基线,语言建模困惑度相对改善约 3%;在 loss-matched 缩放曲线下省 22% FLOPs、降低 15% KV-cache 显存与 I/O,平均层宽缩小 10–11%。
INSIGHT“不同层承担不同计算角色,因此不该等宽”把参数预算从均摊转向按需分配,且 residual resizing 不引入额外参数。对基建的直接价值是 15% 的 KV-cache 显存/IO 下降——这对在线 serving 的吞吐与显存成本是实打实收益;中间层收窄也说明深层网络中段存在冗余容量,可作为压缩/蒸馏的切入点。