Jiakang Wang
Author directory2026
Rethinking Diffusion Large Language Models via Time-aware Weighting and Block-oriented Refinements
Zheyong Xie | Shengrui Li | Li Zhong | Kaiyan Zhao | Fangcheng Shi | Jiakang Wang | Huayi Jin | Zhaokai Luo | Fei Zhao | Haifeng Liu | Jieying Ye | Chonggang Lu | Wenxuan Huang | Yue Liu | Zijie Meng | Shaosheng Cao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Zheyong Xie | Shengrui Li | Li Zhong | Kaiyan Zhao | Fangcheng Shi | Jiakang Wang | Huayi Jin | Zhaokai Luo | Fei Zhao | Haifeng Liu | Jieying Ye | Chonggang Lu | Wenxuan Huang | Yue Liu | Zijie Meng | Shaosheng Cao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Diffusion Large Language Models (DLLMs) have recently emerged as a parallel alternative to autoregressive models, offering bidirectional context modeling and parallel decoding. However, developing highly performant DLLMs is hindered by lifecycle challenges, including suboptimal timestep weighting in pre-training, train-inference distribution gaps in post-training, and underutilized caching during inference. To address these obstacles, we introduce ReDLLM, a 30B-parameter Mixture-of-Experts DLLM. It incorporates a time-aware weighting strategy during continual pre-training that scales gradients according to denoising difficulty. Furthermore, we propose two block-oriented refinements: a terminal-block treatment to narrow the train-inference gap, and a block-aware inference pipeline featuring inline block prefill and block-level scheduling to maximize KV-cache reuse and parallelization. Extensive evaluations demonstrate that ReDLLM surpasses existing DLLMs at the same active-parameter scale, achieves generation quality competitive with 100B DLLMs, and attains the highest inference throughput among evaluated models.
Robust Tool Use via Fission-GRPO: Learning to Recover from Execution Errors
Zhiwei Zhang | Fei Zhao | Rui Wang | Zezhong Wang | Bin Liang | Jiakang Wang | Yao Hu | Shaosheng Cao | Kam-Fai Wong
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Zhiwei Zhang | Fei Zhao | Rui Wang | Zezhong Wang | Bin Liang | Jiakang Wang | Yao Hu | Shaosheng Cao | Kam-Fai Wong
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Large language models (LLMs) can call tools effectively, yet they remain brittle in multi-turn execution: after a tool-call error, smaller models often fall into repetitive invalid re-invocations instead of interpreting the feedback and recovering. This failure mode persists because current training paradigms do not explicitly teach models how to recover from execution errors. In particular, standard reinforcement learning (RL) collapses rich failure experience into sparse negative rewards, while pre-collected error-correction datasets become mismatched to the policy’s evolving failure modes. To bridge this gap, we propose Fission-GRPO, a framework that converts execution errors into on-policy corrective supervision within the RL training loop. Our core mechanism fissions each failed trajectory into a new training instance by augmenting it with diagnostic feedback from a fine-tuned Error Simulator, then resampling multiple recovery rollouts on-policy. This enables the model to learn from the precise errors it makes during exploration, rather than from static, pre-collected error cases. On BFCL v4 Multi-Turn, Fission-GRPO improves the error recovery rate of Qwen3-8B by 5.7% absolute and overall accuracy by 4.0% (from 42.75% to 46.75%), outperforming both RL baselines and specialized tool-use agents. The method further generalizes to TAU-Bench and TAU2-Bench, achieving leading results across most settings with gains up to +17.4%.