Skip to content

Author

Jiajun Chai

We have 9 of 33 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

Preprint Sep 2026

Tool Retrievers Are Underestimated: Annotation Expansion Reveals True Capability

The proposed ToolEX (Tool Equivalent eXpansion), a framework that automatically discovers and annotates the tool combinations functionally equivalent to the labeled ones, and confirms that the one-to-one problem extends beyond tool retrieval.

Yan-Yu Zhu, Chen-Heng Zhang, Shao-Shen Chen et al. · 0 citations
Preprint Aug 2026

Toward Plasticity-Preserving KL Regularization for Capability Retention in LLM Reinforcement Learning

Experiments in controlled multi-solution environments and continual post-training settings across multiple model scales demonstrate that CoKL achieves a more favorable balance between target-task improvement and prior-capability retention than existing regularization methods.

Li Wang, Xiao-Dong Lu, Xiao-Han Wang et al. · 0 citations
#machine learning Preprint Sep 2026

One Step, One Lead: Mitigating Higher-Order Interference in Multi-Domain Reinforcement Learning via Cross-Step Control

OSOL, which designates a focus domain at each iteration, uses the preceding checkpoint footprint to rank token-level rebound risk, and applies a drift-ranked, adaptively scaled correction within the standard GRPO update, is proposed, which suppresses the targeted cross-step output backtracking component.

Zi-Han Lin, Xiao-Han Wang, Jie Cao et al. · 1 citation
#artificial intelligence Preprint Aug 2026

ATLAS: Dual-Horizon Diagnostic Evaluation for Industrial Tool-Use Agents

This work proposes ATLAS, a dual-horizon diagnostic evaluation framework for industrial tool-use agents that instantiates LLM judge interfaces as executable signals with explicit evidence scopes and decision boundaries, and evaluates ATLAS on Meituan Xiaotuan production traffic.

Wei Chen, Pei-Lun Zhou, Zhao-Yu Hu et al. · 0 citations
Preprint Aug 2026

HiDiffTIR: Hierarchical Difficulty-Aware Policy Optimization for Multi-Turn Tool-Integrated Reasoning

HiDiffTIR is proposed, a Hierarchical Difficulty-aware policy optimization framework for multi-turn TIR that consistently improves multi-turn TIR performance and tool invocation accuracy over strong RL baselines, highlighting the necessity of difficulty-aware credit assignment for effective policy optimization in tool-...

Yu-Can Guo, Xiao-Han Wang, Miao Su et al. · 0 citations
Preprint Aug 2026

When Not to Imitate: Boundary-Aware Skill Memory for Reliable Tool-Use LLM Agents

BASM is proposed, which augments each skill with explicit boundary fields, which transforms each retrieved skill from an unconditional action template into state-conditioned guidance: the agent applies the skill when its conditions hold, suppresses inapplicable tool calls when they do not, and issues targeted repairs w...

Zi-Han Lin, Zhen-Yu Chen, Jiawen Wei et al. · 1 citation
Jul 2026

UniMem: Complementary Episodic-to-Parametric Memory for Boundary-Agnostic Task Streams

Inspired by the human brain, which balances plasticity and stability through complementary episodic storage and gradual consolidation, UniMem is proposed, a self-routing framework for autonomous memory management that consistently outperforms baselines while maintaining execution fidelity.

Si-Yu Xia, Chen-Heng Zhang, Yan-Ting Wu et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.