English

Copy-as-Decode: Grammar-Constrained Parallel Prefill for LLM Editing

Computation and Language 2026-05-26 v2 Artificial Intelligence

Abstract

LLMs edit text and code by autoregressively regenerating the full output, even when most tokens appear verbatim in the input. We study Copy-as-Decode, a decoding-layer mechanism that recasts edit generation as structured decoding over a two-primitive grammar: <copy lines="i-j"/> references an input line range, <gen>...</gen> emits new content. A token-level FSM guarantees syntactic validity, and a serving-layer primitive updates the KV cache for each copy span via a single parallel-prefill forward rather than NN autoregressive steps -- sharing the parallel-forward kernel of speculative decoding but with input tokens as the draft and program-enforced acceptance replacing probabilistic verification. We report an upper-bound analysis that requires no end-to-end training. (i) Kernel speedup: on Qwen2.5-{1.5B, 7B}, copying NN tokens via parallel prefill is 6.8×6.8\times--303×303\times faster than autoregressive (N[8,512]N \in [8, 512], A100 80GB bf16). (ii) Copy ceiling: on ProbeEdit and HumanEvalPack-Fix (Py/JS), 7474--98%98\% of gold tokens are reachable under the line-level primitive; composed with the empirical kernel over each corpus's span histogram this yields a closed-form wall-clock bound of 29.0×/3.4×/4.2×29.0\times / 3.4\times / 4.2\times (13.0×13.0\times pooled). A token-level extension reaches 9191--99%99\% coverage with 4.5×4.5\times--6.5×6.5\times floors. (iii) Pipeline losslessness: oracle programs round-trip through the deterministic resolver on all 482482 cases, localizing any downstream failure to span selection rather than the mechanism. A perturbation study shows pooled EM drops from 100%100\% to 15.48%15.48\% under off-by-one noise. A fine-tuning pilot on Qwen2.5-Coder-1.5B lifts HEvalFix-Py EM from 0/330/33 (untrained) to 1212--17%17\%, a learnability signal, not a production selector. Batched-serving integration and multi-file coverage are scoped as follow-up.

Keywords

Cite

@article{arxiv.2604.18170,
  title  = {Copy-as-Decode: Grammar-Constrained Parallel Prefill for LLM Editing},
  author = {Ziyang Liu},
  journal= {arXiv preprint arXiv:2604.18170},
  year   = {2026}
}

Comments

The authors have decided to withdraw this version following internal review regarding authorship and contribution agreements

R2 v1 2026-07-01T12:18:14.102Z