arXiv:2410.16024cs.AI2024-10被引 4

用大模型生成可解释决策树,少试错就能高效完成游戏任务

SMAC-R1: The Emergence of Intelligence in Decision-Making Tasks

  • 基于大模型生成决策树代码,结合奖励反馈自我优化
  • 23个原任务+10个新任务均实现高成功率,探索步数极少
  • 结果可解释且跨环境通用,适合需要透明决策的场景

星际争霸多智能体挑战(SMAC)是多智能体强化学习常用实验环境,任务为控制一定数量友方单位击败敌方部队。传统MARL算法需数百万步环境交互训练,策略难以解释且泛化能力弱。本文提出SMAC-R1,基于从DeepSeek-Coder-v2.5-236B蒸馏出的Qwen2.5-7B-Base大模型。在离线学习中通过行为克隆,利用大模型根据任务描述生成决策树代码,并通过环境奖励进行自我反思。随后,通过监督微调(SFT)和组相对策略优化(GRPO)算法,增强小模型的决策能力和脚本生成能力。在原始23个SMAC任务及10个新设计任务上验证,方法仅用极少环境探索即可生成高质量、可解释的决策树,且在同质环境中无需修改即具备强迁移能力。该方法为决策任务求解与领域特定大模型训练提供了新方向。

原文摘要 · Abstract (English)

StarCraft Multi-Agent Challenge (SMAC) has been one of the most commonly used experimental environments in multi-agent reinforcement learning (MARL), where the specific task is to control a set number of allied units to defeat enemy forces. Traditional MARL algorithms often require interacting with the environment for millions of steps to train a parametric model, of which the resulting policies are typically non-interpretable with weak transferability. In this paper, we introduce SMAC-R1 which is based on the Qwen2.5-7B-Base LLM distilled from DeepSeek-Coder-v2.5-236B. Similar to online reinforcement learning after behavior cloning in offline learning process, in our pipeline, agents leverage the DeepSeek LLM to generate decision tree code by providing task descriptions, and the agents are further self-reflected using feedback from the rewards provided by the environment. Based on that, we augment the generated scripts to fine-tune a small LLM, Qwen2.5-7B-Base, to distill the decision-making ability via Supervised Fine-Tuning (SFT) and enhance the script generation ability by the Group Relative Policy Optimization (GRPO) algorithm. We conduct experiments in the original 23 SMAC tasks and 10 newly-designed tasks to demonstrate that our method can produce high-quality, interpretable decision trees with minimal environmental exploration. Moreover, these scripts exhibit strong transferability, successfully applying to homogeneous SMAC environments without modification. We believe this approach offers a new direction for solving decision-making tasks and domain-specific LLM training pipelines in the future.

大模型决策可解释性强化学习SMAC

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。