Wen Wang
Author directoryOther people with similar names: Wen Wang (王雯), Wen Wang, Wen Wang
2026
Streaming Communication in Multi-Agent Reasoning
Zhen Yang | Xiaogang Xu | Wen Wang | Cong Chen | Xander Xu | Ying-Cong Chen
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Zhen Yang | Xiaogang Xu | Wen Wang | Cong Chen | Xander Xu | Ying-Cong Chen
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Multi-agent reasoning systems adopt a generate-then-transfer paradigm that forces end-to-end latency to scale linearly with pipeline depth. We introduce StreamMA, a multi-agent reasoning system that streams each reasoning step to downstream agents as soon as it is generated, pipelining adjacent agents and thus reducing latency. Surprisingly, this pipelining also improves effectiveness: because multi-step reasoning quality is non-uniform and early steps are more reliable than later ones, working with these reliable early steps instead of the full chain prevents error-prone late steps from misleading downstream agents. We formalize both advantages with the first closed-form joint analysis of stream, serial, and single protocols, deriving the effectiveness ordering, speedup upper bound, and cost ratio. Across eight reasoning benchmarks spanning mathematics, science, and code, three frontier LLMs (Claude Opus 4.6, GPT-5.4, and GLM-5.2), and three topologies (Chain, Tree, Graph), StreamMA outperforms both baselines (avg.+7.3pp, max +22.4pp on HMMT2026; Claude Opus 4.6-High). Beyond these contributions, we discover a step-level scaling law: increasing per-agent steps consistently improves both effectiveness and efficiency, a new scaling dimension orthogonal to and composable with agent-count scaling.
Improving Diffusion Language Model Decoding through Joint Search in Generation Order and Token Space
Yangyi Shen | Tianjian Feng | Jiaqi Han | Wen Wang | Tianlang Chen | Chunhua Shen | Jure Leskovec | Stefano Ermon
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Yangyi Shen | Tianjian Feng | Jiaqi Han | Wen Wang | Tianlang Chen | Chunhua Shen | Jure Leskovec | Stefano Ermon
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Diffusion Language Models (DLMs) offer order-agnostic generation that can explore many possible decoding trajectories. However, current decoding methods commit to a single trajectory, limiting exploration in trajectory space. We introduce Order-Token Search to explore this space through jointly searching over generation order and token values. Its core is a likelihood estimator that scores denoising actions, enabling stable pruning and efficient exploration of diverse trajectories. Across mathematical reasoning and coding benchmarks, Order-Token Search consistently outperforms baselines on GSM8K, MATH500, Countdown, and HumanEval (3.1%, 3.8%, 7.9%, and 6.8% absolute over backbone), reaching accuracy comparable to diffu-GRPO post-training. Our work establishes joint search as a key component for advancing decoding in DLMs.
Beyond Hard Masks: Progressive Token Evolution for Diffusion Language Models
Linhao Zhong | Linyu Wu | Bozhen Fang | Tianjian Feng | Chenchen Jing | Wen Wang | Jiaheng Zhang | Hao Chen | Chunhua Shen
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Linhao Zhong | Linyu Wu | Bozhen Fang | Tianjian Feng | Chenchen Jing | Wen Wang | Jiaheng Zhang | Hao Chen | Chunhua Shen
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Diffusion Language Models (DLMs) offer a promising alternative for language modeling by enabling parallel decoding through iterative refinement. However, most DLMs rely on hard binary masking and discrete token assignments, which hinder the revision of early decisions and underutilize intermediate probabilistic representations. In this paper, we propose EvoToken-DLM, a novel diffusion-based language modeling approach that replaces hard binary masks with evolving soft token distributions. EvoToken-DLM enables a progressive transition from masked states to discrete outputs, supporting revisable decoding. To effectively support this evolution, we introduce continuous trajectory supervision, which aligns training objectives with iterative probabilistic updates. Extensive experiments across multiple benchmarks show that EvoToken-DLM consistently achieves superior performance, outperforming strong diffusion-based and masked DLM baselines. Our code is available at https://github.com/aim-uofa/EvoTokenDLM.
Efficient Self-Evaluation for Diffusion Language Models via Sequence Regeneration
Linhao Zhong | Linyu Wu | Wen Wang | Yuling Xi | Chenchen Jing | Jiaheng Zhang | Hao Chen | Chunhua Shen
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Linhao Zhong | Linyu Wu | Wen Wang | Yuling Xi | Chenchen Jing | Jiaheng Zhang | Hao Chen | Chunhua Shen
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Diffusion large language models (dLLMs) have recently attracted significant attention for their ability to enhance diversity, controllability, and parallelism. However, their non-sequential, bidirectionally masked generation makes quality assessment difficult, underscoring the need for effective self-evaluation. In this work, we propose DiSE, a simple yet effective self-evaluation confidence quantification method for dLLMs. DiSE quantifies confidence by computing the probability of regenerating the tokens in the entire generated sequence, given the full context. This method enables more efficient and reliable quality assessment by leveraging token regeneration probabilities, facilitating both likelihood estimation and robust uncertainty quantification. Building upon DiSE, we further introduce a flexible-length generation framework, which adaptively controls the sequence length based on the model’s self-assessment of its own output. We analyze and validate the feasibility of DiSE from the perspective of dLLM generalization, and empirically demonstrate that DiSE is positively correlated with both semantic coherence and answer accuracy. Extensive experiments on likelihood evaluation, uncertainty quantification, and flexible-length generation further confirm the effectiveness of the proposed DiSE.