arXiv cs.LG· Dongsen Zhang, Peipei Li, Zekun Li, Wenjun Xu·· 6 小时前AI 评分27
KITE:面向任务的关键层 KV 通信实现高效潜在多智能体协作
Task-Oriented Key-Layer KV Communication for Efficient Latent Multi-Agent Collaboration
AI 导读
针对现有 KV 潜在通信方法偏重发送端状态保真、通信与计算开销大的问题,研究者提出免训练框架 KITE,改用接收端任务充分性目标,通过接收端轨迹失真准则定位任务有效关键层,仅传输该层的潜在工作记忆并以其作为自回归潜在推理入口。在两类模型家族、三种模型规模、七个基准上,KITE 相比全层 KV 通信将通信量降低 28-36 倍,端到端推理最高提速 3 倍,准确率最高提升 23.3 个百分点。
来源:arXiv cs.LG · arxiv.org