arXiv:2410.24096cs.LGcs.LO2024-10

用分阶段安全防护机制,让智能体在不依赖模型的前提下安全适应新任务。

Progressive Safeguards for Safe and Model-Agnostic Reinforcement Learning

  • 设计可手动配置的有限状态机作为跨任务安全守护者
  • 在3个环境测试中实现接近最小的安全违规次数
  • 适合需要高安全性的强化学习应用,如游戏控制与大模型微调

本文提出一种形式化、模型无关的元强化学习安全框架,灵感源自父母逐步引导孩子完成风险递增任务时的安全保护机制。将每项任务与一个基于安全规范的有限状态机式守护者同步,该守护者监控安全性并生成形式化奖励信号。安全规范及其对应的守护者可任意复杂且非马尔可夫,提升了训练灵活性与策略可解释性。守护者设计虽需人工参与,但为高层级且模型无关,支持端到端安全学习,适用于从像素级游戏控制到大语言模型微调等场景。基于给定安全规范集,训练模型可仅用少量样本适应新规范,得益于高效的安全偏置迁移,显著减少安全违规。在仿《我的世界》网格世界、VizDoom游戏环境及大模型微调任务中评估,所提方法实现近最小安全违规,优于基线方法。

原文摘要 · Abstract (English)

In this paper we propose a formal, model-agnostic meta-learning framework for safe reinforcement learning. Our framework is inspired by how parents safeguard their children across a progression of increasingly riskier tasks, imparting a sense of safety that is carried over from task to task. We model this as a meta-learning process where each task is synchronized with a safeguard that monitors safety and provides a reward signal to the agent. The safeguard is implemented as a finite-state machine based on a safety specification; the reward signal is formally shaped around this specification. The safety specification and its corresponding safeguard can be arbitrarily complex and non-Markovian, which adds flexibility to the training process and explainability to the learned policy. The design of the safeguard is manual but it is high-level and model-agnostic, which gives rise to an end-to-end safe learning approach with wide applicability, from pixel-level game control to language model fine-tuning. Starting from a given set of safety specifications (tasks), we train a model such that it can adapt to new specifications using only a small number of training samples. This is made possible by our method for efficiently transferring safety bias between tasks, which effectively minimizes the number of safety violations. We evaluate our framework in a Minecraft-inspired Gridworld, a VizDoom game environment, and an LLM fine-tuning application. Agents trained with our approach achieve near-minimal safety violations, while baselines are shown to underperform.

强化学习安全机制元学习模型无关

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