Yaoyuan Wang
Author directory2026
PV-Decoding: Progressive Verification Decoding with Decode Gating for Diffusion Language Models
Renxi Liu | Yiming Li | Shengyin Sun | Chen Chen | Xian Wang | Shuai Yang | Chen xi | Wenbo Gao | Xiaoguang Li | Hui-Ling Zhen | Xianzhi Yu | Weizhe Lin | Xiaosong Li | Yaoyuan Wang
Findings of the Association for Computational Linguistics: EMNLP 2026
Renxi Liu | Yiming Li | Shengyin Sun | Chen Chen | Xian Wang | Shuai Yang | Chen xi | Wenbo Gao | Xiaoguang Li | Hui-Ling Zhen | Xianzhi Yu | Weizhe Lin | Xiaosong Li | Yaoyuan Wang
Findings of the Association for Computational Linguistics: EMNLP 2026
Discrete diffusion language models (dLLMs) offer an attractive alternative to autoregressive generation by updating multiple tokens in parallel, but their inference efficiency is still limited by the cost of safely committing tentative predictions. In particular, speculative decoding for dLLMs is challenging because speculative tokens within the same diffusion block may depend on one another, making aggressive acceptance prone to error propagation and strict verification computationally expensive. We present PV-Decoding, a confidence-driven speculative decoding framework with progressive verification for block diffusion inference. PV-Decoding partitions masked positions into directly decoded tokens, speculative tokens, and unresolved tokens using two confidence thresholds, and then verifies speculative tokens with progressively stronger policies, including single-batch, dual-batch, and multi-batch verification. This design exposes a practical accuracy–efficiency trade-off: weaker verification minimizes overhead, whereas stronger verification improves acceptance precision and robustness. Experiments on three diffusion language models across GSM8K, MBPP, and HumanEval show that PV-Decoding reduces sequential model forward passes (MFPs) by 8%–37% relative to confidence-based decoding, while keeping the largest observed accuracy drop to 1.3%. On Dream-7B-Instruct/GSM8K, PV-Decoding (1-batch) and PV-Decoding (2-batch) improve throughput by 7.0%–13.6% under the same hardware budget, whereas PV-Decoding (multi-batch) is slower under the current runtime. These results suggest that confidence-guided speculation with structured verification is a practical path toward efficient and reliable dLLM inference.
GNMR: Runtime Stability Control for Low-Precision Large Language Model Training
Boao Kong | Weichen Jia | Engao Zhang | Guohong Li | Yonghan Dong | Yao Wang | Yaoyuan Wang | Yunke Peng | Kun Yuan
Findings of the Association for Computational Linguistics: EMNLP 2026
Boao Kong | Weichen Jia | Engao Zhang | Guohong Li | Yonghan Dong | Yao Wang | Yaoyuan Wang | Yunke Peng | Kun Yuan
Findings of the Association for Computational Linguistics: EMNLP 2026
Training stability is a key bottleneck in low-precision language model training: efficient low-cost paths can still produce short-lived numerical risks at a small set of operators. We formulate this as runtime stability control and present **G**radient **N**orm-to-**M**ean **R**atio (**GNMR**), a lightweight controller that compares each recoverable unit’s current gradient norm with its historical mean. Together with 𝛥-GNMR for abrupt short-window increases, GNMR maps local risk signals to bounded recovery actions under a hard maxO budget and a short lock interval, without changing the numerical format, kernel, or backend recipe. Across activation-quantization stress, DeepSeek-style recipe-level training, and LLaMA-2 13B fine-tuning, GNMR preserves high-fidelity quality with sparse, budgeted recovery. These results support GNMR as a backend-agnostic controller to improve low-precision training stability while preserving low-cost execution.
Half-S: Halving the Scale for Near-Lossless 4-Bit LLM Training
Jinyang Du | Ruihao Gong | Linghan Ai | Zining Wang | Yunke Peng | Yao Wang | Lei Yan | Wxuefei | Yaoyuan Wang | Jinyang Guo | Dahua Lin | Xianglong Liu
Findings of the Association for Computational Linguistics: ACL 2026
Jinyang Du | Ruihao Gong | Linghan Ai | Zining Wang | Yunke Peng | Yao Wang | Lei Yan | Wxuefei | Yaoyuan Wang | Jinyang Guo | Dahua Lin | Xianglong Liu
Findings of the Association for Computational Linguistics: ACL 2026
Training large language models (LLMs) at 4-bit precision offers substantial efficiency gains but remains challenging due to the limited dynamic range and coarse numerical resolution. Existing 4-bit training pipelines typically rely on max-scaling, which is ill-suited for heavy-tailed LLM tensor distributions and leads to severe under-utilization of the FP4 quantization grid in the low-magnitude region. This effect causes pronounced representation collapse and large rounding errors for the values that dominate LLM computation. In this work, we derive the theoretically optimal scaling for FP4 under heavy-tailed inputs, revealing why max-scaling is intrinsically suboptimal. Guided by this analysis, we propose Half-S, a simple and efficient scaling strategy that uses half-scaling as a hardware-friendly default and falls back to an MSE-based clipping threshold when needed, yielding a close approximation to the theoretical optimum under real LLM statistics. Extensive experiments on large-scale pretraining and downstream fine-tuning show that Half-S consistently narrows the gap to BF16 in both convergence and final model quality, while preserving the efficiency benefits of 4-bit computation. Under native FP4 support, Half-S is estimated to provide up to 1.8× end-to-end training speedup. These results indicate that Half-S provides a simple and effective correction to max-scaling, substantially improving the stability and accuracy of 4-bit LLM training.
AgentCollab: A Self-Evaluation-Driven Collaboration Paradigm for Efficient LLM Agents
Wenbo Gao | Renxi Liu | Xian Wang | Guo Fang | Shuai Yang | Chen xi | Hui-Ling Zhen | Hanting Chen | Weizhe Lin | Xiaosong Li | Yaoyuan Wang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Wenbo Gao | Renxi Liu | Xian Wang | Guo Fang | Shuai Yang | Chen xi | Hui-Ling Zhen | Hanting Chen | Weizhe Lin | Xiaosong Li | Yaoyuan Wang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Autonomous agents powered by large language models (LLMs) perform complex tasks through long-horizon reasoning and tool interaction, where a fundamental trade-off arises between execution efficiency and reasoning robustness. Models at different capability–cost levels offer complementary advantages: lower-cost models enable fast execution but may struggle on difficult reasoning segments, while stronger models provide more robust reasoning at higher computational cost. We present AgentCollab, a self-driven collaborative inference framework that dynamically coordinates models with different reasoning capacities during agent execution. Instead of relying on external routing modules, the framework uses the agent’s own self-reflection signal to determine whether the current reasoning trajectory is making meaningful progress, and escalates control to a stronger reasoning tier only when necessary. To further stabilize long-horizon execution, we introduce a difficulty-aware cumulative escalation strategy that allocates additional reasoning budget based on recent failure signals. In our experiments, we instantiate this framework using a two-level small–large model setting. Experiments on diverse multi-step agent benchmarks show that AgentCollab consistently improves the accuracy–efficiency Pareto frontier of LLM agents.
Search
Fix author
Co-authors
- Wenbo Gao 2
- Xiaosong Li 2
- Weizhe Lin 2
- Renxi Liu 2
- Yunke Peng 2
- Xian Wang 2
- Yao Wang 2
- Shuai Yang 2
- Hui-Ling Zhen 2
- Chen xi 2
- Linghan Ai 1
- Chen Chen 1
- Hanting Chen 1
- Yonghan Dong 1
- Jinyang Du 1
- Guo Fang 1
- Ruihao Gong 1
- Jinyang Guo 1
- Weichen Jia 1
- Boao Kong 1
- Guohong Li 1
- Xiaoguang Li 1
- Yiming Li 1
- Dahua Lin 1
- Xianglong Liu 1
- Shengyin Sun 1
- Zining Wang 1
- Wxuefei 1
- Lei Yan 1
- Xianzhi Yu 1
- Kun Yuan 1
- Engao Zhang 1