River-LLM: Large Language Model Seamless Exit Based on KV Share
River-LLM:基于KV共享的大型语言模型无缝退出
机构 * Shanghai Jiao Tong University(上海交通大学)
专题命中 效率与部署 :LLM(title,title_cn);large language model(title,abstract);language model(title,abstract);分类 cs.CL
AI总结 针对解码器-only架构中早期退出因KV缓存缺失导致的延迟和精度问题,提出无需训练的River-LLM框架,通过轻量级KV共享退出流实现令牌级无缝退出,并利用状态转移相似性预测KV误差指导退出决策,在数学推理和代码生成任务上获得1.53-2.16倍实际加速且保持高质量。
Comments Accepted to ACL 2026, 13pages, with appendix. Corrected some typos