Zijun Song
Author directory2026
TSRL: Trigger-based Supervised Reinforcement Learning for Continual Agent Training
Yuqian Fu | Kaiwen Jiang | Zijun Song | Jiawei Wang | Yuanheng Zhu | Dongbin Zhao
Findings of the Association for Computational Linguistics: EMNLP 2026
Yuqian Fu | Kaiwen Jiang | Zijun Song | Jiawei Wang | Yuanheng Zhu | Dongbin Zhao
Findings of the Association for Computational Linguistics: EMNLP 2026
Large language model (LLM) agents excel at complex, multi-step tasks but suffer from catastrophic forgetting during continual learning—acquiring new skills often degrades previously mastered capabilities. Existing post-training paradigms face a fundamental trade-off: Supervised Fine-Tuning (SFT) is sample-efficient but prone to severe forgetting, while Reinforcement Learning (RL) preserves prior capabilities but converges slowly without expert guidance. To bridge this gap, we introduce **T**rigger-based **S**upervised **R**einforcement **L**earning (TSRL), a framework that integrates the strengths of both paradigms through an adaptive, state-dependent expert supervision mechanism. TSRL employs RL as the optimization backbone, activating SFT selectively on "hard states" where the policy fails. Theoretically, we show that restricting supervision to failure states derives an upper bound on the cumulative parameter updates, thereby reducing forgetting compared to full SFT. Empirically, TSRL achieves consistent gains across three agent benchmarks (ALFWorld, WebShop, and Deep Search) at both 7B and 32B scales. In continual learning scenarios, our method retains up to 9.6 points more capability on prior tasks than SFT→RL baselines, while matching or exceeding target-task performance.
CriticSearch: Fine-Grained Credit Assignment for Search Agents via a Retrospective Critic
Yaocheng Zhang | Haohuan Huang | Zijun Song | Zijie Zhao | Qichao Zhang | Yuanheng Zhu | Dongbin Zhao
Findings of the Association for Computational Linguistics: ACL 2026
Yaocheng Zhang | Haohuan Huang | Zijun Song | Zijie Zhao | Qichao Zhang | Yuanheng Zhu | Dongbin Zhao
Findings of the Association for Computational Linguistics: ACL 2026
Tool-Integrated Reasoning (TIR) with search engines enables large language models to iteratively retrieve up-to-date external knowledge, enhancing adaptability and generalization in complex question-answering tasks. However, existing search agent pipelines typically depend on reinforcement learning based optimization, which often suffers from sparse outcome rewards, leading to inefficient exploration and unstable training. We introduce CriticSearch, a fine-grained credit-assignment framework that supplies dense, turn-level feedback via a retrospective critic mechanism. During training, a frozen, asymmetric critique LLM retrospectively evaluates each turn using privileged information from the full trajectory and gold answers, converting these assessments into stable, dense rewards that guide policy improvement. Experimental results across diverse multi-hop reasoning benchmarks demonstrate that CriticSearch consistently outperforms existing baselines, achieving faster convergence, improved training stability, and higher performance.
SafeSteer: Localized On-Policy Distillation for Efficient Safety Alignment
Hao Li | Jingkun An | Zijun Song | Pengyu Zhu | Rui Li | Hao Wang | Wendi Feng | Yesheng Liu | Lijun Li | Jin-Ge Yao | Lei Sha
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Hao Li | Jingkun An | Zijun Song | Pengyu Zhu | Rui Li | Hao Wang | Wendi Feng | Yesheng Liu | Lijun Li | Jin-Ge Yao | Lei Sha
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Aligning Large Language Models (LLMs) with human values often degrades their general capabilities, known as the alignment tax. Existing methods mitigate this by balancing dual objectives, which heavily rely on massive general-purpose data or auxiliary reward models.
In this paper, we argue that because safety features are inherently sparse within the output distribution, alignment requires localized modifications rather than global trade-offs. To this end, we propose SafeSteer, which performs on-policy distillation confined to safety tokens. First, we construct a safety teacher via activation steering. Based on this teacher, we develop a safety token selection algorithm. Consequently, SafeSteer restricts the reverse KL penalty to these tokens during training to preserve general capabilities.
Experimental results across diverse models show that our SafeSteer significantly outperforms existing methods on 7 safety benchmarks, with only minimal degradation on 5 general capability benchmarks. Notably, SafeSteer requires only 100 harmful samples without using any general-purpose data, less than 1% of what previous baselines used, considerably reducing alignment cost.
In this paper, we argue that because safety features are inherently sparse within the output distribution, alignment requires localized modifications rather than global trade-offs. To this end, we propose SafeSteer, which performs on-policy distillation confined to safety tokens. First, we construct a safety teacher via activation steering. Based on this teacher, we develop a safety token selection algorithm. Consequently, SafeSteer restricts the reverse KL penalty to these tokens during training to preserve general capabilities.
Experimental results across diverse models show that our SafeSteer significantly outperforms existing methods on 7 safety benchmarks, with only minimal degradation on 5 general capability benchmarks. Notably, SafeSteer requires only 100 harmful samples without using any general-purpose data, less than 1% of what previous baselines used, considerably reducing alignment cost.