Zhi Wang
Author directoryOther people with similar names: Zhi Wang, Zhi Wang, Zhi Wang
2026
LazyMuon: Exploiting Angular Stability to Reduce Newton-Schulz Overhead in Muon for Efficient Large Model Training
Kainan Lin | Rongwei Lu | Haotian Dong | Le Chen | Jingyan Jiang | Huan Yang | Zhi Wang
Findings of the Association for Computational Linguistics: EMNLP 2026
Kainan Lin | Rongwei Lu | Haotian Dong | Le Chen | Jingyan Jiang | Huan Yang | Zhi Wang
Findings of the Association for Computational Linguistics: EMNLP 2026
Recently, the Muon optimizer has demonstrated strong efficiency in large model training through matrix orthogonalization. At each optimization step, Muon orthogonalizes the raw update matrix via Newton-Schulz (NS) iterations before applying it to model parameters. However, orthogonalization requires repeated matrix multiplications whose computational cost scales with O(n3) for an n× n matrix, making the overhead prohibitive at large model scales. In this paper, we observe that the geometric angle derived from cosine similarity between consecutive orthogonal update matrices remains highly stable during training, a phenomenon that we partially explain theoretically and support empirically. Based on this angular stability, we propose LazyMuon, which skips computationally expensive NS iterations through intermittent linear interpolation and employs a residual correction mechanism to mitigate accumulated approximation errors. This design ensures fast iteration while preserving Muon’s orthogonal manifold constraints. Empirical evaluations on language and vision tasks demonstrate that LazyMuon achieves speedups of up to 1.34× and 1.31× over Muon, respectively, with negligible degradation in convergence.
Semantic-Space Exploration and Exploitation in RLVR for LLM Reasoning
Fanding Huang | Guanbo Huang | Xiao Fan | Yi He | Xiao Liang | Xiao Chen | Qinting Jiang | Faisal Nadeem Khan | Jingyan Jiang | Zhi Wang
Findings of the Association for Computational Linguistics: ACL 2026
Fanding Huang | Guanbo Huang | Xiao Fan | Yi He | Xiao Liang | Xiao Chen | Qinting Jiang | Faisal Nadeem Khan | Jingyan Jiang | Zhi Wang
Findings of the Association for Computational Linguistics: ACL 2026
Reinforcement Learning with Verifiable Rewards (RLVR) for LLM reasoning is often framed as balancing exploration and exploitation in action space, typically operationalized with token-level proxies (e.g., output entropy or confidence). We argue that this apparent trade-off is largely a measurement artifact: token-level statistics reflect next-token uncertainty rather than how reasoning progresses over multi-token semantic structures. We therefore study exploration and exploitation in the hidden-state space of response trajectories. We use Effective Rank (ER) to quantify representational exploration and introduce its temporal derivatives, Effective Rank Velocity (ERV) and Effective Rank Acceleration (ERA), to characterize exploitative refinement dynamics. Empirically and theoretically, ER and ERV exhibit near-zero correlation in semantic space, suggesting the two capacities can be improved simultaneously. Motivated by this, we propose Velocity-Exploiting Rank Learning (VERL), which shapes the RL advantage with an auxiliary signal derived from ER/ERV and uses the more stable ERA as a meta-control variable to adaptively balance the incentives. Across multiple base models, RL algorithms, and reasoning benchmarks, VERL yields consistent improvements, including large gains on challenging tasks (e.g., 21.4% in Gaokao 2024).
HDSL: A Hierarchical Domain-Specific Language for Structured 3D Indoor Scene Generation and Localized Editing with LLM Agents
Letian Li | Chao Shen | Shuzhao Xie | Chenghao Gu | ZhengXiao He | Yu Meng | Xin Yang | Wenyuan Jiang | Zhi Wang
Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers)
Letian Li | Chao Shen | Shuzhao Xie | Chenghao Gu | ZhengXiao He | Yu Meng | Xin Yang | Wenyuan Jiang | Zhi Wang
Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers)
Text-driven indoor scene generation and editing require an intermediate representation that language models can both produce and revise. Existing LLM-based systems often rely on scene graphs or global constraint lists, which are compact but underspecify local geometry and make instruction-based edits difficult to localize. We frame the task as structured program generation and local program repair, and propose the Hierarchical Descriptive Scene Language (HDSL), an XML/CSS-style domain-specific language in which rooms, regions, objects, and support surfaces form a tree with parent-relative local coordinates. Our pipeline generates HDSL subtrees recursively with bounded verification, grounds non-virtual nodes through multimodal asset retrieval, and applies force-directed layout optimization to repair boundary and collision errors. For editing, Hierarchical Retrieval-Augmented Generation retrieves the relevant subtree, rewrites only that local context, and merges the result back through a deterministic three-way merge. In our reproduced benchmark, HDSL improves average object coverage, text-scene alignment, and generation time over full text-to-scene baselines while remaining competitive with layout-only reproductions on geometry metrics. On thirty paired editing instructions, HRAG reduces token use by 67.6% and runtime by 68.1%, and cuts non-target removals by 87.5% over full-scene rewriting; attribute ablations confirm that local coordinates and virtual/leaf flags are necessary for valid, locality-preserving edits.
2025
Beyond A Single AI Cluster: A Survey of Decentralized LLM Training
Haotian Dong | Jingyan Jiang | Rongwei Lu | Jiajun Luo | Jiajun Song | Bowen Li | Ying Shen | Zhi Wang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Haotian Dong | Jingyan Jiang | Rongwei Lu | Jiajun Luo | Jiajun Song | Bowen Li | Ying Shen | Zhi Wang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
The emergence of large language models (LLMs) has revolutionized AI development, yet their resource demands beyond a single cluster or even datacenter, limiting accessibility to well-resourced organizations. Decentralized training has emerged as a promising paradigm to leverage dispersed resources across clusters, datacenters and even regions, offering the potential to democratize LLM development for broader communities. As the first comprehensive exploration of this emerging field, we present decentralized LLM training as a resource-driven paradigm and categorize existing efforts into community-driven and organizational approaches. We further clarify this through: (1) a comparison with related paradigms, (2) characterization of decentralized resources, and (3) a taxonomy of recent advancements. We also provide up-to-date case studies and outline future directions to advance research in decentralized LLM training.
2022
bert2BERT: Towards Reusable Pretrained Language Models
Cheng Chen | Yichun Yin | Lifeng Shang | Xin Jiang | Yujia Qin | Fengyu Wang | Zhi Wang | Xiao Chen | Zhiyuan Liu | Qun Liu
Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Cheng Chen | Yichun Yin | Lifeng Shang | Xin Jiang | Yujia Qin | Fengyu Wang | Zhi Wang | Xiao Chen | Zhiyuan Liu | Qun Liu
Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
In recent years, researchers tend to pre-train ever-larger language models to explore the upper limit of deep models. However, large language model pre-training costs intensive computational resources, and most of the models are trained from scratch without reusing the existing pre-trained models, which is wasteful. In this paper, we propose bert2BERT, which can effectively transfer the knowledge of an existing smaller pre-trained model to a large model through parameter initialization and significantly improve the pre-training efficiency of the large model. Specifically, we extend the previous function-preserving method proposed in computer vision on the Transformer-based language model, and further improve it by proposing a novel method, advanced knowledge for large model’s initialization. In addition, a two-stage learning method is proposed to further accelerate the pre-training. We conduct extensive experiments on representative PLMs (e.g., BERT and GPT) and demonstrate that (1) our method can save a significant amount of training cost compared with baselines including learning from scratch, StackBERT and MSLT; (2) our method is generic and applicable to different types of pre-trained models. In particular, bert2BERT saves about 45% and 47% computational cost of pre-training BERT BASE and GPT BASE by reusing the models of almost their half sizes.
Search
Fix author
Co-authors
- Jingyan Jiang 3
- Haotian Dong 2
- Rongwei Lu 2
- Cheng Chen 1
- Le Chen 1
- Xiao Chen 1
- Xiao Chen 1
- Xiao Fan 1
- Chenghao Gu 1
- Yi He 1
- ZhengXiao He 1
- Fanding Huang 1
- Guanbo Huang 1
- Qinting Jiang 1
- Wenyuan Jiang 1
- Xin Jiang 1
- Faisal Nadeem Khan 1
- Bowen Li 1
- Letian Li 1
- Xiao Liang (梁霄) 1
- Kainan Lin 1
- Qun Liu 1
- Zhiyuan Liu 1
- Jiajun Luo 1
- Yu Meng 1
- Yujia Qin 1
- Lifeng Shang 1
- Chao Shen 1
- Ying Shen 1
- Jiajun Song 1
- Fengyu Wang 1
- Shuzhao Xie 1
- Huan Yang 1
- Xin Yang 1
- Yichun Yin 1