arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.05183cs.LGcs.AI
收藏

面向大语言模型多智能体强化学习的数据导向可解释性

Data-Centric Interpretability for LLM-based Multi-Agent Reinforcement Learning

goodstartlabs(Good Start Labs)

John Yan, Michael Yu, Yuqi Sun, Alexander Duffy, Tyler Marques, Matthew Lyle Olson

AI总结:

本文提出Meta-Autointerp方法,利用SAEs和LLM总结器分析多智能体强化学习行为,发现细粒度行为及奖励黑客,验证SAE特征的有效性并提升智能体性能。

摘要:

大型语言模型(LLMs)在复杂的强化学习多智能体环境中日益受到训练,这使得难以理解行为如何随训练变化。稀疏自编码器(SAEs)最近显示出在数据导向可解释性方面的用处。在本工作中,我们通过应用预训练的SAEs,对Full-Press Diplomacy复杂环境中的大规模强化学习训练运行进行分析,并结合LLM总结方法。我们引入了Meta-Autointerp,一种将SAE特征分组为训练动态可解释假设的方法。我们发现了细粒度行为,包括角色扮演模式、退化输出、语言切换,以及高层战略行为和环境特定的bug。通过自动化评估,我们验证了发现的90%的SAE Meta-特征是显著的,并发现了一种令人惊讶的奖励黑客行为。然而,通过两次用户研究,我们发现即使主观上有趣且看似有用的SAE特征可能对人类来说比无用或大多数LLM生成的假设更差。然而,一部分由SAE衍生的假设对下游任务具有预测性用途。我们进一步通过增强未训练智能体的系统提示,将分数提高了+14.2%。总体而言,我们展示了SAEs和LLM总结器为智能体行为提供了互补的视角,且我们的框架共同构成了未来数据导向可解释性工作的实用起点,以确保在训练过程中LLM行为的可信度。

英文摘要:

Large language models (LLMs) are increasingly trained in complex Reinforcement Learning, multi-agent environments, making it difficult to understand how behavior changes over training. Sparse Autoencoders (SAEs) have recently shown to be useful for data-centric interpretability. In this work, we analyze large-scale reinforcement learning training runs from the sophisticated environment of Full-Press Diplomacy by applying pretrained SAEs, alongside LLM-summarizer methods. We introduce Meta-Autointerp, a method for grouping SAE features into interpretable hypotheses about training dynamics. We discover fine-grained behaviors including role-playing patterns, degenerate outputs, language switching, alongside high-level strategic behaviors and environment-specific bugs. Through automated evaluation, we validate that 90% of discovered SAE Meta-Features are significant, and find a surprising reward hacking behavior. However, through two user studies, we find that even subjectively interesting and seemingly helpful SAE features may be worse than useless to humans, along with most LLM generated hypotheses. However, a subset of SAE-derived hypotheses are predictively useful for downstream tasks. We further provide validation by augmenting an untrained agent's system prompt, improving the score by +14.2%. Overall, we show that SAEs and LLM-summarizer provide complementary views into agent behavior, and together our framework forms a practical starting point for future data-centric interpretability work on ensuring trustworthy LLM behavior throughout training.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑