StateBench:四级状态追踪基准揭示 GDN 与 Transformer 能力边界
A Fine-Grained Hierarchy and Benchmark for State Tracking in Language Models
这篇论文给状态追踪能力画了张精细地图,GDN 和 Transformer 各能解到哪一层都讲清楚了,还配套开源了 StateBench 可以直接测自己的模型。
论文提出按难度分四层的状态追踪问题层级:r-trivial、aperiodic、soluble 和一般有限状态追踪。理论上证明 Gated DeltaNet(GDN)在特征值为正时可表达所有 aperiodic 问题,而 Transformer 能识别的正则语言恰好对应 soluble 层。作者据此构建 StateBench 四级长程状态追踪基准,直接训练实验结果与理论预期基本一致。对开源混合架构与闭源前沿模型的评测显示,没有任何预训练模型能稳定解决 soluble 层任务。
A Fine-Grained Hierarchy and Benchmark for State Tracking in Language Models
Theoretical limitations of transformers on state tracking---the ability to track information about entities or variables over long contexts---have motivated new recurrent language modeling (LM) architectures such as Gated DeltaNet (GDN). These architectures have, in particular, been benchmarked against the $S_5$ word problem, a theoretically hard state-tracking task that captures abilities such as code evaluation under arbitrary variable assignment. It, however, remains open whether other, simpler forms of state tracking might both be more relevant in practice and already explain observed differences between these architectures. We propose evaluating state tracking using a hierarchy of word problems with four increasing levels of difficulty: r-trivial, corresponding to immutable assignment to variables in programming; aperiodic, where assignments are mutable; soluble, where assignments are self-referential; and general finite state tracking. Theoretically, we show that the class of problems transformers are thought to length-generalize on does not even contain all aperiodic state tracking. On the other hand, we prove that GDN with positive eigenvalues expresses all aperiodic problems, and the regular languages recognized by certain transformers are precisely all soluble ones, thus providing a fine-grained picture of state tracking capabilities of various architectures. Taking these ideas to practice, we construct StateBench, a 4-tier long-horizon state-tracking benchmark. Our systematic direct training experiments across architectures largely align with theoretical expectation, albeit with interesting observations at very large scale length-generalization. Additionally, we also evaluate pretrained models---both open hybrid and closed frontier ones---on StateBench, finding that none of them robustly solves even soluble state tracking, suggesting a direction for future improvement.
- marktechpost11-02 07:28原文