PreSTO 预测式子树预取将 LLM Power Sampling 提速至多 2.33 倍

du_yilun · x · 2026-10-12

Nan Jiang 等人发布 arXiv 论文 PreSTO(PreSTO: Predictive Subtree Prefetching for Fast LLM Power Sampling),在不动模型和采样器语义的前提下大幅加速 power sampling——一种在测试期通过从锐化分布采样提升 LLM 推理能力、无需额外训练的方法。

核心问题:power sampling 的采样器 PowerMH 依赖 Metropolis–Hastings 更新,接受/拒绝决策是串行的,每一步状态决定下一步提案来源,导致很慢。

方法:

结果:跨 7 个数据集、5 个基础 LLM,PreSTO 对 PowerMH、EntropyCut、MultiTryMH 分别取得 1.71×、1.91×、2.33× 中位加速;对前两者模型调用次数中位减少 3.9×,单次调用仅长 2.1×。

所属事件:PreSTO 预测式子树预取为 LLM Power Sampling 提速至 2.33 倍(2 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →