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

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

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

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

图片

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

图片

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

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

图片

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

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

图片

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

实验遵循原始思维链论文,评估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%。

图片

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

图片

图片

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

图片

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

图片

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

图片

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

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

责任编辑:张燕妮 来源: 量子位
相关推荐

2025-03-24 13:45:56

2025-03-13 10:40:00

2024-03-20 11:07:57

AI计算CPUGPU

2011-06-10 13:21:09

SEO垃圾链接

2022-03-08 14:31:37

区块链汽车技术

2024-12-18 14:53:28

2023-01-05 21:25:06

毫末

2020-02-02 19:00:28

区块链供应链区块链技术

2025-02-08 17:00:11

2024-11-11 11:05:00

大语言模型系统

2025-03-17 08:15:00

AI技术模型

2023-10-16 21:52:01

人工智能人工智能算力

2024-08-12 09:40:00

模型数据

2024-05-14 09:22:51

模型技术

2023-06-05 10:01:18

模型测评

2024-01-08 12:51:34

AI数据

2018-12-17 14:00:44

公有链联盟链私有链

2023-01-04 20:00:02

SparseGPT降低算力成本

2025-03-12 09:48:19

点赞
收藏

51CTO技术栈公众号