Zijian Zhou
Author directory2026
When Can Large Reasoning Models Save Thinking? Mechanistic Analysis of Behavioral Divergence in Reasoning
Rongzhi Zhu | Yi Liu | Jiancheng Wang | Xiangyu Liu | Zequn Sun | Yiwei Wang | Yu Deng | Zijian Zhou | Wei Hu
Findings of the Association for Computational Linguistics: EMNLP 2026
Rongzhi Zhu | Yi Liu | Jiancheng Wang | Xiangyu Liu | Zequn Sun | Yiwei Wang | Yu Deng | Zijian Zhou | Wei Hu
Findings of the Association for Computational Linguistics: EMNLP 2026
Large reasoning models (LRMs) have achieved remarkable success on complex tasks, yet their tendency to "overthink” leads to inefficiencies. Although "save-thinking” prompts are intended to mitigate this issue, we find that LRMs still frequently enter the "Still-thinking” mode instead of the expected "No-thinking” mode, especially when facing difficult queries. First, we examine the LRM’s confidence when terminating thinking process, finding that high perplexity signals internal uncertainty and precedes the thinking process. Second, we analyze the divergence in internal attention distributions across different LRMs. Third, we investigate where the LRM allocates its attention within the prompt, identifying a critical "re-reading” pattern. Based on these analyses, we propose an attention intervention method to regulate this behavior. While this intervention successfully suppresses thinking, we find it causes a drop in accuracy, confirming that the thinking process is functional. Our work provides a mechanistic explanation for this behavior, highlighting the trade-off between instruction following and reasoning correctness, and offers insights for developing more controllable LRMs.
2025
Uncovering Scaling Laws for Large Language Models via Inverse Problems
Arun Verma | Zhaoxuan Wu | Zijian Zhou | Xiaoqiang Lin | Zhiliang Chen | Rachael Hwee Ling Sim | Rui Qiao | Jingtan Wang | Nhung Bui | Xinyuan Niu | Wenyang Hu | Gregory Kang Ruey Lau | Zi-Yu Khoo | Zitong Zhao | Xinyi Xu | Apivich Hemachandra | See-Kiong Ng | Bryan Kian Hsiang Low
Findings of the Association for Computational Linguistics: EMNLP 2025
Arun Verma | Zhaoxuan Wu | Zijian Zhou | Xiaoqiang Lin | Zhiliang Chen | Rachael Hwee Ling Sim | Rui Qiao | Jingtan Wang | Nhung Bui | Xinyuan Niu | Wenyang Hu | Gregory Kang Ruey Lau | Zi-Yu Khoo | Zitong Zhao | Xinyi Xu | Apivich Hemachandra | See-Kiong Ng | Bryan Kian Hsiang Low
Findings of the Association for Computational Linguistics: EMNLP 2025
Large Language Models (LLMs) are large-scale pretrained models that have achieved remarkable success across diverse domains. These successes have been driven by unprecedented complexity and scale in both data and computations. However, due to the high costs of training such models, brute-force trial-and-error approaches to improve LLMs are not feasible. Inspired by the success of inverse problems in uncovering fundamental scientific laws, this position paper advocates that inverse problems can also efficiently uncover scaling laws that guide the building of LLMs to achieve the desirable performance with significantly better cost-effectiveness.
TETRIS: Optimal Draft Token Selection for Batch Speculative Decoding
Zhaoxuan Wu | Zijian Zhou | Arun Verma | Alok Prakash | Daniela Rus | Bryan Kian Hsiang Low
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Zhaoxuan Wu | Zijian Zhou | Arun Verma | Alok Prakash | Daniela Rus | Bryan Kian Hsiang Low
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
We propose TETRIS, a novel method that optimizes the total throughput of batch speculative decoding in multi-request settings. Unlike existing methods that optimize for a single request or a group of requests as a whole, TETRIS actively selects the most promising draft tokens (for every request in a batch) to be accepted when verified in parallel, resulting in fewer rejected tokens and hence less wasted computing resources. Such an effective resource utilization to achieve fast inference in large language models (LLMs) is especially important to service providers with limited inference capacity. Compared to baseline speculative decoding, TETRIS yields a consistently higher acceptance rate and more effective utilization of the limited inference capacity. We show theoretically and empirically that TETRIS outperforms baseline speculative decoding and existing methods that dynamically select draft tokens, leading to a more efficient batch inference in LLMs.
2024
Position Paper: Data-Centric AI in the Age of Large Language Models
Xinyi Xu | Zhaoxuan Wu | Rui Qiao | Arun Verma | Yao Shu | Jingtan Wang | Xinyuan Niu | Zhenfeng He | Jiangwei Chen | Zijian Zhou | Gregory Kang Ruey Lau | Hieu Dao | Lucas Agussurja | Rachael Hwee Ling Sim | Xiaoqiang Lin | Wenyang Hu | Zhongxiang Dai | Pang Wei Koh | Bryan Kian Hsiang Low
Findings of the Association for Computational Linguistics: EMNLP 2024
Xinyi Xu | Zhaoxuan Wu | Rui Qiao | Arun Verma | Yao Shu | Jingtan Wang | Xinyuan Niu | Zhenfeng He | Jiangwei Chen | Zijian Zhou | Gregory Kang Ruey Lau | Hieu Dao | Lucas Agussurja | Rachael Hwee Ling Sim | Xiaoqiang Lin | Wenyang Hu | Zhongxiang Dai | Pang Wei Koh | Bryan Kian Hsiang Low
Findings of the Association for Computational Linguistics: EMNLP 2024
This position paper proposes a data-centric viewpoint of AI research, focusing on large language models (LLMs). We start by making a key observation that data is instrumental in the developmental (e.g., pretraining and fine-tuning) and inferential stages (e.g., in-context learning) of LLMs, and advocate that data-centric research should receive more attention from the community. We identify four specific scenarios centered around data, covering data-centric benchmarks and data curation, data attribution, knowledge transfer, and inference contextualization. In each scenario, we underscore the importance of data, highlight promising research directions, and articulate the potential impacts on the research community and, where applicable, the society as a whole. For instance, we advocate for a suite of data-centric benchmarks tailored to the scale and complexity of data for LLMs. These benchmarks can be used to develop new data curation methods and document research efforts and results, which can help promote openness and transparency in AI and LLM research.
Search
Fix author
Co-authors
- Bryan Kian Hsiang Low 3
- Arun Verma 3
- Zhaoxuan Wu 3
- Wenyang Hu 2
- Gregory Kang Ruey Lau 2
- Xiaoqiang Lin 2
- Xinyuan Niu 2
- Rui Qiao 2
- Rachael Hwee Ling Sim 2
- Jingtan Wang 2
- Xinyi Xu 2
- Lucas Agussurja 1
- Nhung Bui 1
- Jiangwei Chen 1
- Zhiliang Chen 1
- Zhongxiang Dai 1
- Hieu Dao 1
- Yu Deng 1
- Zhenfeng He 1
- Apivich Hemachandra 1
- Wei Hu 1
- Zi-Yu Khoo 1
- Pang Wei Koh 1
- Xiangyu Liu 1
- Yi Liu 1
- See Kiong Ng 1
- Alok Prakash 1
- Daniela Rus 1
- Yao Shu 1
- Zequn Sun 1
- Jiancheng Wang 1
- Yiwei Wang 1
- Zitong Zhao 1
- Rongzhi Zhu 1