RidgeFill:免训练跳算上层预填充,TTFT 提速至 1.6 倍、64K 请求 82.6s→47.5s

Public_Umpire_1099 · reddit · 2026-11-03

Reddit 上发布的论文 RidgeFill:一种完全免训练的方法,通过不为 prompt 计算模型大部分上层来缩短 prefill。低层精确运行到切分点;从该层状态出发,用一次闭式拟合的岭回归预测其后各层的输入,各晚层以自身投影不变的方式从预测值写入缓存。选定的行、最后几千 token 和每个解码步仍走完整模型——不训练、不丢 token。

实测数据:

方法论脉络:想法源自 DeepSeek V4.1 架构——该模型整体训练时即内建 prefill 近似,误差被模型吸收;作者尝试移植到其他模型时发现上下文召回明显下降,于是转向定位与测量「填充区」的误差。他们否决了更强训练拟合组件的路线(会变得任务依赖、损害泛化),改为直接绕过 prefill 最敏感部分的投影,使社区可用本地推理级硬件在不同模型上自行复现。作者强烈建议按模型逐个调参,论文含详尽消融。

已提供 llama.cpp fork、HF 上的五个模型映射、tcclaviger 的 vLLM fork 及 Deadcode 的 Radiance 引擎(v1.3.0)实现。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →