Tongyao Zhu
Author directory2026
From Harm to Help: Turning Reasoning In-Context Demos into Assets for Reasoning LMs
Weida Liang | Zihang Fu | Haonan Wang | Zheng Nie | Yifan Zhang | Yao Tong | Tongyao Zhu | Hao Jiang | Chuang Li | Jiaying Wu | Kenji Kawaguchi
Findings of the Association for Computational Linguistics: EMNLP 2026
Weida Liang | Zihang Fu | Haonan Wang | Zheng Nie | Yifan Zhang | Yao Tong | Tongyao Zhu | Hao Jiang | Chuang Li | Jiaying Wu | Kenji Kawaguchi
Findings of the Association for Computational Linguistics: EMNLP 2026
Recent reasoning LLMs (RLMs), especially those trained with verifier-based reinforcement learning, often perform worse with few-shot CoT than with direct answering. We revisit this paradox using high-quality reasoning traces from DeepSeek-R1 as demonstrations and find that adding more exemplars consistently degrades accuracy, even when demonstrations are optimal. A detailed analysis reveals two mechanisms behind this decline: (i) *semantic misguidance*, where high textual similarity leads the model to treat the target as the same as the exemplar and to copy intermediate steps verbatim; and (ii) *strategy transfer failure*, where the model struggles to extract useful reasoning strategies and apply them to target questions. Guided by these findings, we introduce Insight-to-Solve (I2S), which converts demonstrations into target-conditioned insights before solving, and I2S+, which adds iterative refinement. Across open- and closed-source models, both variants outperform direct answering and sequential test-time baselines. On a frozen 500-problem Omni-MATH transfer set, I2S+ improves R1-Distill-Qwen-7B and Qwen3-14B by +4.2 and +4.0 points, respectively. The same prompt stack also transfers to GPT-4.1, o1-mini, and R1-Distill-Llama-8B, showing that reasoning demonstrations can be converted from liabilities into useful test-time guidance.
Why Do LLM Agents Fail in Exploring New Environments? A World-Modeling Perspective
Shiqi Chen | Tongyao Zhu | Zian Wang | Jinghan Zhang | Kangrui Wang | Ruochen Zhou | Siyang Gao | Teng Xiao | Yee Whye Teh | Junxian He | Manling Li
Findings of the Association for Computational Linguistics: EMNLP 2026
Shiqi Chen | Tongyao Zhu | Zian Wang | Jinghan Zhang | Kangrui Wang | Ruochen Zhou | Siyang Gao | Teng Xiao | Yee Whye Teh | Junxian He | Manling Li
Findings of the Association for Computational Linguistics: EMNLP 2026
Large Language Models (LLMs) as agents often fail to improve in new environments. We identify and characterize a failure mode we call exploration collapse: under reinforcement learning (RL) in environments whose states are unfamiliar to the policy, Pass@k, the probability that at least one of k sampled trajectories succeeds, drops markedly over training even as Pass@1 edges up, revealing increasingly brittle exploration; environments closer to the pretraining distribution show no such decline. We trace this collapse to weak grounding in environment states and dynamics, and study a simple remedy: explicitly teaching the agent to estimate the current state and predict its transitions before optimizing for reward. We instantiate it as SPA, an explore-then-exploit recipe that cold-starts the policy with a Self-Experience supervised finetuning (SFT) stage, collecting the model’s own interaction trajectories and supervising state and next-state prediction, and then runs standard RL. The resulting world model serves as a grounded initialization for RL rather than an inference-time planner. Across unseen environments, SPA consistently and substantially improves over vanilla RL: for example, it raises the Sokoban success rate from 25.6% to 59.8% on Qwen2.5-1.5B-Instruct, letting sub-3B models surpass a 20B baseline on these tasks. Controlled studies indicate that the gains track four factors: grounded state representations, explicit transition modeling, self-experience trajectories from a sufficiently strong exploration policy, and adequate coverage of transition data.
When Models Edit Too Much: On the Fidelity of Minimal Code Edits
Tongyao Zhu | Lim Wei Hern | Min-Yen Kan
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Tongyao Zhu | Lim Wei Hern | Min-Yen Kan
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Large language models (LLMs) are increasingly used to edit existing code, but correctness alone is not enough: useful repairs should also be minimal, reviewable, and faithful to the original implementation. We study over-editing, the tendency of a model to rewrite code beyond what is required to fix a bug. We construct an evaluation framework from 400 BigCodeBench problems by injecting controlled AST-level corruptions into reference solutions, giving each repair task a known minimal patch. Across frontier LLMs, over-editing is widespread even among strong models like GPT-5.5: high Pass@1 can coexist with unnecessarily large edits and added cognitive complexity. A preservation instruction substantially reduces this behavior, lowering average excess Levenshtein distance from 0.195 to 0.131, reducing added cognitive complexity by 26.6%, and increasing Pass@1 by 2.3 points. However, these gains do not simply follow from a larger reasoning budget or larger models. We next ask whether minimal editing can be learned directly during post-training. We observe that supervised fine-tuning overfits to seen corruption patterns, whereas reinforcement learning gives the best out-of-domain edit-fidelity and performance-retention trade-off. These results position edit fidelity as a distinct axis of code-repair quality and show that it can be measured and learned.
When Does Mixing Help? Analyzing Query Embedding Interpolation in Multilingual Dense Retrieval
Tongyao Zhu | Huang Chao Ming | Min-Yen Kan
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Tongyao Zhu | Huang Chao Ming | Min-Yen Kan
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
While mixed-language querying is ubiquitous in multilingual communities, the sensitivity of dense retrievers to such queries remains poorly understood. We present a ratio-controlled study on mMARCO that systematically evaluates retrieval performance by varying the mixing proportion of parallel query translations via embedding-level mixing—constructing mixed queries as an interpolation of monolingual embeddings. Experiments with BGE-M3 demonstrate that an optimal mixing ratio outperforms the best monolingual endpoint in 88/105 cases. We uncover a distinct asymmetry driven by English dominance: mixing is uniformly beneficial when retrieving from non-English document indices, whereas indices containing English are best served by pure English queries. Furthermore, English acts as the strongest mixing partner for every non-English document language. Finally, when controlling for English dominance, mixing gains correlate negatively with typological distance. We conclude that language-mix sensitivity is structured and predictable, and we validate the robustness of these patterns across model families and scales.
How Can Synthetic Data Improve Multilingual Language Model Pretraining? A Data Quality Perspective
Tongyao Zhu | Qian Liu | Chang Ma | Jinghan Zhang | Longxu Dou | Junxian He | Shiqi Chen
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Tongyao Zhu | Qian Liu | Chang Ma | Jinghan Zhang | Longxu Dou | Junxian He | Shiqi Chen
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Low-resource languages challenge multilingual LLMs due to limited high-quality training data, leading to weaker performance on complex reasoning and knowledge tasks. To address this, we propose improving training data quality through data synthesis, moving beyond simple resource scaling. First, we introduce SynTrans, which translates high-quality, knowledge-rich English data into low-resource languages during pre-training to inject world knowledge, though at the cost of semantic fluency. To overcome low-quality data issues while maintaining fluency, we also propose SynRank. SynRank leverages synthetic data as positive samples to train a classifier that ranks and filters noisy real-world data, enabling the extraction of high-quality subsets without expensive human cleaning. Experiments show SynRank matches handcrafted rule-based filtering by human experts and significantly improves knowledge-intensive task performance at the same filtering rate. Remarkably, higher filtering rates even improve performance with less data, demonstrating the efficiency and effectiveness of our method, surpassing expert filtering. Lastly, we introduce DA-QwenScore, a training-free metric that evaluates corpus quality by normalizing model loss with diversity measures, further enhancing evaluation efficiency. Our insights into knowledge injection could advance low-resource multilingual LLM development.
2024
Beyond Memorization: The Challenge of Random Memory Access in Language Models
Tongyao Zhu | Qian Liu | Liang Pang | Zhengbao Jiang | Min-Yen Kan | Min Lin
Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Tongyao Zhu | Qian Liu | Liang Pang | Zhengbao Jiang | Min-Yen Kan | Min Lin
Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Recent developments in Language Models (LMs) have shown their effectiveness in NLP tasks, particularly in knowledge-intensive tasks.However, the mechanisms underlying knowledge storage and memory access within their parameters remain elusive. In this paper, we investigate whether a generative LM (e.g., GPT-2) is able to access its memory sequentially or randomly. Through carefully-designed synthetic tasks, covering the scenarios of full recitation, selective recitation and grounded question answering, we reveal that LMs manage to sequentially access their memory while encountering challenges in randomly accessing memorized content. We find that techniques including recitation and permutation improve the random memory access capability of LMs. Furthermore, by applying this intervention to realistic scenarios of open-domain question answering, we validate that enhancing random access by recitation leads to notable improvements in question answering. The code to reproduce our experiments can be found at https://github.com/sail-sg/lm-random-memory-access.
Search
Fix author
Co-authors
- Min-Yen Kan 3
- Shiqi Chen 2
- Junxian He 2
- Qian Liu 2
- Jinghan Zhang 2
- Longxu Dou 1
- Zihang Fu 1
- Siyang Gao 1
- Lim Wei Hern 1
- Hao Jiang 1
- Zhengbao Jiang 1
- Kenji Kawaguchi 1
- Chuang Li 1
- Manling Li 1
- Weida Liang 1
- Min Lin 1
- Chang Ma 1
- Huang Chao Ming 1
- Zheng Nie 1
- Liang Pang (庞亮) 1
- Yee Whye Teh 1
- Yao Tong 1
- Haonan Wang 1
- Kangrui Wang 1
- Zian Wang 1
- Jiaying Wu 1
- Teng Xiao 1
- Yifan Zhang 1
- Ruochen Zhou 1