Junhao Ruan
Author directory2026
Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence
Xinyu Liu | Kechen Jiao | Chunyang Xiao | Runsong Zhao | Junhao Ruan | Bei Li | Jiahao Liu | Qifan Wang | Xin Chen | Jingang Wang | Chenglong Wang | Tong Xiao | JingBo Zhu
Findings of the Association for Computational Linguistics: EMNLP 2026
Xinyu Liu | Kechen Jiao | Chunyang Xiao | Runsong Zhao | Junhao Ruan | Bei Li | Jiahao Liu | Qifan Wang | Xin Chen | Jingang Wang | Chenglong Wang | Tong Xiao | JingBo Zhu
Findings of the Association for Computational Linguistics: EMNLP 2026
On-policy distillation (OPD) has emerged as a promising paradigm for reasoning-oriented post-training of large language models (LLMs), especially when combined with reinforcement learning from verifiable rewards (RLVR). Existing OPD methods rely on reverse KL (RKL)-based teacher supervision over trajectories sampled from the student policy. However, we identify a critical limitation: under large teacher–student policy divergence, RL-driven exploration often produces trajectories that the teacher assigns low probability, resulting in uninformative negative feedback. To address this, we propose Teacher-Guided Policy Optimization (TGPO). Unlike existing OPD methods that evaluate student-sampled actions using the teacher, TGPO directly guides next-token generation using the teacher’s predicted distribution on student-generated contexts, yielding an on-policy forward-KL objective. Combined with RLVR, this guidance steers exploration toward improved continuations. Experiments on reasoning benchmarks show that TGPO outperforms existing RKL-based OPD methods and remains robust across different teacher models. Furthermore, we find that teacher top-1 predictions suffice for effective guidance, without requiring the full teacher distribution.
HybridThinker: Efficient Chain-of-Thought Reasoning via Compressed Memory and Transient Thought Steps
Xin Liu | Runsong Zhao | Xinyu Liu | Junhao Ruan | Pengcheng Huang | Shichao Dong | Chunyang Xiao | Chenglong Wang | Changliang Li | JingBo Zhu | Tong Xiao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Xin Liu | Runsong Zhao | Xinyu Liu | Junhao Ruan | Pengcheng Huang | Shichao Dong | Chunyang Xiao | Chenglong Wang | Changliang Li | JingBo Zhu | Tong Xiao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Extended chain-of-thought (CoT) traces improve LLM reasoning but incur substantial computational and memory costs. While existing CoT compression methods mitigate this by condensing thought steps into compact representations via memory tokens and retaining only these representations at inference time, the loss of fine-grained information makes subsequent steps more error-prone. To alleviate this, we propose HybridThinker, where in addition to preserving these representations, thought steps are also temporarily retained to provide fine-grained details. However, we observe that naively keeping thought steps accessible to subsequent steps during training lets the model bypass memory tokens by retrieving information directly from these steps, leaving the model’s ability to compress and retrieve information through memory tokens insufficiently trained. We therefore introduce a hybrid training scheme, in which only some thought steps are directly accessible through attention to subsequent steps, while the other thought steps are masked, forcing the model to use memory tokens for compression and retrieval. Across four reasoning benchmarks, HybridThinker matches Vanilla on Qwen2.5-7B and approaches it on Llama3.1-8B, improving over LightThinker by 5.8 and 2.7 points, respectively, with similar inference time. Ablation studies confirm that both temporary thought-step retention and the hybrid training scheme contribute to these gains.
ToFu: A White-Box, Token-Efficient Agent Harness for Researchers
Junhao Ruan | Yuan Ge | Bei Li | Yongjing Yin | Yuchun Fan | Xin Chen | Jingang Wang | Chenglong Wang | JingBo Zhu | Tong Xiao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: System Demonstrations
Junhao Ruan | Yuan Ge | Bei Li | Yongjing Yin | Yuchun Fan | Xin Chen | Jingang Wang | Chenglong Wang | JingBo Zhu | Tong Xiao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: System Demonstrations
Agentic coding tools present new opportunities to transform research workflows. The performance of agent systems built depends on both large language models (LLMs) and the harness around LLMs, which is the orchestration code that determines an agent’s behavior. We present ToFu, an agentic harness for researchers that reads your codebase, edits files, runs commands, and integrates with your development tools.
ToFu plays a dual role in research. As a research assistant, it supports practical research workflows with superior token efficiency, lower cost, and multilingual capability compared with existing agentic harnesses. Its release under the MIT License further enables local deployment for privacy-sensitive users. As a research object, ToFu provides a white-box agentic harness that allows researchers to inspect, modify, and evaluate its orchestration logic, tool-use behavior, and harness design, while retaining strong benchmark performance and an application-level user experience.
ToFu plays a dual role in research. As a research assistant, it supports practical research workflows with superior token efficiency, lower cost, and multilingual capability compared with existing agentic harnesses. Its release under the MIT License further enables local deployment for privacy-sensitive users. As a research object, ToFu provides a white-box agentic harness that allows researchers to inspect, modify, and evaluate its orchestration logic, tool-use behavior, and harness design, while retaining strong benchmark performance and an application-level user experience.
MTR-Suite: A Framework for Evaluating and Synthesizing Conversational Retrieval Benchmarks
Junhao Ruan | Abudukeyumu Abudula | Bei Li | Yongjing Yin | Xinyu Liu | Kechen Jiao | Xin Chen | Jingang Wang | Xunliang Cai | Tong Xiao | JingBo Zhu
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Junhao Ruan | Abudukeyumu Abudula | Bei Li | Yongjing Yin | Xinyu Liu | Kechen Jiao | Xin Chen | Jingang Wang | Xunliang Cai | Tong Xiao | JingBo Zhu
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Accurate evaluation of conversational retrieval is pivotal for advancing Retrieval-Augmented Generation (RAG) systems. However, existing conversational retrieval benchmarks suffer from costly, sparse human annotation or rigid, unnatural automated heuristics. To address these challenges, we introduce MTR-Suite, a unified framework for auditing, synthesizing, and benchmarking retrieval. It features: (1) MTR-Eval, an LLM-based auditor quantifying alignment gaps in previous benchmarks; (2) MTR-Pipeline, a multi-agent system using greedy traversal clustering to generate high-fidelity dialogues at 1/400th human cost; and (3) MTR-Bench, a rigorous general-domain benchmark. MTR-Bench mimics production-style challenges (hard topic switching, verbosity), offering superior discriminative power. We make our code and data publicly available to facilitate future research.
2025
TCPO: Thought-Centric Preference Optimization for Effective Embodied Decision-making
Kechen Jiao | Zhirui Fang | Jiahao Liu | Bei Li | Qifan Wang | Xinyu Liu | Junhao Ruan | Zhongjian Qiao | Yifan Zhu | Yaxin Xu | Jingang Wang | Xiu Li
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Kechen Jiao | Zhirui Fang | Jiahao Liu | Bei Li | Qifan Wang | Xinyu Liu | Junhao Ruan | Zhongjian Qiao | Yifan Zhu | Yaxin Xu | Jingang Wang | Xiu Li
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Using effective generalization capabilities of vision language models (VLMs) in context-specific dynamic tasks for embodied artificial intelligence remains a significant challenge. Although supervised fine-tuned models can better align with the real physical world, they still exhibit sluggish responses and hallucination issues in dynamically changing environments, necessitating further alignment. Existing post-SFT methods, reliant on reinforcement learning and chain-of-thought (CoT) approaches, are constrained by sparse rewards and action-only optimization, resulting in low sample efficiency, poor consistency, and model degradation. To address these issues, this paper proposes Thought-Centric Preference Optimization (TCPO) for effective embodied decision-making. Specifically, TCPO introduces a stepwise preference-based optimization approach, transforming sparse reward signals into richer step sample pairs. It emphasizes the alignment of the model’s intermediate reasoning process, mitigating the problem of model degradation. Moreover, by incorporating Action Policy Consistency Constraint (APC), it further imposes consistency constraints on the model output. Experiments in the ALFWorld environment demonstrate an average success rate of 26.67%, achieving a 6% improvement over RL4VLM and validating the effectiveness of our approach in mitigating model degradation after fine-tuning. These results highlight the potential of integrating preference-based learning techniques with CoT processes to enhance the decision-making capabilities of vision-language models in embodied agents.
IIET: Efficient Numerical Transformer via Implicit Iterative Euler Method
Xinyu Liu | Bei Li | Jiahao Liu | Junhao Ruan | Kechen Jiao | Hongyin Tang | Jingang Wang | Tong Xiao | JingBo Zhu
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Xinyu Liu | Bei Li | Jiahao Liu | Junhao Ruan | Kechen Jiao | Hongyin Tang | Jingang Wang | Tong Xiao | JingBo Zhu
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
High-order numerical methods enhance Transformer performance in tasks like NLP and CV, but introduce a performance-efficiency trade-off due to increased computational overhead. Our analysis reveals that conventional efficiency techniques, such as distillation, can be detrimental to the performance of these models, exemplified by PCformer. To explore more optimizable ODE-based Transformer architectures, we propose the Iterative Implicit Euler Transformer (IIET), which simplifies high-order methods using an iterative implicit Euler approach. This simplification not only leads to superior performance but also facilitates model compression compared to PCformer. To enhance inference efficiency, we introduce Iteration Influence-Aware Distillation (IIAD). Through a flexible threshold, IIAD allows users to effectively balance the performance-efficiency trade-off. On lm-evaluation-harness, IIET boosts average accuracy by 2.65% over vanilla Transformers and 0.8% over PCformer. Its efficient variant, E-IIET, significantly cuts inference overhead by 55% while retaining 99.4% of the original task accuracy. Moreover, the most efficient IIET variant achieves an average performance gain exceeding 1.6% over vanilla Transformer with comparable speed.
SLAM: Towards Efficient Multilingual Reasoning via Selective Language Alignment
Yuchun Fan | Yongyu Mu | YiLin Wang | Lei Huang | Junhao Ruan | Bei Li | Tong Xiao | Shujian Huang | Xiaocheng Feng | Jingbo Zhu
Proceedings of the 31st International Conference on Computational Linguistics
Yuchun Fan | Yongyu Mu | YiLin Wang | Lei Huang | Junhao Ruan | Bei Li | Tong Xiao | Shujian Huang | Xiaocheng Feng | Jingbo Zhu
Proceedings of the 31st International Conference on Computational Linguistics
Despite the significant improvements achieved by large language models (LLMs) in English reasoning tasks, these models continue to struggle with multilingual reasoning. Recent studies leverage a full-parameter and two-stage training paradigm to teach models to first understand non-English questions and then reason. However, this method suffers from both substantial computational resource computing and catastrophic forgetting. The fundamental cause is that, with the primary goal of enhancing multilingual comprehension, an excessive number of irrelevant layers and parameters are tuned during the first stage. Given our findings that the representation learning of languages is merely conducted in lower-level layers, we propose an efficient multilingual reasoning alignment approach that precisely identifies and fine-tunes the layers responsible for handling multilingualism. Experimental results show that our method, SLAM, only tunes 6 layers’ feed-forward sub-layers including 6.5-8% of all parameters within 7B and 13B LLMs, achieving superior average performance than all strong baselines across 10 languages. Meanwhile, SLAM only involves one training stage, reducing training time by 4.1-11.9× compared to the two-stage method.
Enhancing Neural Machine Translation Through Target Language Data: A kNN-LM Approach for Domain Adaptation
Abudurexiti Reheman | Hongyu Liu | Junhao Ruan | Abudukeyumu Abudula | Yingfeng Luo | Tong Xiao | JingBo Zhu
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Abudurexiti Reheman | Hongyu Liu | Junhao Ruan | Abudukeyumu Abudula | Yingfeng Luo | Tong Xiao | JingBo Zhu
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Neural machine translation (NMT) has advanced significantly, yet challenges remain in adapting to new domains . In scenarios where bilingual data is limited, this issue is further exacerbated. To address this, we propose kNN-LM-NMT, a method that leverages semantically similar target language sentences in the kNN framework. Our approach generates a probability distribution over these sentences during decoding, and this distribution is then interpolated with the NMT model’s distribution. Additionally, we introduce an n-gram-based approach to focus on similar fragments, enabling the model to avoid the noise introduced by the non-similar parts. To enhance accuracy, we further incorporate cross-lingual retrieval similarity to refine the kNN probability distribution. Extensive experiments on multi-domain datasets demonstrate significant performance improvements in both high-resource and low-resource scenarios. Our approach effectively extracts translation knowledge from limited target domain data, and well benefits from large-scale monolingual data for robust context representation.
2024
Exploiting Target Language Data for Neural Machine Translation Beyond Back Translation
Abudurexiti Reheman | Yingfeng Luo | Junhao Ruan | Chunliang Zhang | Anxiang Ma | Tong Xiao | JingBo Zhu
Findings of the Association for Computational Linguistics: ACL 2024
Abudurexiti Reheman | Yingfeng Luo | Junhao Ruan | Chunliang Zhang | Anxiang Ma | Tong Xiao | JingBo Zhu
Findings of the Association for Computational Linguistics: ACL 2024
Neural Machine Translation (NMT) encounters challenges when translating in new domains and low-resource languages. To address these issues, researchers have proposed methods to integrate additional knowledge into NMT, such as translation memories (TMs). However, finding TMs that closely match the input sentence remains challenging, particularly in specific domains. On the other hand, monolingual data is widely accessible in most languages, and back-translation is seen as a promising approach for utilizing target language data. Nevertheless, it still necessitates additional training. In this paper, we introduce Pseudo-kNN-MT, a variant of k-nearest neighbor machine translation (kNN-MT) that utilizes target language data by constructing a pseudo datastore. Furthermore, we investigate the utility of large language models (LLMs) for the kNN component. Experimental results demonstrate that our approach exhibits strong domain adaptation capability in both high-resource and low-resource machine translation. Notably, LLMs are found to be beneficial for robust NMT systems.
Search
Fix author
Co-authors
- JingBo Zhu (朱靖波) 8
- Tong Xiao 7
- Bei Li 6
- Xinyu Liu 5
- Jingang Wang 5
- Kechen Jiao 4
- Xin Chen 3
- Jiahao Liu 3
- Chenglong Wang 3
- Abudukeyumu Abudula 2
- Yuchun Fan 2
- Yingfeng Luo 2
- Abudurexiti Reheman 2
- Qifan Wang 2
- Chunyang Xiao 2
- Yongjing Yin 2
- Runsong Zhao 2
- Xunliang Cai 1
- Shichao Dong 1
- Zhirui Fang 1
- Xiaocheng Feng (冯骁骋) 1
- Yuan Ge 1
- Lei Huang 1
- Pengcheng Huang 1
- Shujian Huang (书剑 黄) 1
- Changliang Li 1
- Xiu Li 1
- Hongyu Liu 1
- Xin Liu 1
- Anxiang Ma (马安香) 1
- Yongyu Mu 1
- Zhongjian Qiao 1
- Hongyin Tang 1
- Yilin Wang 1
- Tong Xiao 1
- Yaxin Xu 1
- Chunliang Zhang 1
- Yifan Zhu 1