草稿链代替思维链,推理token砍掉80%,显著降低算力成本和延迟

梦晨 发自 凹非寺
量子位 | 公众号 QbitAI

推理token减少80%-90%,准确率变化不大,某些任务还能增加。

Zoom团队提出思维链替代品“草稿链”,显著降低延迟和算力成本。

d1b54f03ddaec8bbf90ac7e03916f453.png

原理很简单,要求模型为每个推理步骤生成简洁、信息密集的token。

1306e1ee789235699ccfb440dda1b127.png

这一思路受到人类解题过程启发,人类通常不会详细阐述每一个细节,只是简单几下关键的中间结果,作为草稿来辅助思考。

此外,草稿链方法简单且易于实现,不需要修改模型、微调或强化学习,只需更新提示词中的示例即可,相关代码和数据已在GitHub上开源。

a0052a8f65d27f6466894a161aed2865.png

研究团队认为,与另一种降低延迟和计算成本的方法“在连续潜空间推理”相比,草稿链保留了可解释性,且可以应用于闭源的黑盒模型。

第三方分析测算,对于每个月处理100万个推理请求的企业, 可以将成本从思维链的3800美元降低到760美元,每月节省超过3000美元。

f147dd415821116d73e1d10832eafd0c.png

实验验证:准确率不降,成本大幅降低

实验遵循原始思维链论文,评估3类任务:算术推理、常识推理和符号推理。

算数推理任务选择GSM8k数据集,从准确率看,标准提示下GPT-4o和Claude 3.5 Sonnet的准确率分别仅53.3%和64.6%,思维链使两者均超95%,草稿链也达到91%左右。

在token使用上,思维链生成约200个token/响应,草稿链仅约40个,减少约80%。

延迟方面,草稿链使GPT-4o和Claude 3.5 Sonnet的平均延迟分别降低76.2%和48.4%。

680203f1ba0b90af5e2b008116e107da.png

常识推理选择BIG-bench 的日期理解和运动理解任务也取得了相似的结果,特别值得注意的是Claude 3.5 Sonnet在草稿链下准确率还有所上升。

e8358b46e1676a8cae264dc07e215f2b.png

b5b8f0bd6b279abe789d0941724b43a4.png

符号推理选择抛硬币任务,由于原数据集未公开,作者合成了250个示例的测试集。
草稿链在保持100%准确率的情况下大幅减少了推理token和延迟。

d32a845b1cb26a5bd417920428575456.png

但草稿链方法也有一定的局限性:在零样本设置下有效性会显著下降。

907e34294b93dcb61f70af1d4383994c.png

此外,在参数量小于30亿的小模型上,虽然草稿链仍能有效减少每个回答所需的token数并提高准确率,但与思维链相比,性能差距更大。

6d56d2bf6ac0c2e5bcdeec19b453803c.png

研究团队推测,可能是由于训练数据中缺乏草稿链风格的推理模式,在不提供few-shot样本指导的情况下,很难生成简洁而有帮助的”草稿”。

论文地址:
https://arxiv.org/abs/2502.18600v2

参考链接:
[1]https://ajithp.com/2025/03/02/chain-of-draft-llm-prompting/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值