SkillGen: Learning Domain Skills for In-Context Sequential Decision Making
DOI:
https://doi.org/10.1609/aaai.v40i36.40305Abstract
Large language models (LLMs) are increasingly applied to sequential decision-making through in-context learning (ICL), yet their effectiveness is highly sensitive to prompt quality. Effective prompts should meet three principles: focus on decision-critical information, provide step-level granularity, and minimize reliance on expert annotations through label efficiency. However, existing ICL methods often fail to satisfy all three criteria simultaneously. Motivated by these challenges, we introduce SkillGen, a skill-based ICL framework for structured sequential reasoning. It constructs an action-centric, domain-level graph from sampled trajectories, identifies high-utility actions via temporal-difference credit assignment, and retrieves step-wise skills to generate fine-grained, context-aware prompts. We further present a theoretical analysis showing that focusing on high-utility segments supports task identifiability and informs more effective ICL prompt design. Experiments on ALFWorld, BabyAI, and ScienceWorld, using both open-source and proprietary LLMs, show that SkillGen achieves consistent gains, improving progress rate by 5.9%–16.5% on average across models.Downloads
Published
2026-03-14
How to Cite
Ding, R., Cheng, W., Shao, M., & Zhao, C. (2026). SkillGen: Learning Domain Skills for In-Context Sequential Decision Making. Proceedings of the AAAI Conference on Artificial Intelligence, 40(36), 30512-30520. https://doi.org/10.1609/aaai.v40i36.40305
Issue
Section
AAAI Technical Track on Natural Language Processing I