Beyond KV Reconstruction: Functional Reconstruction for MLA Draft Models in Speculative Decoding · Research brief

Request deep report Read original paper ↗View page data (JSON)Download research skill ↓Discuss this paper

For solo builders

A possible application for this research area. Paper-specific feasibility still needs a full-paper review.

Your self-hosted model spends too long generating answers token by token.

What you would change

Check whether a compatible draft model and serving integration exist before renting compute.

Try the idea on your product

A proposed first test. Set your budget and success criteria before running it.

Check whether a compatible draft model and serving integration exist before renting compute.

Compare against

The same target with speculation off, then your existing drafter if you have one.

Locate the paper's implementation and check its license, model support, data needs, and hardware requirements. Stop here if you cannot obtain the required artifacts.
Save a small set of representative product tasks and expected outcomes. Keep evaluation tasks out of any training or tuning, and choose quality and cost limits before running.
Run the baseline and the proposed change on identical inputs. Keep other settings fixed and repeat paired runs to expose variation.
Save per-task outputs, failures, timings, and all usage costs. Review regressions before expanding the test.

What to measure

Accepted draft length, tokens per second, tail latency, output quality, and total GPU cost.

When to keep it

Keep the change only if it meets your preselected quality and latency limits and improves the metric you care about after setup and operating costs. Otherwise retain the baseline. This is an exploration plan, not a validated recommendation.

Download test plan ↓

abstract indexedPaperraft next step

Do not allocate compute yet.

This page uses the abstract and public adoption signal only. Request the deep report before relying on its recommendation.

Abstract-level screening only. Risks and infrastructure may not have been evaluated. This recommendation is not a reproduced result.

minimum test setup: not reported
published evidence setup: not reported
source-linked claims: 0 of 2
exposure checks: 0 of 8

Why it was shortlisted

The technical change, research area, and reason it entered the archive.

recommendation: test
research area: speculative decoding
technical change: Functional reconstruction of MLA draft models for speculative decoding
published: 2026-07-29

The technique addresses a real acceptance-rate regression in converted MLA drafts and is feasible on a single GPU, but the gain is inconsistent across configurations, so the reader should validate it on their specific draft-target pair and backend before production adoption.

Central equations

Original equations, typeset for reading. Paperraft explanations connect the mechanism, its aim, and the evidence.

For layer $ℓ$ and calibration minibatch $b$ , let $H_{b}^{ℓ} \in ℝ^{T_{b} \times D}$ contain the cached layer inputs and let $M_{b} \in {0, 1}^{T_{b}}$ mark valid tokens, broadcast over the hidden dimension. The implementation optimizes each layer independently with

ℒ_{E2E}^{(ℓ)} = 𝔼_{b} ​ [\frac{{‖ M_{b} ⊙ (A_{MLA}^{ℓ} ​ (H_{b}^{ℓ}) - sg [A_{orig}^{ℓ} ​ (H_{b}^{ℓ})]) ‖}_{F}^{2}}{D ​ \sum_{t} M_{b, t}}] .

equation (11) · §Method · loss function

Speculative decoding makes this residual operationally visible. At a fixed context $x$ , write $p_{D}^{x} (y) = p_{D} (y ∣ x)$ and similarly for $p_{T}^{x}$ . Stochastic rejection-sampling verification has one-token expected acceptance (Leviathan et al. 2023; Chen et al. 2023)

α_{sample} ​ (x) = \sum_{y} \min (p_{D}^{x} ​ (y), p_{T}^{x} ​ (y))

= 1 - TV (p_{D}^{x}, p_{T}^{x}) .

equation (9) · §Preliminary: Why Converted MLA Is Not Draft-Preserving · metric

Our experiments instead use greedy decoding, whose one-token acceptance event is

α_{greedy} (x) = [\arg \max_{y} p_{D}^{x} (y) = \arg \max_{y} p_{T}^{x} (y)] .

equation (10) · §Preliminary: Why Converted MLA Is Not Draft-Preserving · metric

equations from arXiv HTML · fetched 2026-09-15

Claims and evidence

A claim is source-linked only when the PDF page and matching excerpt are available.

claim 01Paperraft inference
Read the brief interpretation
The method replaces direct low-rank MHA/GQA-to-MLA conversion for draft models with a post-conversion calibration procedure that fine-tunes each converted MLA attention module to reproduce the original module's post-output-projection output on calibration hidden states, without verifier logits or supervision. Its cost is an additional per-module optimization stage after conversion, while the converted MLA cache layout and inference graph are preserved, so no extra serving-time memory or latency is introduced. It can fail when acceptance gains do not materialize (26 of 64 cells unchanged, one degraded), when calibration hidden states are unrepresentative of the production workload, or when the reader's stack uses MHA/GQA drafts directly and the MLA conversion step itself is not warranted.
This is a Paperraft inference. It is not pinned to a verified PDF passage.
Inspect the source
claim 02Paperraft inference
Across 64 matched task cells (four Llama/Qwen draft-target pairs, TransMLA and MHA2MLA converters, HF and vLLM backends, four 200-prompt tasks), functional reconstruction materially improves draft-token acceptance in 37 cells, leaves 26 unchanged within a 0.5-percentage-point tolerance, and decreases one; no aggregate speedup factor is reported.
This is a Paperraft inference. It is not pinned to a verified PDF passage.
Inspect the source

Exposure map

Author findings, Paperraft interpretations, and the next check. Missing analysis never counts as evidence of safety.

quality

Still unknown

Which tasks improve, and which regress against the baseline?

compute and memory

Still unknown

What memory, hardware, and total compute does the complete method require?

latency

Still unknown

Are startup time, typical response time, and slow requests measured separately?

operations

Still unknown

What must be monitored, maintained, and rolled back?

software compatibility

Still unknown

Does the implementation support your model, runtime, and hardware?

security

Still unknown

What new data access, permissions, or trust boundaries does it introduce?

data and training

Still unknown

What data preparation or training is required, and is it included in the cost?

reproducibility

Still unknown

Are code, versions, inputs, and evaluation scripts available?

Continue the investigation

A reading guide for speculative decoding.

Editorial prompts, not findings from this paper or a validated test plan. Detailed analysis is still pending for the topics below.

What to inspect

How are proposals verified, and what happens after rejection? Check the assumptions behind any quality guarantee.

Plan a fair comparison

Can you compare the same target model with speculation off and on, including proposal and verification time?

Questions for adoption

Which draft model or prediction heads are needed, and how does acceptance change with the workload?

Read the paper with these questions

Inside the paper

Preview the first 3 pages of the original PDF. Click the stack to turn the page.

The downloadable skill includes SKILL.md and evidence.json, with recorded findings and unanswered checks. TRY-IT.md adds a product test plan and a blank results sheet.

A reading aid for your agent. Findings still need validation.

Page 1 of 3

Open this page in the PDF ↗

Provisional research brief updated 2026-10-18. Paperraft has not reproduced this experiment.