pith. sign in

arxiv: 2601.02780 · v2 · submitted 2026-01-06 · 💻 cs.CL · cs.AI

MiMo-V2-Flash Technical Report

Pith reviewed 2026-05-12 11:27 UTC · model grok-4.3

classification 💻 cs.CL cs.AI
keywords Mixture of Expertsmulti-token predictionspeculative decodingdistillationhybrid attentionlanguage modelparameter efficiencyopen-source model
0
0 comments X

The pith

MiMo-V2-Flash matches top open-weight models like DeepSeek-V3.2 using half their total parameters via sparse MoE design.

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The paper presents MiMo-V2-Flash as a Mixture-of-Experts model with 309 billion total parameters but only 15 billion active during use. It pre-trains on 27 trillion tokens with multi-token prediction and a hybrid attention setup that alternates sliding windows with global attention, then extends context from 32k to 256k. A new Multi-Teacher On-Policy Distillation method lets specialized teachers supply token-level guidance so the student fully acquires their capabilities. This enables the model to rival larger systems while delivering inference speedups by reusing its prediction layers as a draft model for speculative decoding.

Core claim

MiMo-V2-Flash is a Mixture-of-Experts model with 309B total parameters and 15B active parameters that rivals top-tier open-weight models such as DeepSeek-V3.2 and Kimi-K2 despite using only half or one-third their total parameters. It adopts a hybrid attention architecture interleaving Sliding Window Attention with global attention under a 5:1 ratio and a 128-token window, pre-trains with Multi-Token Prediction on 27 trillion tokens, and introduces Multi-Teacher On-Policy Distillation where domain-specialized teachers provide dense token-level rewards. The model extends to 256k context and repurposes MTP layers for speculative decoding to reach up to 3.6 acceptance length and 2.6x speedup,开放

What carries the argument

Mixture-of-Experts architecture with 15B active parameters out of 309B total, supported by Multi-Teacher On-Policy Distillation that transfers expertise from specialized teachers via token-level rewards.

If this is right

  • The model reaches comparable reasoning and agentic performance to systems with two or three times more total parameters.
  • Inference runs up to 2.6 times faster with 3.6 token acceptance length by treating MTP layers as a speculative draft model.
  • Context length extends to 256k after initial 32k training without separate long-context pre-training.
  • Open release of the model weights and three-layer MTP weights supports community use and further development.

Where Pith is reading between the lines

These are editorial extensions of the paper, not claims the author makes directly.

  • Sparse activation paired with targeted distillation may let future models achieve high capability at lower memory and compute cost during deployment.
  • Hybrid sliding-window and global attention offers a practical balance for long-context tasks that avoids full quadratic scaling.
  • Reusing pre-training prediction heads for inference acceleration could generalize to other auxiliary objectives in language models.

Load-bearing premise

The unreported benchmark results and training details actually demonstrate performance rivaling DeepSeek-V3.2 and Kimi-K2 under comparable conditions.

What would settle it

Independent runs on the same public benchmarks where MiMo-V2-Flash scores noticeably below DeepSeek-V3.2 or Kimi-K2.

read the original abstract

We present MiMo-V2-Flash, a Mixture-of-Experts (MoE) model with 309B total parameters and 15B active parameters, designed for fast, strong reasoning and agentic capabilities. MiMo-V2-Flash adopts a hybrid attention architecture that interleaves Sliding Window Attention (SWA) with global attention, with a 128-token sliding window under a 5:1 hybrid ratio. The model is pre-trained on 27 trillion tokens with Multi-Token Prediction (MTP), employing a native 32k context length and subsequently extended to 256k. To efficiently scale post-training compute, MiMo-V2-Flash introduces a novel Multi-Teacher On-Policy Distillation (MOPD) paradigm. In this framework, domain-specialized teachers (e.g., trained via large-scale reinforcement learning) provide dense and token-level reward, enabling the student model to perfectly master teacher expertise. MiMo-V2-Flash rivals top-tier open-weight models such as DeepSeek-V3.2 and Kimi-K2, despite using only 1/2 and 1/3 of their total parameters, respectively. During inference, by repurposing MTP as a draft model for speculative decoding, MiMo-V2-Flash achieves up to 3.6 acceptance length and 2.6x decoding speedup with three MTP layers. We open-source both the model weights and the three-layer MTP weights to foster open research and community collaboration.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit. Tearing a paper down is the easy half of reading it; the pith above is the substance, this is the friction.

Referee Report

2 major / 2 minor

Summary. The manuscript presents MiMo-V2-Flash, a Mixture-of-Experts model with 309B total and 15B active parameters that uses hybrid sliding-window attention (128-token window at 5:1 ratio) interleaved with global attention. It is pre-trained on 27 trillion tokens with multi-token prediction (MTP), extended from 32k to 256k context, and post-trained via a novel Multi-Teacher On-Policy Distillation (MOPD) paradigm. The paper claims this model rivals DeepSeek-V3.2 and Kimi-K2 while using only half and one-third their parameters, respectively, and achieves up to 3.6 acceptance length and 2.6x decoding speedup by repurposing MTP layers for speculative decoding. The model and MTP weights are open-sourced.

Significance. If the performance claims hold under matched evaluation conditions, the work would demonstrate practical advances in parameter-efficient MoE scaling for reasoning and agentic capabilities, with the hybrid attention and MOPD methods offering reusable design insights. The open-sourcing of weights and MTP layers would provide immediate value for community replication and further research on speculative decoding.

major comments (2)
  1. [Abstract] Abstract: The claim that MiMo-V2-Flash 'rivals top-tier open-weight models such as DeepSeek-V3.2 and Kimi-K2, despite using only 1/2 and 1/3 of their total parameters, respectively' is unsupported by any benchmark scores, tables, or evaluation details. No side-by-side results on MMLU, GSM8K, HumanEval or similar tasks are supplied, nor is there information on prompting, shot count, or whether baselines were re-evaluated under identical conditions. This is load-bearing for the central contribution.
  2. [Abstract] Abstract: The inference claim of 'up to 3.6 acceptance length and 2.6x decoding speedup with three MTP layers' is presented without experimental setup, hardware details, baseline comparisons, or acceptance-length distributions. This prevents assessment of whether the speedup is reproducible or generalizes beyond the reported conditions.
minor comments (2)
  1. [Abstract] The hybrid attention ratio is described as '5:1' without clarifying whether this denotes the fraction of SWA layers, the interleaving pattern, or another quantity; a diagram or explicit definition in the main text would improve clarity.
  2. [Abstract] The context-length extension from native 32k to 256k is mentioned without describing the method (e.g., RoPE scaling factors, continued pre-training schedule, or long-context benchmark results).

Simulated Author's Rebuttal

2 responses · 0 unresolved

We thank the referee for the constructive feedback on the abstract. We address each major comment below and will revise the manuscript accordingly to strengthen the presentation of results and experimental details.

read point-by-point responses
  1. Referee: [Abstract] Abstract: The claim that MiMo-V2-Flash 'rivals top-tier open-weight models such as DeepSeek-V3.2 and Kimi-K2, despite using only 1/2 and 1/3 of their total parameters, respectively' is unsupported by any benchmark scores, tables, or evaluation details. No side-by-side results on MMLU, GSM8K, HumanEval or similar tasks are supplied, nor is there information on prompting, shot count, or whether baselines were re-evaluated under identical conditions. This is load-bearing for the central contribution.

    Authors: We agree that the abstract claim would benefit from direct supporting evidence. The full manuscript contains benchmark tables and evaluation details in the Experiments section, but to make the abstract self-contained we will revise it to include key side-by-side scores on MMLU, GSM8K, HumanEval and related tasks, along with notes on prompting, shot counts, and confirmation that baselines were run under matched conditions. We will also add explicit references to the relevant tables. revision: yes

  2. Referee: [Abstract] Abstract: The inference claim of 'up to 3.6 acceptance length and 2.6x decoding speedup with three MTP layers' is presented without experimental setup, hardware details, baseline comparisons, or acceptance-length distributions. This prevents assessment of whether the speedup is reproducible or generalizes beyond the reported conditions.

    Authors: We acknowledge the abstract is overly concise on the inference results. The manuscript includes a dedicated section on speculative decoding that describes the MTP-layer repurposing, hardware setup, baseline autoregressive decoding, and acceptance-length statistics. We will revise the abstract to briefly summarize the experimental conditions, hardware, baseline, and key statistics (including distributions), and ensure the full section provides all reproducibility details. revision: yes

Circularity Check

0 steps flagged

No circularity: empirical technical report with no derivation chain

full rationale

The document is a model release report describing architecture choices (hybrid SWA/global attention, MTP pre-training, MOPD post-training), parameter counts, and benchmark rivalry claims. No equations, first-principles derivations, fitted parameters renamed as predictions, or self-citation chains appear in the provided text. Central performance assertions rest on external benchmark comparisons rather than internal definitions that loop back to the same quantities. The paper is therefore self-contained against external benchmarks with no load-bearing circular steps.

Axiom & Free-Parameter Ledger

2 free parameters · 1 axioms · 0 invented entities

The central claims rest on standard transformer and MoE training assumptions plus the unreported empirical results; no new physical entities or ad-hoc constants are introduced beyond typical hyper-parameters such as the 5:1 attention ratio and 128-token window.

free parameters (2)
  • hybrid attention ratio
    5:1 SWA-to-global ratio chosen to balance speed and quality; value stated without derivation from first principles.
  • sliding window size
    128 tokens; selected for inference efficiency.
axioms (1)
  • domain assumption Standard transformer attention and MoE routing assumptions hold at this scale.
    Invoked implicitly when claiming the architecture trains and runs as described.

pith-pipeline@v0.9.0 · 6053 in / 1376 out tokens · 68802 ms · 2026-05-12T11:27:42.763790+00:00 · methodology

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 60 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Omni-DeepSearch: A Benchmark for Audio-Driven Omni-Modal Deep Search

    cs.SD 2026-05 unverdicted novelty 8.0

    Omni-DeepSearch is a 640-sample benchmark for audio-driven omni-modal search where the best model reaches only 43.44% accuracy, exposing bottlenecks in audio inference, tool use, and cross-modal reasoning.

  2. Lightning OPD: Efficient Post-Training for Large Reasoning Models with Offline On-Policy Distillation

    cs.LG 2026-04 unverdicted novelty 8.0

    Lightning OPD enforces teacher consistency by precomputing log-probabilities over SFT rollouts, matching standard OPD performance with bounded gradient discrepancy and achieving 4x speedup on math and code reasoning tasks.

  3. When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizon

    cs.LG 2026-06 unverdicted novelty 7.0

    Online IL overcomes an information-theoretic bottleneck that offline IL faces in non-realizable settings even at horizon 1, under a new structural characterization of reward-relative misspecification.

  4. Measuring & Mitigating Over-Alignment for LLMs in Multilingual Criminal Law Courts

    cs.CL 2026-06 conditional novelty 7.0

    New benchmark quantifies language- and model-dependent over-alignment in criminal law LLM use and identifies abliteration as an effective mitigation with minimal performance cost.

  5. Each Judge Its Own Yardstick: Discovering Per-VLM Taxonomies for Physical Video Evaluation

    cs.CV 2026-06 unverdicted novelty 7.0

    JudgeFit produces per-VLM physical video evaluation taxonomies that improve held-out accuracy by a mean 32% relative to a single global schema across 16 models from eight families.

  6. On the Position Bias of On-Policy Distillation

    cs.LG 2026-06 unverdicted novelty 7.0

    Importance-weighted on-policy distillation counters position bias by scaling token weights according to cumulative student-teacher distribution discrepancy, improving efficiency and final performance over uniform averaging.

  7. Learning from the Self-future: On-policy Self-distillation for dLLMs

    cs.CL 2026-06 unverdicted novelty 7.0

    d-OPSD reframes on-policy self-distillation for dLLMs via suffix conditioning from self-generated answers and step-level supervision, outperforming RLVR and SFT on reasoning benchmarks with ~10% of the optimization steps.

  8. When Context Returns: Toward Robust Internalization in On-Policy Distillation

    cs.LG 2026-06 unverdicted novelty 7.0

    A stop-gradient consistency regularizer mitigates context-induced degradation in on-policy distillation, improving robustness across 12 configurations.

  9. Escaping the KL Agreement Trap in On-Policy Distillation

    cs.LG 2026-06 unverdicted novelty 7.0

    KAT detects persistent low-KL agreement traps in on-policy distillation via a dynamic threshold to filter weak supervision, improving avg@k by 2.66% and pass@k by 3.43% on four math benchmarks while shortening rollout...

  10. OPRD: On-Policy Representation Distillation

    cs.LG 2026-06 unverdicted novelty 7.0

    OPRD performs distillation in hidden-state space on on-policy data for deterministic gradients and better math benchmark performance, plus OPRD-Bridge for cross-architecture transfer via low-rank projectors.

  11. Imagine Before You Predict: Interleaved Latent Visual Reasoning for Video Event Prediction

    cs.CV 2026-06 unverdicted novelty 7.0

    Future-L1 interleaves latent visual spans with text in MLLM decoding, trained on a custom Future-L1-50K dataset via LA-DAPO RL, and reports SOTA gains on FutureBench (61.0 to 85.4) and TwiFF-Bench (2.44 to 3.04).

  12. Reinforcement Learning from Rich Feedback with Distributional DAgger

    cs.LG 2026-06 unverdicted novelty 7.0

    DistIL applies distributional DAgger with forward cross-entropy to achieve monotonic policy improvement and better Pass@N from rich feedback in RL for reasoning tasks.

  13. OmniOPD: Logit-Free On-Policy Distillation via Speculative Verification

    cs.LG 2026-05 unverdicted novelty 7.0

    OmniOPD replaces token-level logit matching in on-policy distillation with Monte Carlo chunk-level semantic verification and a peak-entropy scheduler.

  14. Adapting the Interface, Not the Model: Runtime Harness Adaptation for Deterministic LLM Agents

    cs.AI 2026-05 unverdicted novelty 7.0

    Life-Harness evolves reusable interventions from training trajectories to enhance frozen LLM agents on unseen tasks across seven deterministic environments, yielding 88.5% average relative improvement in 116 of 126 mo...

  15. GRASP: Deterministic argument ranking in interaction graphs

    cs.LG 2026-05 unverdicted novelty 7.0

    GRASP aggregates stable local LLM interaction judgments into global argument rankings via a convergent attack-defense propagation operator on interaction graphs, yielding higher reproducibility than holistic judging a...

  16. BacktestBench: Benchmarking Large Language Models for Automated Quantitative Strategy Backtesting

    cs.CL 2026-05 unverdicted novelty 7.0

    Introduces BacktestBench benchmark with 18k QA pairs across four backtesting tasks and evaluates 23 LLMs via the AutoBacktest multi-agent system.

  17. BacktestBench: Benchmarking Large Language Models for Automated Quantitative Strategy Backtesting

    cs.CL 2026-05 unverdicted novelty 7.0

    BacktestBench is the first large-scale benchmark for LLM-automated quantitative backtesting, with 18,246 QA pairs from real market data and a multi-agent baseline called AutoBacktest.

  18. Decoupling KL and Trajectories: A Unified Perspective for SFT, DAgger, Offline RL, and OPD in LLM Distillation

    cs.LG 2026-05 unverdicted novelty 7.0

    Decoupling prefix source from token-level KL direction in autoregressive sequence KL yields four objectives unifying SFT, DAgger, offline RL and OPD, with KL mixing and entropy-gated curriculum improving math reasonin...

  19. Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why

    cs.LG 2026-05 unverdicted novelty 7.0

    Distillation signals align better with ideal updates on incorrect student rollouts than correct ones, with optimal teacher context depending on student capacity and task.

  20. Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation

    cs.CL 2026-05 unverdicted novelty 7.0

    Persistent 'Rock Tokens' in on-policy distillation resist teacher corrections, consume large gradient norms, yet add negligible value to reasoning, allowing targeted bypassing to streamline alignment.

  21. Beyond the All-in-One Agent: Benchmarking Role-Specialized Multi-Agent Collaboration in Enterprise Workflows

    cs.MA 2026-05 unverdicted novelty 7.0

    EntCollabBench shows that today's LLM agents still struggle with delegation, context transfer, parameter grounding, workflow closure, and decision commitment when tested in a simulated enterprise with 11 role-speciali...

  22. Training with Harnesses: On-Policy Harness Self-Distillation for Complex Reasoning

    cs.CL 2026-05 unverdicted novelty 7.0

    OPHSD uses harness-augmented models as teachers to distill reasoning capabilities into base LLMs, yielding strong standalone performance on classification and math tasks.

  23. Flow-OPD: On-Policy Distillation for Flow Matching Models

    cs.CV 2026-05 conditional novelty 7.0

    Flow-OPD applies on-policy distillation to flow matching models via specialized teachers, cold-start initialization, and manifold anchor regularization, lifting GenEval from 63 to 92 and OCR from 59 to 94 on Stable Di...

  24. GameGen-Verifier: Parallel Keypoint-Based Verification for LLM-Generated Games via Runtime State Injection

    cs.LG 2026-05 unverdicted novelty 7.0

    GameGen-Verifier decomposes game specifications into keypoints, injects runtime states for targeted checks, and achieves 92.2% accuracy on 100 games while running up to 16.6x faster than agent-based baselines.

  25. Rubric-based On-policy Distillation

    cs.LG 2026-05 unverdicted novelty 7.0

    Rubric-based on-policy distillation allows training student models using only teacher responses by generating scoring rubrics from contrasts and using them for on-policy optimization, achieving superior performance an...

  26. Attention Sink in Transformers: A Survey on Utilization, Interpretation, and Mitigation

    cs.LG 2026-04 unverdicted novelty 7.0

    The first survey on Attention Sink in Transformers structures the literature around fundamental utilization, mechanistic interpretation, and strategic mitigation.

  27. Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Models

    cs.CL 2026-04 unverdicted novelty 7.0

    OPD for LLMs suffers length inflation and repetition collapse; StableOPD uses reference divergence and rollout mixing to prevent it and improve math reasoning performance by 7.2% on average.

  28. Sell More, Play Less: Benchmarking LLM Realistic Selling Skill

    cs.CL 2026-04 conditional novelty 7.0

    SalesLLM provides an automatic evaluation framework for LLM sales dialogues that correlates 0.98 with human experts and shows top models approaching human performance while weaker ones lag.

  29. DetailVerifyBench: A Benchmark for Dense Hallucination Localization in Long Image Captions

    cs.CV 2026-04 unverdicted novelty 7.0

    DetailVerifyBench supplies 1,000 images and densely annotated long captions to evaluate precise hallucination localization in multimodal large language models.

  30. Self-Distilled RLVR

    cs.LG 2026-04 unverdicted novelty 7.0

    RLSD mixes self-distillation for token-level policy difference magnitudes with RLVR for reliable update directions from response correctness to reach higher convergence and better training stability.

  31. Video-OPD: Efficient Post-Training of Multimodal Large Language Models for Temporal Video Grounding via On-Policy Distillation

    cs.CV 2026-02 unverdicted novelty 7.0

    Video-OPD uses on-policy distillation from a frontier teacher to turn sparse episode rewards into dense step-wise signals for more efficient post-training of MLLMs on temporal video grounding.

  32. Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models

    cs.LG 2026-01 unverdicted novelty 7.0

    A single LLM improves its own reasoning by self-distilling from privileged verified traces as teacher to its question-only student policy, outperforming off-policy distillation and RL on math benchmarks with better to...

  33. AsyncOPD: How Stale Can On-Policy Distillation Be?

    cs.LG 2026-06 conditional novelty 6.0

    AsyncOPD shows asynchronous OPD training reaches 1.6-3.8x higher throughput than synchronous baselines with comparable accuracy by using forward-KL estimators and multi-sample Monte Carlo correction for finite teacher caches.

  34. Measuring & Mitigating Over-Alignment for LLMs in Multilingual Criminal Law Courts

    cs.CL 2026-06 unverdicted novelty 6.0

    Creates TF-RefusalBench to quantify over-alignment in LLMs on criminal-law tasks across four languages and shows abliteration mitigates refusals with little performance loss.

  35. On the Position Bias of On-Policy Distillation

    cs.LG 2026-06 unverdicted novelty 6.0

    Position bias in on-policy distillation degrades later-token supervision; IW-OPD weights tokens by accumulated discrepancy, yielding faster convergence and up to 6.9 point gains on AIME-2025.

  36. JetSpec: Breaking the Scaling Ceiling of Speculative Decoding with Parallel Tree Drafting

    cs.CL 2026-06 unverdicted novelty 6.0

    JetSpec trains a causal draft head to produce branch-consistent trees aligned with target autoregressive scores, achieving up to 9.64x speedup on MATH-500 and outperforming prior SD baselines on Qwen3 models.

  37. ConSA: Controllable Sparsity in Hybrid Attention via Learnable Allocation

    cs.CL 2026-06 unverdicted novelty 6.0

    ConSA learns FA/SWA allocation via L0 masks and augmented Lagrangian constraints, outperforming rule-based baselines on 0.6B and 1.7B models with consistent layer patterns.

  38. Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillation

    cs.LG 2026-06 unverdicted novelty 6.0

    On-policy distillation produces coordinate-sparse, FFN-heavy updates that are full-rank but spectrally concentrated away from principal singular subspaces and near-zero source weights.

  39. RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation

    cs.LG 2026-06 unverdicted novelty 6.0

    RLCSD contrasts teacher-student distributional gaps under correct versus wrong hints to suppress privilege-induced style drift and concentrate supervision on task tokens, outperforming GRPO and prior OPSD on Qwen3 and...

  40. Consolidating Rewarded Perturbations for LLM Post-Training

    cs.CL 2026-05 unverdicted novelty 6.0

    CoRP consolidates reward-weighted perturbations into a single model via low-rank structure, improving base LLMs by 8.1 points on average while using one-tenth the budget of prior ensembles and one forward pass.

  41. DeepSurvey: Enhancing Analytical Depth and Citation Reliability in Automated Survey Generation

    cs.AI 2026-05 unverdicted novelty 6.0

    DeepSurvey introduces an agentic system for automated survey generation that improves depth through full-text keynotes, cross-paper clustering, and code analysis, while boosting citation reliability via graph expansio...

  42. CollectionLoRA: Collecting 50 Effects in 1 LoRA via Multi-Teacher On-Policy Distillation

    cs.CV 2026-05 unverdicted novelty 6.0

    A multi-teacher distillation framework that packs 50 effect LoRAs and fast sampling into a single adapter while aiming to avoid concept interference.

  43. Adapting the Interface, Not the Model: Runtime Harness Adaptation for Deterministic LLM Agents

    cs.AI 2026-05 unverdicted novelty 6.0

    Life-Harness evolves reusable runtime interventions from training failures to improve frozen LLM agents by 88.5% on average across 126 settings in seven deterministic environments while transferring across 18 model backbones.

  44. Hack-Verifiable Environments: Towards Evaluating Reward Hacking at Scale

    cs.LG 2026-05 unverdicted novelty 6.0

    Presents Hack-Verifiable TextArena, a benchmark that embeds verifiable reward hacking opportunities into environments to enable deterministic measurement of exploitation by language models.

  45. Self-Supervised On-Policy Distillation for Reasoning Language Models

    cs.LG 2026-05 unverdicted novelty 6.0

    SSOPD converts intra-group correct-wrong contrast into process supervision by distilling a teacher distribution from the shortest correct completion into prefixes of the longest wrong completion, improving GRPO on AIM...

  46. Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence

    cs.LG 2026-05 unverdicted novelty 6.0

    TGPO improves on-policy reasoning distillation in LLMs by using teacher-guided token generation conditioned on student contexts together with trajectory rewards, outperforming RKL-based methods under large policy divergence.

  47. Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence

    cs.LG 2026-05 unverdicted novelty 6.0

    TGPO improves on-policy LLM distillation by using teacher predictions conditioned on student rollouts to supply informative guidance when the two distributions diverge.

  48. Learning with Rare Success but Rich Feedback via Reflection-Enhanced Self-Distillation

    cs.LG 2026-05 unverdicted novelty 6.0

    RESD turns failure trajectories into token-level supervision via retrospective reflections and a persistent global playbook, enabling faster improvement than standard self-distillation or GRPO with only one rollout pe...

  49. Beyond GRPO and On-Policy Distillation: An Empirical Sparse-to-Dense Reward Principle for Language-Model Post-Training

    cs.LG 2026-05 unverdicted novelty 6.0

    Sparse RL on capable teachers followed by dense distillation to students beats direct GRPO on students for verifiable math reasoning.

  50. Missing Old Logits in Asynchronous Agentic RL: Semantic Mismatch and Repair Methods for Off-Policy Correction

    cs.LG 2026-05 unverdicted novelty 6.0

    Missing old logits in async agentic RL entangle discrepancy and staleness terms in PPO off-policy correction; exact acquisition methods and revised PPO-EWMA restore decoupled updates with reported gains in speed and p...

  51. Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation

    cs.CL 2026-05 unverdicted novelty 6.0

    On-policy distillation gains efficiency from early foresight in module focus and update directions, enabling EffOPD to accelerate training 3x with comparable performance.

  52. Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation

    cs.CL 2026-05 unverdicted novelty 6.0

    On-policy distillation gains efficiency from early foresight in module allocation and low-rank update directions, enabling EffOPD to accelerate training by 3x via adaptive extrapolation without extra modules or tuning.

  53. Anti-Self-Distillation for Reasoning RL via Pointwise Mutual Information

    cs.LG 2026-05 unverdicted novelty 6.0

    Anti-Self-Distillation reverses self-distillation signals via PMI to fix overconfidence on structural tokens, matching GRPO baseline accuracy 2-10x faster with up to 11.5 point gains across 4B-30B models.

  54. An Information-Theoretic Criterion for Efficient Data Synthesis

    cs.LG 2026-05 unverdicted novelty 6.0

    Synthetic data improves models only in information-open generation-training loops with external signals, and coarser signals like binary correctness enable better generalization by converging to the most information-e...

  55. Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation

    cs.CL 2026-05 unverdicted novelty 6.0

    Rock Tokens in on-policy distillation persist at high loss, account for up to 18% of outputs, absorb large gradient norms, but add negligible value to reasoning performance.

  56. Flow-OPD: On-Policy Distillation for Flow Matching Models

    cs.CV 2026-05 unverdicted novelty 6.0

    Flow-OPD applies on-policy distillation to flow-matching text-to-image models, lifting GenEval from 63 to 92 and OCR accuracy from 59 to 94 while preserving fidelity.

  57. Flow-OPD: On-Policy Distillation for Flow Matching Models

    cs.CV 2026-05 unverdicted novelty 6.0

    Flow-OPD applies on-policy distillation to flow matching models, achieving GenEval of 92 and OCR accuracy of 94 on Stable Diffusion 3.5 Medium while avoiding the seesaw effect of multi-reward optimization.

  58. Flow-OPD: On-Policy Distillation for Flow Matching Models

    cs.CV 2026-05 unverdicted novelty 6.0

    Flow-OPD applies on-policy distillation to Flow Matching models through specialized teachers, cold-start initialization, task routing, and manifold regularization, lifting GenEval from 63 to 92 and OCR from 59 to 94 o...

  59. Flow-OPD: On-Policy Distillation for Flow Matching Models

    cs.CV 2026-05 conditional novelty 6.0

    Flow-OPD is a two-stage on-policy distillation method for flow matching models that lifts GenEval from 63 to 92 and OCR from 59 to 94 on SD 3.5 Medium while preserving fidelity.

  60. UniPrefill: Universal Long-Context Prefill Acceleration via Block-wise Dynamic Sparsification

    cs.CL 2026-05 unverdicted novelty 6.0

    UniPrefill accelerates LLM prefill via block-wise dynamic sparsification, achieving up to 2.1x TTFT speedup while supporting hybrid architectures and native vLLM continuous batching.

Reference graph

Works this paper leans on

51 extracted references · 51 canonical work pages · cited by 76 Pith papers · 29 internal anchors

  1. [1]

    URLhttps://api.semanticscholar.org/CorpusID: 263610088. S. Agarwal, L. Ahmad, J. Ai, S. Altman, A. Applebaum, E. Arbus, R. K. Arora, Y. Bai, B. Baker, H. Bao, et al. gpt-oss-120b & gpt-oss-20b model card.arXiv preprint arXiv:2508.10925,

  2. [2]

    GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints

    J. Ainslie, J. Lee-Thorp, M. De Jong, Y. Zemlyanskiy, F. Lebrón, and S. Sanghai. Gqa: Training generalized multi-query transformer models from multi-head checkpoints. arXiv preprint arXiv:2305.13245,

  3. [4]

    URLhttps://arxiv.org/abs/2108.07732. Y. Bai, S. Tu, J. Zhang, H. Peng, X. Wang, X. Lv, S. Cao, J. Xu, L. Hou, Y. Dong, et al. Long- bench v2: Towards deeper understanding and reasoning on realistic long-context multitasks. 22 In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume1: Long Papers), pages 3639–3664,

  4. [5]

    MathArena: Evaluating LLMs on Uncontaminated Math Competitions

    M. Balunović, J. Dekoninck, I. Petrov, N. Jovanović, and M. Vechev. Matharena: Evaluating llms on uncontaminated math competitions.arXiv preprint arXiv:2505.23281,

  5. [6]

    $\tau^2$-Bench: Evaluating Conversational Agents in a Dual-Control Environment

    V. Barres, H. Dong, S. Ray, X. Si, and K. Narasimhan.𝜏2-bench: Evaluating conversational agents in a dual-control environment.arXiv preprint arXiv:2506.07982,

  6. [7]

    Longformer: The Long-Document Transformer

    I. Beltagy, M. E. Peters, and A. Cohan. Longformer: The long-document transformer.arXiv preprint arXiv:2004.05150,

  7. [8]

    Brown, B

    T. Brown, B. Mann, N. Ryder, M. Subbiah, J. D. Kaplan, P. Dhariwal, A. Neelakantan, P. Shyam, G.Sastry,A.Askell,etal. Languagemodelsarefew-shotlearners. Advancesinneuralinformation processing systems, 33:1877–1901,

  8. [9]

    arXiv preprint arXiv:2208.08227 , year=

    F. Cassano, J. Gouwar, D. Nguyen, S. Nguyen, L. Phipps-Costin, D. Pinckney, M.-H. Yee, Y. Zi, C. J. Anderson, M. Q. Feldman, et al. Multipl-e: A scalable and extensible approach to benchmarking neural code generation.arXiv preprint arXiv:2208.08227,

  9. [11]

    URLhttps://arxiv.org/abs/1803.05457. K. Cobbe, V. Kosaraju, M. Bavarian, M. Chen, H. Jun, L. Kaiser, M. Plappert, J. Tworek, J. Hilton, R. Nakano, et al. Training verifiers to solve math word problems.ArXiv preprint, abs/2110.14168,

  10. [12]

    URLhttps://arxiv.org/abs/2110.14168. X. Du, Y. Yao, K. Ma, B. Wang, T. Zheng, K. Zhu, M. Liu, Y. Liang, X. Jin, Z. Wei, et al. Supergpqa: Scaling llm evaluation across 285 graduate disciplines.ArXiv preprint, abs/2502.14739,

  11. [13]

    SuperGPQA: Scaling LLM Evaluation across 285 Graduate Disciplines

    URLhttps://arxiv.org/abs/2502.14739. D.Dua,Y.Wang,P.Dasigi,G.Stanovsky,S.Singh,andM.Gardner. DROP:Areadingcomprehension benchmark requiring discrete reasoning over paragraphs. In J. Burstein, C. Doran, and T. Solorio, editors, Proceedings ofthe 2019 Conferenceofthe North American Chapter ofthe Association for Computational Linguistics: Human Language Tech...

  12. [14]

    Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies,

    Association for Computational Linguistics. doi: 10.18653/v1/N19-1246. URLhttps://aclanthology.org/N19-1246. W. Fu, J. Gao, X. Shen, C. Zhu, Z. Mei, C. He, S. Xu, G. Wei, J. Mei, J. Wang, et al. Areal: A large-scale asynchronous reinforcement learning system for language reasoning.arXiv preprint arXiv:2505.24298,

  13. [15]

    W. Gao, Y. Zhao, D. An, T. Wu, L. Cao, S. Xiong, J. Huang, W. Wang, S. Yang, W. Su, et al. Rollpacker: Mitigating long-tail rollouts for fast, synchronous rl post-training.arXiv preprint arXiv:2509.21009,

  14. [17]

    Are we done with mmlu?arXiv preprint arXiv:2406.04127, 2024

    URL https://arxiv.org/abs/2406.04127. Gemma Team. Gemma 2: Improving open language models at a practical size.arXiv preprint arXiv:2408.00118,

  15. [18]

    Gemma 3 Technical Report

    Gemma Team. Gemma 3 technical report.arXiv preprint arXiv:2503.19786,

  16. [19]

    Better & Faster Large Language Models via Multi-token Prediction

    23 F. Gloeckle, B. Y. Idrissi, B. Rozière, D. Lopez-Paz, and G. Synnaeve. Better & faster large language models via multi-token prediction.arXiv preprint arXiv:2404.19737,

  17. [20]

    A. Gu, B. Rozière, H. J. Leather, A. Solar-Lezama, G. Synnaeve, and S. Wang. Cruxeval: A bench- mark for code reasoning, understanding and execution. InForty-first International Conference on Machine Learning, ICML 2024, Vienna, Austria, July 21-27,

  18. [21]

    When Attention Sink Emerges in Language Models: An Empirical View

    OpenReview.net, 2024a. URLhttps://openreview.net/forum?id=Ffpg52swvg. X. Gu, T. Pang, C. Du, Q. Liu, F. Zhang, C. Du, Y. Wang, and M. Lin. When attention sink emerges in language models: An empirical view.arXiv preprint arXiv:2410.10781, 2024b. Y. Gu, L. Dong, F. Wei, and M. Huang. Minillm: Knowledge distillation of large language models. In Proceedings o...

  19. [22]

    Hendrycks, C

    D. Hendrycks, C. Burns, S. Basart, A. Zou, M. Mazeika, D. Song, and J. Steinhardt. Measur- ing massive multitask language understanding. In9th International Conference on Learning Representations, ICLR 2021, Virtual Event, Austria, May 3-7,

  20. [23]

    Measuring Mathematical Problem Solving With the MATH Dataset

    OpenReview.net, 2021a. URLhttps://openreview.net/forum?id=d7KBjmI3GmQ. D. Hendrycks, C. Burns, S. Kadavath, A. Arora, S. Basart, E. Tang, D. Song, and J. Steinhardt. Mea- suring mathematical problem solving with the math dataset.ArXiv preprint, abs/2103.03874, 2021b. URLhttps://arxiv.org/abs/2103.03874. C.-P. Hsieh, S. Sun, S. Kriman, S. Acharya, D. Rekes...

  21. [24]

    Huang, Y

    Y. Huang, Y. Bai, Z. Zhu, J. Zhang, J. Zhang, T. Su, J. Liu, C. Lv, Y. Zhang, J. Lei, Y. Fu, M. Sun, and J. He. C-eval: A multi-level multi-discipline chinese evaluation suite for foundation models. In A. Oh, T. Naumann, A. Globerson, K. Saenko, M. Hardt, and S. Levine, edi- tors, Advances in Neural Information Processing Systems 36: Annual Conference on ...

  22. [26]

    URLhttps://arxiv.org/abs/2403.07974. C. E. Jimenez, J. Yang, A. Wettig, S. Yao, K. Pei, O. Press, and K. R. Narasimhan. Swe-bench: Can language models resolve real-world github issues? InThe Twelfth International Conference on Learning Representations, ICLR 2024, Vienna, Austria, May 7-11,

  23. [27]

    Weld and Luke Zettlemoyer , editor =

    Association for Computational Linguistics. doi: 10.18653/v1/P17-1147. URLhttps://aclanthology.org/P17-1147. Kimi Team. Kimi linear: An expressive, efficient attention architecture. arXiv preprint arXiv:2510.26692, 2025a. KimiTeam. Kimik1.5: Scalingreinforcementlearningwithllms. arXivpreprintarXiv:2501.12599, 2025b. Kimi Team. Kimi k2: Open agentic intelli...

  24. [29]

    URLhttps://arxiv.org/abs/2306.09212. T. Li, W.-L. Chiang, E. Frick, L. Dunlap, B. Zhu, J. E. Gonzalez, and I. Stoica. From live data to high-quality benchmarks: The arena-hard pipeline, April

  25. [30]

    URLhttps://lmsys.org/ blog/2024-04-19-arena-hard/. A. Liu, B. Feng, B. Xue, B. Wang, B. Wu, C. Lu, C. Zhao, C. Deng, C. Zhang, C. Ruan, et al. Deepseek-v3 technical report.arXiv preprint arXiv:2412.19437,

  26. [31]

    A. Liu, A. Mei, B. Lin, B. Xue, B. Wang, B. Xu, B. Wu, B. Zhang, C. Lin, C. Dong, et al. Deepseek-v3. 2: Pushing the frontier of open large language models.arXiv preprint arXiv:2512.02556,

  27. [32]

    J. Liu, C. S. Xia, Y. Wang, and L. Zhang. Is your code generated by chatgpt really correct? rigorous evaluation of large language models for code generation. In A. Oh, T. Naumann, A. Globerson, K.Saenko, M.Hardt, andS.Levine, editors, AdvancesinNeuralInformationProcessingSystems 36: Annual Conference on Neural Information Processing Systems 2023, NeurIPS ...

  28. [33]

    URLhttp://papers.nips.cc/paper_f iles/paper/2023/hash/43e9d647ccd3e4b7b5baab53f0368686-Abstract-Confere nce.html. K. Lu and T. M. Lab. On-policy distillation.Thinking Machines Lab: Connectionism,

  29. [34]

    Lu and T

    doi: 10.64434/tml.20251026. https://thinkingmachines.ai/blog/on-policy-distillation. W. Ma, H. Zhang, L. Zhao, Y. Song, Y. Wang, Z. Sui, and F. Luo. Stabilizing moe reinforcement learning by aligning training and inference routers.arXiv preprint arXiv:2510.11370,

  30. [35]

    URL https://maa.org/math-competition s/american-invitational-mathematics-examination-aime. 25 A. Modarressi, H. Deilamsalehy, F. Dernoncourt, T. Bui, R. A. Rossi, S. Yoon, and H. Schütze. Nolima: Long-context evaluation beyond literal matching.arXiv preprint arXiv:2502.05167,

  31. [36]

    L. Phan, A. Gatti, Z. Han, N. Li, J. Hu, H. Zhang, C. B. C. Zhang, M. Shaaban, J. Ling, S. Shi, et al. Humanity’s last exam.arXiv preprint arXiv:2501.14249,

  32. [37]

    Z. Qiu, Z. Wang, B. Zheng, Z. Huang, K. Wen, S. Yang, R. Men, L. Yu, F. Huang, S. Huang, et al. Gated attention for large language models: Non-linearity, sparsity, and attention-sink-free.arXiv preprint arXiv:2505.06708,

  33. [38]

    Introducing miles — rl framework to fire up large-scale moe training.https: //lmsys.org/blog/2025-11-19-miles/, Nov

    RadixArk Team. Introducing miles — rl framework to fire up large-scale moe training.https: //lmsys.org/blog/2025-11-19-miles/, Nov

  34. [39]

    Include: Evaluating multilingual language understanding with regional knowledge.arXiv preprint arXiv:2411.19799, 2024

    A. Romanou, N. Foroutan, A. Sotnikova, Z. Chen, S. H. Nelaturu, S. Singh, R. Maheshwary, M. Altomare, M. A. Haggag, A. Amayuelas, et al. Include: Evaluating multilingual language understanding with regional knowledge.arXiv preprint arXiv:2411.19799,

  35. [40]

    Sakaguchi, R

    K. Sakaguchi, R. L. Bras, C. Bhagavatula, and Y. Choi. Winogrande: An adversarial winograd schema challenge at scale. InThe Thirty-Fourth AAAI Conference on Artificial Intelligence, AAAI 2020, The Thirty-Second Innovative Applications of Artificial Intelligence Conference, IAAI2020,TheTenthAAAISymposiumonEducationalAdvancesinArtificialIntelligence,EAAI 20...

  36. [41]

    URL https://aaai.org/ojs/index.php/AAAI/article/view/6399. J. Schulman, F. Wolski, P. Dhariwal, A. Radford, and O. Klimov. Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347,

  37. [42]

    URLhttps://arxiv.org/abs/2402.03300. N. Shazeer, A. Mirhoseini, K. Maziarz, A. Davis, Q. Le, G. Hinton, and J. Dean. Outrageously large neural networks: The sparsely-gated mixture-of-experts layer.arXivpreprint arXiv:1701.06538,

  38. [43]

    Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism

    M. Shoeybi, M. Patwary, R. Puri, P. LeGresley, J. Casper, and B. Catanzaro. Megatron-lm: Training multi-billion parameter language models using model parallelism.arXiv preprint arXiv:1909.08053,

  39. [44]

    M. Sun, X. Chen, J. Z. Kolter, and Z. Liu. Massive activations in large language models.arXiv preprint arXiv:2402.17762,

  40. [45]

    Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Them , booktitle =

    M. Suzgun, N. Scales, N. Schärli, S. Gehrmann, Y. Tay, H. W. Chung, A. Chowdhery, Q. Le, E. Chi, D. Zhou, and J. Wei. Challenging BIG-bench tasks and whether chain-of-thought can solve them. In A. Rogers, J. Boyd-Graber, and N. Okazaki, editors,Findings of the Association for ComputationalLinguistics: ACL2023, pages13003–13051, Toronto, Canada, 2023.Assoc...

  41. [46]

    Jason Weston, Antoine Bordes, Sumit Chopra, Alexan- der M

    K.Vodrahalli, S.Ontanon, N.Tripuraneni, K.Xu, S.Jain, R.Shivanna, J.Hui, N.Dikkala, M.Kazemi, B. Fatemi, et al. Michelangelo: Long context evaluations beyond haystacks via latent structure queries. arXiv preprint arXiv:2409.12640,

  42. [47]

    Y.Wang, X.Ma, G.Zhang, Y.Ni, A.Chandra, S.Guo, W.Ren, A.Arulraj, X.He, Z.Jiang, T.Li, M.Ku, K. Wang, A. Zhuang, R. Fan, X. Yue, and W. Chen. Mmlu-pro: A more robust and challenging multi-task language understanding benchmark. In A. Globersons, L. Mackey, D. Belgrave, A. Fan, U. Paquet, J. M. Tomczak, and C. Zhang, editors,Advances in Neural Information Pr...

  43. [48]

    URLhttp://pape rs.nips.cc/paper_files/paper/2024/hash/ad236edc564f3e3156e1b2feafb99a2 4-Abstract-Datasets_and_Benchmarks_Track.html. J. Wei, Z. Sun, S. Papay, S. McKinney, J. Han, I. Fulford, H. W. Chung, A. T. Passos, W. Fedus, and A. Glaese. Browsecomp: A simple yet challenging benchmark for browsing agents.arXiv preprint arXiv:2504.12516,

  44. [49]

    B. Xia, B. Shen, D. Zhu, D. Zhang, G. Wang, H. Zhang, H. Liu, J. Xiao, J. Dong, L. Zhao, et al. Mimo: Unlocking the reasoning potential of language model–from pretraining to posttraining. arXiv preprint arXiv:2505.07608,

  45. [50]

    C. S. Xia, Y. Deng, S. Dunn, and L. Zhang. Agentless: Demystifying llm-based software engineering agents. arXiv preprint arXiv:2407.01489,

  46. [51]

    G. Xiao, Y. Tian, B. Chen, S. Han, and M. Lewis. Efficient streaming language models with attention sinks.arXiv preprint arXiv:2309.17453,

  47. [52]

    URLhttps: //arxiv.org/abs/2504.21798. 27 F. Yao, L. Liu, D. Zhang, C. Dong, J. Shang, and J. Gao. Your efficient rl framework secretly brings you off-policy rl training, Aug

  48. [53]

    HellaSwag: Can a Machine Really Finish Your Sentence? , booktitle =

    Association for Computational Linguistics. doi: 10.18653/v1/P19-1472. URL https://aclanthology.org/P19-1472. X. Zhao, Y. Liu, K. Xu, J. Guo, Z. Wang, Y. Sun, X. Kong, Q. Cao, L. Jiang, Z. Wen, Z. Zhang, and J. Zhou. Small leak can sink a great ship–boost rl training on moe with icepop!, Sep

  49. [54]

    URL https://ringtech.notion.site/icepop. C. Zheng, S. Liu, M. Li, X.-H. Chen, B. Yu, C. Gao, K. Dang, Y. Liu, R. Men, A. Yang, et al. Group sequence policy optimization.arXiv preprint arXiv:2507.18071,

  50. [55]

    Y. Zhou, H. Liu, Z. Chen, Y. Tian, and B. Chen. Gsm-infinite: How do your llms behave over infinitelyincreasingcontextlengthandreasoningcomplexity? arXivpreprintarXiv:2502.05252,

  51. [56]

    T. Y. Zhuo, M. C. Vu, J. Chim, H. Hu, W. Yu, R. Widyasari, I. N. B. Yusuf, H. Zhan, J. He, I. Paul, et al. Bigcodebench: Benchmarking code generation with diverse function calls and complex instructions. arXiv preprint arXiv:2406.15877,