Files
myWiki/concepts/prefix-survival-probability.md

1.3 KiB
Raw Permalink Blame History

title, created, updated, type, tags, sources
title created updated type tags sources
Prefix Survival Probability 2026-06-28 2026-06-28 concept
speculative-decoding
probability
acceptance-metric
DSpark

Prefix Survival Probability

前缀存活概率是投机解码中衡量草稿前缀被目标模型接受的可能性的核心度量。对于草稿位置 $j$,其前缀存活概率定义为累积条件接受率的乘积:

a_{r,j} = \prod_{i \le j} c_{r,i}

其中 c_{r,i} = P(\text{accept}_i | \text{accept}_{1:i-1}) 是位置 i 的条件存活概率。

性质

单调非增$a_{r,j} \le a_{r,j-1}$,即越靠后的 token 存活概率越低。这一性质是 DSparkhardware-aware-prefix-scheduler贪心算法正确性的关键保证。

在调度中的应用

调度器将每个请求 r 在位置 j 的期望收益直接建模为 $a_{r,j}$——即接纳该 token 所带来的额外期望接受数。全局排序所有 a_{r,j} 后按降序接纳,自然遵守块内前缀依赖(因为 a_{r,j} 的单调性保证 $a_{r,j} \le a_{r,j-1}$)。

参考