本文へスキップ
原文
arXiv 机器学习· Aaryam Sharma·· 9 時間前AI スコア38

TVP:推测解码中 TV 几乎就是全部所需

TVP: TV is Almost All You Need for Speculative Decoding

未翻訳のため中国語で表示しています。

AI ガイド

研究人员提出针对推测解码的 TVP 损失函数族,基于期望接受长度与全变差(TV)距离优化草稿模型。针对 TV 梯度消失等问题,TVP 通过幂参数在覆盖率与接受率目标间插值,在 3 种草稿架构和 3 种目标模型上均取得最高或并列最高接受长度。微调时 TVP 相比 CE 将接受长度提升 9.8%,较 DFlash2 提升 6.2% 至 12.9%。

出典:arXiv 机器学习 · arxiv.org