跳到正文
原文
arXiv:cs.LG(机器学习,全量分类)· Zhiyuan Ma·· 17 小时前AI 评分34

LumoTree:面向混合语言模型的路径并行推测验证

LumoTree: Path-Parallel Speculative Verification for Hybrid Language Models

AI 导读

LumoTree 是一种面向混合语言模型的树形推测解码验证器,通过并行执行循环路径、复用路径内状态块,并在共享逻辑树中协调循环重放、卷积历史收集与注意力缓存重映射。

正文

View PDF HTML (experimental)

Abstract:Tree speculative decoding for hybrid language models must preserve one coherent continuation across recurrent, convolution, and attention state. We present LumoTree, a verifier that executes recurrent paths in parallel, reuses state tiles within each path, and coordinates native recurrent replay, convolution-history gathering, and attention-cache remapping through a shared logical tree. Fused candidate selection, GPU-resident acceptance, and grouped split-K attention support the verification cycle. Component experiments show exact candidate-selection parity and recurrent agreement within paired error bounds. An exploratory Qwen3.8-27B NVFP4 deployment on a single NVIDIA DGX Spark (GB10) records 25.63 pooled tokens/s on ten SWE-bench Verified Astropy tasks. The results characterize component-level numerical agreement and coding-agent deployment; complete-model continuation and controlled application speedups remain open.
Comments: 11 pages, 3 figures
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2609.23900 [cs.LG]
  (or arXiv:2609.23900v2 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2609.23900

arXiv-issued DOI via DataCite

Submission history

From: Zhiyuan Ma [view email]
[v1] Sun, 20 Sep 2026 22:21:01 UTC (88 KB)
[v2] Thu, 1 Oct 2026 01:53:32 UTC (96 KB)

来源:arXiv:cs.LG(机器学习,全量分类) · arxiv.org