Jiaxin Mao
Author directory2026
Entropy-Guided Token Dropout: Training Autoregressive Language Models with Limited Domain Data
Jiapeng Wang | Yiwen Hu | Yanzipeng Gao | Haoyu Wang | Shuo Wang | Hongyu Lu | Jiaxin Mao | Xin Zhao | Junyi Li | Xiao Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Jiapeng Wang | Yiwen Hu | Yanzipeng Gao | Haoyu Wang | Shuo Wang | Hongyu Lu | Jiaxin Mao | Xin Zhao | Junyi Li | Xiao Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
As access to high-quality, domain-specific data grows increasingly scarce, multi-epoch training has become a practical strategy for adapting large language models (LLMs). However, autoregressive models often suffer from performance degradation under repeated data exposure, where overfitting leads to a marked decline in model capability. Through empirical analysis, we trace this degradation to an imbalance in learning dynamics: predictable, low-entropy tokens are learned quickly and come to dominate optimization, while the model’s ability to generalize on high-entropy tokens deteriorates with continued training. To address this, we introduce EntroDrop, an entropy-guided token dropout method that functions as structured data regularization. EntroDrop selectively masks low-entropy tokens during training and employs a curriculum schedule to adjust regularization strength in alignment with training progress. Experiments across model scales from 0.6B to 8B parameters show that EntroDrop consistently outperforms standard regularization baselines and maintains robust performance throughout extended multi-epoch training. These findings underscore the importance of aligning regularization with token-level learning dynamics when training on limited data. Our approach offers a promising pathway toward more effective adaptation of LLMs in data-constrained domains.
E2Rank: Unifying Text Embedding and Listwise Reranking for Effective and Efficient Search
Qi Liu | Yanzhao Zhang | Mingxin Li | Dingkun Long | Pengjun Xie | Jiaxin Mao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Qi Liu | Yanzhao Zhang | Mingxin Li | Dingkun Long | Pengjun Xie | Jiaxin Mao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Text embedding models deliver competitive retrieval performance with high efficiency, but their ranking fidelity remains limited compared to LLM-based listwise rerankers, which capture fine-grained query-document and document-document interactions at high computational cost. We propose E2Rank (Efficient Embedding-based Ranking), a unified framework that extends a single text embedding model to perform both retrieval and listwise reranking via continued training under a listwise ranking objective. The key insight is to treat the listwise prompt—constructed from the query and its top-K candidates—as a pseudo-relevance feedback (PRF) query, enabling reranking via cosine similarity against precomputed document embeddings without autoregressive decoding. Empirically, E2Rank achieves state-of-the-art results on BEIR, competitive performance on the reasoning-intensive BRIGHT benchmark, significantly lower latency than existing LLM-based rerankers, and improved embedding performance on MTEB—all within a single model.
Proof-of-Use: Mitigating tool-call hacking in Deep Research Agents
Shengjie Ma | Chenlong Deng | Jiaxin Mao | Jiadeng Huang | Xuhui Jiang | Teng Wang | Junjie Wu | Changwang Zhang | Jun Wang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Shengjie Ma | Chenlong Deng | Jiaxin Mao | Jiadeng Huang | Xuhui Jiang | Teng Wang | Junjie Wu | Changwang Zhang | Jun Wang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Deep research agents rely on iterative tool calls to gather external evidence for multi-step reasoning. While reinforcement learning (RL) improves planning and retrieval behavior, we identify a critical failure mode: tool-call hacking, where agents exploit format- and outcome-level rewards by generating superficially valid tool-use traces without genuinely grounding their reasoning in retrieved evidence. Unlike execution-based tools (e.g., code or math), whose effects are directly observable, the weak observability of causal dependencies between retrieved evidence and reasoning under format- and outcome-level supervision enables agents to maximize surface-level reward signals without genuinely grounding their reasoning in the returned evidence. This results in tool overuse, mode collapse, and decorative or hallucinated tool calls.
We propose Proof-of-Use (PoU), an evidence-grounded RL framework that strengthens the dependency between retrieval, reasoning, and final answers. PoU requires agents to cite normalized evidence identifiers during reasoning and optimizes a multi-objective reward combining step-level citation validity, global Answer–Support Alignment, and adaptive reward mixing. Experiments show that PoU mitigates tool-call hacking, improves task performance, and induces robust tool-use behavior under domain and tool shifts, even without explicit optimization for routing adaptation.
We propose Proof-of-Use (PoU), an evidence-grounded RL framework that strengthens the dependency between retrieval, reasoning, and final answers. PoU requires agents to cite normalized evidence identifiers during reasoning and optimizes a multi-objective reward combining step-level citation validity, global Answer–Support Alignment, and adaptive reward mixing. Experiments show that PoU mitigates tool-call hacking, improves task performance, and induces robust tool-use behavior under domain and tool shifts, even without explicit optimization for routing adaptation.
Tournament-GRPO: Group-Wise Tournament Rewards for Reinforcement Learning in Open-Ended Long-Form Generation
Zixuan Yang | Yiqun Chen | Wei Yang | Erhan Zhang | Zihan Shen | Xiaochi Wei | Yan Gao | Yi Wu | Yao Hu | Jiaxin Mao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Zixuan Yang | Yiqun Chen | Wei Yang | Erhan Zhang | Zihan Shen | Xiaochi Wei | Yan Gao | Yi Wu | Yao Hu | Jiaxin Mao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Reinforcement learning in open-ended long-form generation is challenging because reliable reference answers and automatic metrics are often unavailable. Existing rubric-based methods typically rely on pointwise LLM-as-a-judge scoring, but absolute scores are difficult to calibrate across complex responses, may provide weak discrimination among same-query rollouts, and can become saturated during optimization. We propose Tournament-GRPO, a group-wise reward framework that converts rubric-guided LLM judgments into relative rewards through repeated multi-round tournaments among same-query rollouts. Tournament-GRPO compares candidates within groups, accumulates tournament outcomes, and normalizes them into group-wise rewards for GRPO training. Experiments on Deep Research Bench show that Tournament-GRPO consistently outperforms existing reward-design baselines, achieving a 4.52-point overall-score improvement over the strongest baseline. Further analyses show that tournament rewards provide a favorable effectiveness–efficiency trade-off and that tournament design affects training dynamics. These results suggest that rubric-guided tournament comparison provides an effective reward signal for reinforcement learning in open-ended long-form generation.
2025
Towards Effective and Efficient Continual Pre-training of Large Language Models
Jie Chen | Zhipeng Chen | Jiapeng Wang | Kun Zhou | Yutao Zhu | Jinhao Jiang | Yingqian Min | Wayne Xin Zhao | Zhicheng Dou | Jiaxin Mao | Yankai Lin | Ruihua Song | Jun Xu | Xu Chen | Rui Yan | Zhewei Wei | Di Hu | Wenbing Huang | Ji-Rong Wen
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Jie Chen | Zhipeng Chen | Jiapeng Wang | Kun Zhou | Yutao Zhu | Jinhao Jiang | Yingqian Min | Wayne Xin Zhao | Zhicheng Dou | Jiaxin Mao | Yankai Lin | Ruihua Song | Jun Xu | Xu Chen | Rui Yan | Zhewei Wei | Di Hu | Wenbing Huang | Ji-Rong Wen
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Continual pre-training (CPT) has been an important approach for adapting language models to specific domains or tasks. In this paper, we comprehensively study its key designs to balance the new abilities while retaining the original abilities, and present an effective CPT method that can greatly improve the Chinese language ability and scientific reasoning ability of LLMs. To achieve it, we design specific data mixture and curriculum strategies based on existing datasets and synthetic high-quality data. Concretely, we synthesize multidisciplinary scientific QA pairs based on related web pages to guarantee the data quality, and also devise the performance tracking and data mixture adjustment strategy to ensure the training stability. For the detailed designs, we conduct preliminary studies on a relatively small model, and summarize the findings to help optimize our CPT method. Extensive experiments on a number of evaluation benchmarks show that our approach can largely improve the performance of Llama-3 (8B), including both the general abilities (+8.81 on C-Eval and +6.31 on CMMLU) and the scientific reasoning abilities (+12.00 on MATH and +4.13 on SciEval). Our model, data, and codes are available at https://github.com/RUC-GSAI/Llama-3-SynE.
2024
Prompt Refinement with Image Pivot for Text-to-Image Generation
Jingtao Zhan | Qingyao Ai | Yiqun Liu | Yingwei Pan | Ting Yao | Jiaxin Mao | Shaoping Ma | Tao Mei
Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Jingtao Zhan | Qingyao Ai | Yiqun Liu | Yingwei Pan | Ting Yao | Jiaxin Mao | Shaoping Ma | Tao Mei
Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
For text-to-image generation, automatically refining user-provided natural language prompts into the keyword-enriched prompts favored by systems is essential for the user experience. Such a prompt refinement process is analogous to translating the prompt from “user languages” into “system languages”. However, the scarcity of such parallel corpora makes it difficult to train a prompt refinement model. Inspired by zero-shot machine translation techniques, we introduce Prompt Refinement with Image Pivot (PRIP). PRIP innovatively uses the latent representation of a user-preferred image as an intermediary “pivot” between the user and system languages. It decomposes the refinement process into two data-rich tasks: inferring representations of user-preferred images from user languages and subsequently translating image representations into system languages. Thus, it can leverage abundant data for training. Extensive experiments show that PRIP substantially outperforms a wide range of baselines and effectively transfers to unseen systems in a zero-shot manner.
Search
Fix author
Co-authors
- Jiapeng Wang 2
- Wayne Xin Zhao 2
- Qingyao Ai 1
- Jie Chen 1
- Xu Chen 1
- Yiqun Chen 1
- Zhipeng Chen 1
- Chenlong Deng 1
- Zhicheng Dou (窦志成) 1
- Yan Gao 1
- Yanzipeng Gao 1
- Di Hu 1
- Yao Hu 1
- Yiwen Hu 1
- Jiadeng Huang 1
- Wenbing Huang 1
- Jinhao Jiang 1
- Xuhui Jiang 1
- Junyi Li 1
- Mingxin Li 1
- Yankai Lin (林衍凯) 1
- Qi Liu 1
- Yiqun Liu 1
- Dingkun Long 1
- Hongyu Lu 1
- Shaoping Ma 1
- Shengjie Ma 1
- Tao Mei 1
- Yingqian Min 1
- Yingwei Pan 1
- Zihan Shen 1
- Ruihua Song 1
- Haoyu Wang 1
- Jun Wang 1
- Shuo Wang 1
- Teng Wang 1
- Xiaochi Wei 1
- Zhewei Wei 1
- Ji-Rong Wen 1
- Junjie Wu 1
- Yi Wu 1
- Pengjun Xie 1
- Jun Xu 1
- Rui Yan 1
- Wei Yang 1
- Zixuan Yang 1
- Ting Yao 1
- Jingtao Zhan 1
- Changwang Zhang 1
- Erhan Zhang 1
- Xiao Zhang 1
- Yanzhao Zhang 1
- Kun Zhou 1
- Yutao Zhu (朱余韬) 1