Jiajun Chai

According to our database, Jiajun Chai authored at least 37 papers between 2022 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

AutoSearch: Adaptive Search Depth for Efficient Agentic RAG via Reinforcement Learning.

Jingbo Sun, Wenyue Chong, Songjun Tu, Qichao Zhang, Yaocheng Zhang, Jiajun Chai, Xiaohan Wang, Wei Lin, Guojun Yin, Dongbin Zhao

CoRR, April, 2026

π-Play: Multi-Agent Self-Play via Privileged Self-Distillation without External Data.

Yaocheng Zhang, Yuanheng Zhu, Wenyue Chong, Songjun Tu, Qichao Zhang, Jiajun Chai, Xiaohan Wang, Wei Lin, Guojun Yin, Dongbin Zhao

CoRR, April, 2026

CDRRM: Contrast-Driven Rubric Generation for Reliable and Interpretable Reward Modeling.

Dengcan Liu, Fengkai Yang, Xiaohan Wang, Shurui Yan, Jiajun Chai, Jiahao Li, Yikun Ban, Zhendong Mao, Wei Lin, Guojun Yin

CoRR, March, 2026

Rethinking Personalization in Large Language Models at the Token Level.

Chenheng Zhang, Yijun Lu, Lizhe Fang, Chunyuan Zheng, Jiajun Chai, Xiaohan Wang, Guojun Yin, Wei Lin, Yisen Wang, Zhouchen Lin

CoRR, March, 2026

SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Training.

Qi Zhang, Yifei Wang, Xiaohan Wang, Jiajun Chai, Guojun Yin, Wei Lin, Yisen Wang

CoRR, March, 2026

CPIG: Leveraging Consistency Policy With Intention Guidance for Multiagent Exploration.

Yuqian Fu, Yuanheng Zhu, Haoran Li, Zijie Zhao, Jiajun Chai, Dongbin Zhao

IEEE Trans. Cogn. Dev. Syst., February, 2026

Contextual Rollout Bandits for Reinforcement Learning with Verifiable Rewards.

Xiaodong Lu, Xiaohan Wang, Jiajun Chai, Guojun Yin, Wei Lin, Zhijun Chen, Yu Luo, Fuzhen Zhuang, Yikun Ban, Deqing Wang

CoRR, February, 2026

Your Group-Relative Advantage Is Biased.

Fengkai Yang, Zherui Chen, Xiaohan Wang, Xiaodong Lu, Jiajun Chai, Guojun Yin, Wei Lin, Shuai Ma, Fuzhen Zhuang, Deqing Wang, Yaodong Yang, Jianxin Li, Yikun Ban

CoRR, January, 2026

Tacit mechanism: Bridging pre-training of individuality to multi-agent adversarial coordination.

Shiqing Yao, Jiajun Chai, Haixin Yu, Yongzhe Chang, Tian-Tian Zhang, Yuanheng Zhu, Xueqian Wang

Neural Networks, 2026

Promoting Efficient Reasoning with Verifiable Stepwise Reward.

Chuhuai Yue, Chengqi Dong, Yinan Gao, Hang He, Jiajun Chai, Wei Lin, Guojun Yin

Proceedings of the Fortieth AAAI Conference on Artificial Intelligence, 2026

2025

AWPO: Enhancing Tool-Use of Large Language Models through Explicit Integration of Reasoning Rewards.

Zihan Lin, Xiaohan Wang, Hexiong Yang, Jiajun Chai, Jie Cao, Guojun Yin, Wei Lin, Ran He

CoRR, December, 2025

ToolForge: A Data Synthesis Pipeline for Multi-Hop Search without Real-World APIs.

Hao Chen, Zhexin Hu, Jiajun Chai, Haocheng Yang, Hang He, Xiaohan Wang, Wei Lin, Luhang Wang, Guojun Yin, Zhuofeng Zhao

CoRR, December, 2025

Training Multi-Image Vision Agents via End2End Reinforcement Learning.

Chengqi Dong, Chuhuai Yue, Hang He, Rongge Mao, Fenghe Tang, S. Kevin Zhou, Zekun Xu, Xiaohan Wang, Jiajun Chai, Wei Lin, Guojun Yin

CoRR, December, 2025

LocalSearchBench: Benchmarking Agentic Search in Real-World Local Life Services.

Hang He, Chuhuai Yue, Chengqi Dong, Mingxue Tian, Zhenfeng Liu, Jiajun Chai, Xiaohan Wang, Yufei Zhang, Qun Liao, Guojun Yin, Wei Lin, Chengcheng Wan, Haiying Sun, Ting Su

CoRR, December, 2025

From Experience to Strategy: Empowering LLM Agents with Trainable Graph Memory.

Siyu Xia, Zekun Xu, Jiajun Chai, Wentian Fan, Yan Song, Xiaohan Wang, Guojun Yin, Wei Lin, Haifeng Zhang, Jun Wang

CoRR, November, 2025

MTIR-SQL: Multi-turn Tool-Integrated Reasoning Reinforcement Learning for Text-to-SQL.

Zekun Xu, Siyu Xia, Chuhuai Yue, Jiajun Chai, Mingxue Tian, Xiaohan Wang, Wei Lin, Haoxuan Li, Guojun Yin

CoRR, October, 2025

SSL4RL: Revisiting Self-supervised Learning as Intrinsic Reward for Visual-Language Reasoning.

Xiaojun Guo, Runyu Zhou, Yifei Wang, Qi Zhang, Chenheng Zhang, Stefanie Jegelka, Xiaohan Wang, Jiajun Chai, Guojun Yin, Wei Lin, Yisen Wang

CoRR, October, 2025

ResT: Reshaping Token-Level Policy Gradients for Tool-Use Large Language Models.

Zihan Lin, Xiaohan Wang, Jie Cao, Jiajun Chai, Guojun Yin, Wei Lin, Ran He

CoRR, September, 2025

RLFactory: A Plug-and-Play Reinforcement Learning Post-Training Framework for LLM Multi-Turn Tool-Use.

Jiajun Chai, Guojun Yin, Zekun Xu, Chuhuai Yue, Yi Jia, Siyu Xia, Xiaohan Wang, Jiwen Jiang, Xiaoguang Li, Chengqi Dong, Hang He, Wei Lin

CoRR, September, 2025

Meta Learning Task Representation in Multiagent Reinforcement Learning: From Global Inference to Local Inference.

Zijie Zhao, Yuqian Fu, Jiajun Chai, Yuanheng Zhu, Dongbin Zhao

IEEE Trans. Neural Networks Learn. Syst., August, 2025

SRFT: A Single-Stage Method with Supervised and Reinforcement Fine-Tuning for Reasoning.

Yuqian Fu, Tinghong Chen, Jiajun Chai, Xihuai Wang, Songjun Tu, Guojun Yin, Wei Lin, Qichao Zhang, Yuanheng Zhu, Dongbin Zhao

CoRR, June, 2025

LDR: Learning Discrete Representation to Improve Noise Robustness in Multiagent Tasks.

Yuqian Fu, Yuanheng Zhu, Jiajun Chai, Dongbin Zhao

IEEE Trans. Syst. Man Cybern. Syst., January, 2025

Learning Pre-Trained Tacit Behavior for Efficient Multi-Agent Adversarial Coordination.

Shiqing Yao, Jiajun Chai, Haixin Yu, Yongzhe Chang, Yuanheng Zhu, Xueqian Wang

Proceedings of the 24th International Conference on Autonomous Agents and Multiagent Systems, 2025

DipLLM: Fine-Tuning LLM for Strategic Decision-making in Diplomacy.

Kaixuan Xu, Jiajun Chai, Sicheng Li, Yuqian Fu, Yuanheng Zhu, Dongbin Zhao

Proceedings of the Forty-second International Conference on Machine Learning, 2025

INS: Interaction-aware Synthesis to Enhance Offline Multi-agent Reinforcement Learning.

Yuqian Fu, Yuanheng Zhu, Jian Zhao, Jiajun Chai, Dongbin Zhao

Proceedings of the Thirteenth International Conference on Learning Representations, 2025

Empowering LLM Agents with Zero-Shot Optimal Decision-Making through Q-learning.

Jiajun Chai, Sicheng Li, Yuqian Fu, Dongbin Zhao, Yuanheng Zhu

Proceedings of the Thirteenth International Conference on Learning Representations, 2025

UIOrchestra: Generating High-Fidelity Code from UI Designs with a Multi-agent System.

Chuhuai Yue, Jiajun Chai, Yufei Zhan, Zixiang Ding, Xihao Liang, Peixin Wang, Shihai Chen, Wang Yixuan, Wang Yanping, Guojun Yin, Wei Lin

Proceedings of the Findings of the Association for Computational Linguistics: EMNLP 2025, 2025

RLAE: Reinforcement Learning-Assisted Ensemble for LLMs.

Yuqian Fu, Yuanheng Zhu, Jiajun Chai, Guojun Yin, Wei Lin, Qichao Zhang, Dongbin Zhao

Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing, 2025

2024

NVIF: Neighboring Variational Information Flow for Cooperative Large-Scale Multiagent Reinforcement Learning.

Jiajun Chai, Yuanheng Zhu, Dongbin Zhao

IEEE Trans. Neural Networks Learn. Syst., December, 2024

CPEG: Leveraging Consistency Policy with Consensus Guidance for Multi-agent Exploration.

Yuqian Fu, Yuanheng Zhu, Haoran Li, Zijie Zhao, Jiajun Chai, Dongbin Zhao

CoRR, 2024

Aligning Credit for Multi-Agent Cooperation via Model-based Counterfactual Imagination.

Jiajun Chai, Yuqian Fu, Dongbin Zhao, Yuanheng Zhu

Proceedings of the 23rd International Conference on Autonomous Agents and Multiagent Systems, 2024

2023

A Hierarchical Deep Reinforcement Learning Framework for 6-DOF UCAV Air-to-Air Combat.

Jiajun Chai, Wenzhang Chen, Yuanheng Zhu, Zong-xin Yao, Dongbin Zhao

IEEE Trans. Syst. Man Cybern. Syst., September, 2023

UNMAS: Multiagent Reinforcement Learning for Unshaped Cooperative Scenarios.

Jiajun Chai, Weifan Li, Yuanheng Zhu, Dongbin Zhao, Zhe Ma, Kewu Sun, Jishiyu Ding

IEEE Trans. Neural Networks Learn. Syst., April, 2023

2022

NVIF: Neighboring Variational Information Flow for Large-Scale Cooperative Multi-Agent Scenarios.

Jiajun Chai, Yuanheng Zhu, Dongbin Zhao

CoRR, 2022

UNMAS: Multi-Agent Reinforcement Learning for Unshaped Cooperative Scenarios.

Jiajun Chai, Weifan Li, Yuanheng Zhu, Dongbin Zhao, Zhe Ma, Kewu Sun, Jishiyu Ding

CoRR, 2022

Learning Continuous 3-DoF Air-to-Air Close-in Combat Strategy using Proximal Policy Optimization.

Luntong Li, Zhiming Zhou, Jiajun Chai, Zhen Liu, Yuanheng Zhu, Jianqiang Yi

Proceedings of the IEEE Conference on Games, CoG 2022, Beijing, 2022

LILAC: Learning a Leader for Cooperative Reinforcement Learning.

Yuqian Fu, Jiajun Chai, Yuanheng Zhu, Dongbin Zhao

Proceedings of the IEEE Conference on Games, CoG 2022, Beijing, 2022