比较扩散与自回归模型学抽象规则的能力,发现各有优劣。
Diverse capability and scaling of diffusion and auto-regressive models when learning abstract rules
- 用结构化数据集测试模型从少量样本中提炼规律
- 扩散模型生成新颖样本更优,自回归模型补全更准
- 数据量达千级时规则学习出现,但两类模型增长趋势不同
人类能从有限样本中发现规律并应用于新情境。我们研究现代生成模型是否具备类似能力,通过类瑞文渐进矩阵任务设计了GenRAVEN数据集,每个样本含三行,由40种关系规则(位置、数量或属性)统一控制。将样本编码为整数数组以聚焦规则学习,对比了扩散模型(EDM、DiT、SiT)与自回归模型(GPT2、Mamba)在无条件与条件采样下的表现。结果表明:扩散模型在无条件生成中更优,产生更多新颖且一致的样本,记忆更少;但在面板补全任务中表现较差,即使使用先进条件采样方法。自回归模型则在规则一致的面板补全上更佳,但无条件生成一致性差。两类模型均在约1000个样本/规则时出现规则学习能力。随着数据增加,扩散模型在两类任务中均提升;而自回归模型补全能力增强,但无条件生成一致性下降。研究揭示两类模型在规则学习与推理中的互补特性,提示未来需深入理解其机制与类人推理潜力。
原文摘要 · Abstract (English)
Humans excel at discovering regular structures from limited samples and applying inferred rules to novel settings. We investigate whether modern generative models can similarly learn underlying rules from finite samples and perform reasoning through conditional sampling. Inspired by Raven's Progressive Matrices task, we designed GenRAVEN dataset, where each sample consists of three rows, and one of 40 relational rules governing the object position, number, or attributes applies to all rows. We trained generative models to learn the data distribution, where samples are encoded as integer arrays to focus on rule learning. We compared two generative model families: diffusion (EDM, DiT, SiT) and autoregressive models (GPT2, Mamba). We evaluated their ability to generate structurally consistent samples and perform panel completion via unconditional and conditional sampling. We found diffusion models excel at unconditional generation, producing more novel and consistent samples from scratch and memorizing less, but performing less well in panel completion, even with advanced conditional sampling methods. Conversely, autoregressive models excel at completing missing panels in a rule-consistent manner but generate less consistent samples unconditionally. We observe diverse data scaling behaviors: for both model families, rule learning emerges at a certain dataset size - around 1000s examples per rule. With more training data, diffusion models improve both their unconditional and conditional generation capabilities. However, for autoregressive models, while panel completion improves with more training data, unconditional generation consistency declines. Our findings highlight complementary capabilities and limitations of diffusion and autoregressive models in rule learning and reasoning tasks, suggesting avenues for further research into their mechanisms and potential for human-like reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。