用对抗序列检测生成模型是否能正确下棋,发现所有模型都不严谨。
Verification of the Implicit World Model in a Generative Model via Adversarial Sequences
- 设计对抗序列迫使模型预测非法着法,检验其逻辑正确性
- 实验表明所有棋类生成模型均不满足合法性要求,但训练方法影响程度不同
- 适合研究生成模型缺陷、训练策略优化的开发者和研究人员
生成式序列模型通常基于自然语言或形式语言的样本序列进行训练。一个重要问题是:基于样本的训练能否捕捉到这些语言的真实结构,即所谓的“世界模型”。理论表明,我们最多只能期望模型生成合法序列(即具备完整性),而无法保证覆盖全部合法序列。因此,发展实用工具验证模型的合法性至关重要。本研究以国际象棋为对象,因其复杂度适中且规则清晰,具备明确的世界模型。提出一种对抗序列生成方法,用于验证序列模型的合法性:生成有效棋局序列,迫使模型预测出非法下一步。该方法不仅能揭示模型不合规的问题,还可用于精细分析失败模式及训练过程中的不同选择的影响。我们设计多种对抗生成方法,并在大量棋类生成模型上进行了评估,涵盖随机与高质量对局数据集,采用多种训练方式。结果表明,所有模型均不满足合法性要求,但部分训练技术和数据选择可显著提升表现。此外,我们还研究了棋盘状态探测器在训练与攻击中的应用,发现大多数模型中提取的棋盘状态对下一步预测无因果作用。
原文摘要 · Abstract (English)
Generative sequence models are typically trained on sample sequences from natural or formal languages. It is a crucial question whether -- or to what extent -- sample-based training is able to capture the true structure of these languages, often referred to as the ``world model''. Theoretical results indicate that we can hope for soundness at best, that is, generating valid sequences, but not necessarily all of them. However, it is still important to have practical tools that are able to verify whether a given sequence model is sound. In this study, we focus on chess, as it is a domain that provides enough complexity while having a simple rule-based world model. We propose adversarial sequence generation for verifying the soundness of the sequence model. Our adversaries generate valid sequences so as to force the sequence model to generate an invalid next move prediction. Apart from the falsification of soundness, this method is also suitable for a more fine-grained analysis of the failure modes and the effects of different choices during training. To demonstrate this, we propose a number of methods for adversarial sequence generation and evaluate the approach on a large set of chess models. We train models on random as well as high-quality chess games, using several training recipes. We find that none of the models are sound, but some training techniques and dataset choices are able to improve soundness remarkably. We also investigate the potential application of board state probes in both our training and attack methods. Our findings indicate that the extracted board states have no causal role in next token prediction in most of the models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。