AUSO: Action-Level Unified Skill Optimization from Internalization to Utilization
2026-08-21 • Artificial Intelligence
Artificial Intelligence
AI summaryⓘ
The authors propose AUSO, a method that helps an agent learn and use skills more effectively throughout its training. Early on, AUSO guides the agent with example skills while still focusing on task results, then shifts to improving the agent's own problem-solving. Eventually, AUSO checks if each action benefits from skills and updates training to encourage helpful use of skills and discourage harmful reliance. This approach results in better performance and generalization compared to other methods that treat skills and learning separately.
skill learningreinforcement learningpolicy optimizationaction-level evaluationteacher guidancetask outcomemodel internalizationtrajectorygeneralizationagent training
Authors
Huizu Lin, Chengkai Huang, Tianqi Gao, Tao Huang, Daijiao Liu, Tongxin Li, Xiaoyan Sun, Lina Yao
Abstract
Skills play different roles as an agent's policy evolves: they should first provide learnable knowledge, then support capability formation, and finally be invoked only when they improve individual decisions. Existing methods rarely model this lifecycle. They either keep skills outside the model, fully internalize them, or select among internalization and utilization objectives through noisy task-level success rates. Such designs fragment training and assign uniform importance to actions within the same trajectory, even though skill guidance may help some decisions while distracting others. To solve these problems, we introduce AUSO (Action-level Unified Skill Optimization), which unifies skill learning and skill use through a progressive, action-aware optimization process. At the beginning of training, AUSO jointly learns from teacher guidance and environmental outcomes, enabling the policy to acquire foundational skills without losing task-oriented feedback. It subsequently emphasizes outcome-based policy optimization to consolidate autonomous problem-solving ability. As the policy matures, AUSO evaluates each sampled action under both skill-conditioned and skill-free contexts. The resulting action-level information signal is coupled with the trajectory outcome advantage, allowing beneficial skill-sensitive actions to receive stronger updates and harmful ones to be suppressed. Therefore, skills gradually transition from an external source of supervision into decision knowledge whose utilization is adapted to its action-level benefit, while reinforcement learning remains the shared backbone across all stages. Experiments on ALFWorld, WebShop, and SearchQA show that AUSO consistently improves agent performance and out-of-distribution generalization over competitive baselines.