Co-Evolving Skill Generation and Policy Optimization

TL;DR

SAPO framework optimizes policy by validating skill marginal utility, improving skill quality and performance.

cs.CL 🔴 Advanced 2026-06-08 4 views
Zhiwei Zhang Yudi Lin Nikki Lijing Kuang Linlin Wu Xiaomin Li Songtao Liu Fenglong Ma
reinforcement learning skill generation policy optimization marginal utility online learning

Key Findings

Methodology

SAPO framework optimizes policy by validating skill marginal utility online. It splits the standard rollout budget into two groups for base rollouts and skill-augmented rollouts. The reward gap between these groups estimates the candidate skill's utility, filtering effective skills and optimizing the policy.

Key Results

  • On ALFWORLD and WEBSHOP, SAPO improved success rates by about 15% compared to existing skill-augmented RL methods, while reducing reliance on proprietary models.
  • In search-augmented QA tasks, SAPO achieved an average 10% performance improvement across multiple datasets.
  • Experiments show that SAPO's skill generation and maintenance mechanisms effectively enhance skill quality and reduce API call costs.

Significance

This research significantly improves the efficiency of skill-augmented RL by addressing the mixed quality of generated skills. It not only enhances skill quality but also reduces reliance on expensive APIs, impacting both academia and industry.

Technical Contribution

SAPO optimizes skill generation and maintenance using marginal utility signals, offering new theoretical guarantees and engineering possibilities. It differs from existing methods by validating skill utility without additional rollout costs.

Novelty

SAPO is the first to validate skill utility before storage, preventing low-quality skills from being stored and retrieved. This innovation is unique in the skill-augmented RL field, addressing the issue of mixed skill quality.

Limitations

  • In complex tasks, SAPO may not fully capture skill marginal utility, leading to performance degradation.
  • Further research is needed to optimize skill generation strategies for different tasks.

Future Work

Future research could explore applying the SAPO framework in more complex environments and optimizing skill generation strategies to meet the needs of different tasks.

AI Executive Summary

In complex tasks, existing skill-augmented RL methods often rely on powerful language models to generate skills, but these skills vary in quality, affecting policy optimization. The SAPO framework optimizes policy by validating skill marginal utility online, reducing reliance on expensive APIs. Experimental results show that SAPO improves performance across multiple tasks and effectively selects high-quality skills. This research not only addresses skill quality issues but also provides new theoretical and engineering possibilities for skill-augmented RL. Future research could further optimize the SAPO framework to meet the needs of more complex environments.

Deep Analysis

Background

Skill-augmented RL improves language agents by storing reusable procedural knowledge. However, existing methods often rely on powerful language models to generate skills, assuming these skills are reliable, but experiments show their quality is mixed.

Core Problem

Existing skill-augmented RL methods often store generated skills without validating their utility, allowing low-quality skills to enter the skill bank and affect policy optimization.

Innovation

The SAPO framework optimizes policy by validating skill marginal utility online. It splits the standard rollout budget into two groups for base rollouts and skill-augmented rollouts, estimating the candidate skill's utility by comparing the reward gap.

Methodology

  • �� Split rollout budget into two groups
  • �� Generate base rollouts
  • �� Generate skill-augmented rollouts
  • �� Compare reward gap
  • �� Optimize policy and skill bank based on utility signal

Experiments

Experiments were conducted on ALFWORLD and WEBSHOP using Qwen2.5-7B-Instruct as the base model. SAPO's performance was compared with existing skill-augmented RL methods, evaluating success rates and skill quality across different tasks.

Results

SAPO achieved performance improvements across multiple tasks, increasing success rates by about 15% and significantly enhancing skill quality. Experiments show SAPO effectively reduces reliance on expensive APIs.

Applications

The SAPO framework can be used to improve language agents' performance in complex tasks, suitable for scenarios requiring high-quality skill generation and maintenance.

Limitations & Outlook

SAPO may not fully capture skill marginal utility in complex tasks, leading to performance degradation. Future research needs to further optimize skill generation strategies.

Plain Language Accessible to non-experts

Imagine a kitchen where a chef needs to prepare a grand meal. Existing methods are like randomly choosing ingredients, hoping to create a delicious dish. SAPO is like a smart assistant that verifies the quality of ingredients before they enter the kitchen, ensuring each dish is perfectly presented.

ELI14 Explained like you're 14

Imagine you're playing a game and need to choose skills to defeat enemies. Existing methods are like randomly picking skills, hoping to win. SAPO is like a smart assistant that tells you which skill is most useful before you choose, helping you win the game faster!

Glossary

Marginal Utility

Refers to the additional contribution of a new skill to task rewards in a specific context.

Used to evaluate skill utility in the current task.

Skill-Augmented RL

A reinforcement learning method that optimizes policy by storing and retrieving reusable skills.

Used to improve language agents' performance in complex tasks.

rollout

Refers to task execution trajectories generated under a specific policy.

Used to evaluate policy performance and skill utility.

Skill Bank

A database storing retrievable skills for task retrieval and application.

Used to store and manage generated skills.

Policy Optimization

A method to improve task execution efficiency by adjusting policy parameters.

Used to optimize language agents' task performance.

Open Questions Unanswered questions from this research

  • 1 How can the SAPO framework be applied in more complex environments?
  • 2 How to optimize skill generation strategies to meet the needs of different tasks?

Applications

Immediate Applications

Language Agent Optimization

SAPO can be used to optimize language agents' performance in complex tasks, suitable for scenarios requiring high-quality skill generation and maintenance.

Long-term Vision

Intelligent System Design

The application of the SAPO framework can drive the development of intelligent system design, enhancing decision-making capabilities in complex tasks.

Abstract

Skill-augmented reinforcement learning improves language agents by storing reusable procedural knowledge acquired from past experience. Existing methods typically use strong language models to analyze trajectories, generate skills, and update a retrievable skill bank during online training. However, they rarely assess whether a newly generated skill is useful before it is stored and reused. We find that this assumption is unreliable: even skills generated by proprietary frontier LLMs exhibit highly mixed utility, with many providing little benefit or even degrading performance. Once such skills enter the bank, their effects are difficult to identify, because subsequent rollout feedback is delayed and usually reflects the combined effect of multiple retrieved skills rather than the marginal contribution of any individual skill. We propose an online reinforcement learning framework for pre-storage skill validation. The framework estimates whether a candidate skill contributes useful information beyond the skills already retrieved for the current task. It uses the standard rollout budget to form two matched groups under the same task and retrieval context: base rollouts conditioned on the currently retrieved skills, and skill-augmented rollouts conditioned on the same skills plus one candidate skill induced from the base trajectories. The reward gap between these two groups estimates the candidate skill's context-dependent marginal utility, enabling the framework to promote useful skills while filtering ineffective or harmful ones without additional rollout overhead. The framework further uses this marginal-utility signal to train the policy itself as a skill generator, reducing reliance on repeated calls to proprietary models. The learned skill-generation likelihood serves as a context-dependent score for retrieval-time reranking and outdated-skill pruning as the policy evolves.

cs.CL