跳到正文
原文
arXiv cs.CL· Mingyu Zhao, Shuhang Lin, Ligong Han, Xiaojing Yu, Zhenting Wang, Yifei Pang, Yang Zhou, Dimitris N. Metaxas·· 2 小时前AI 评分35

元认知解码:面向扩散语言模型智能体的高效推理

Metacognitive Decoding: Efficient Inference for Diffusion Language Agents

AI 导读

Metacognitive Decoding 用一个轻量元认知控制器引导扩散语言模型的生成,学习何时定稿 token、生成应延伸多远,以减少单次响应内的整体计算。该控制器作为扩散主干网络的插件,在智能体轨迹上训练以平衡正确性与生成成本。在五个 benchmark 上,相比最高效基线平均减少 20.6% 推理 FLOPs,同时保持有竞争力的任务准确率。

来源:arXiv cs.CL · arxiv.org