Search papers, labs, and topics across Lattice.
This paper introduces the Model-Aware Skill Alignment (MASA) framework, which customizes procedural skills for large language model (LLM) agents based on their specific capabilities, rather than applying a one-size-fits-all approach. Through controlled experiments, it was demonstrated that skill effectiveness varies significantly across different model architectures, with MASA achieving performance improvements of up to 25.8 points over existing baselines. The framework employs a two-stage process that includes a hierarchical skill evolution pipeline and a model-conditioned skill rewriter, enabling adaptation without altering the agent's weights and ensuring generalization to new tasks and environments.
Skills tailored to specific LLM architectures can lead to a performance boost of nearly 26 points, challenging the notion of model-agnostic skill libraries.
LLM agents increasingly retrieve externally curated skills-procedural instructions retrieved at decision time-to improve performance on long-horizon interactive tasks. Existing skill libraries are typically treated as model-agnostic, reusing the same skill formulations across backbones with substantially different capacities and behaviors. However, our controlled experiments across multiple model scales show that skill effectiveness is strongly model-dependent: a skill that benefits one backbone can harm another. Motivated by this observation, we propose MASA Model-Aware Skill Alignment, a framework that adapts skills to each target backbone without modifying agent weights. MASA operates in two stages: (1) a hierarchical skill evolution pipeline that iteratively rewrites general and task-specific skills using hill climbing and UCB-driven tree search, guided by environment feedback and model capability profiles; and (2) a lightweight model-conditioned skill rewriter trained on evolution trajectories to reproduce the adaptation in a single forward pass. Experiments across three interactive environments and four backbones show that MASA consistently achieves the best overall performance, with gains of up to 25.8 points over the strongest baseline. The learned rewriter further generalizes to unseen tasks and environments without additional search, consistently outperforming a much larger teacher LLM at a fraction of the inference cost.
