ViT在抽象视觉推理任务中表现不佳,新模型通过位置与物体感知显著提升性能。
Tackling the Abstraction and Reasoning Corpus with Vision Transformers: the Importance of 2D Representation, Positions, and Objects
- 采用像素级输入与物体感知的位置编码,增强对2D结构的理解
- 在400个任务中,超半数实现接近100%的解题率
- 适合研究视觉推理与Transformer架构改进的学者
抽象与推理语料库(ARC)是评估人工智能系统视觉推理能力的流行基准。原始设定下,一个ARC任务需基于少量输入-输出图像对解决程序合成问题。本文采用数据驱动方法,探究视觉变换器(ViT)是否能学习输入图像到输出图像的隐含映射。结果显示,即使每个任务训练一百万样本,现有最先进的ViT模型在多数任务上仍表现惨淡,暴露出其在捕捉简单结构化映射方面的表征缺陷。基于此,我们提出ViTARC——一种类ViT架构,引入像素级输入表示、空间感知分块机制及基于自动分割的对象位置编码等改进。该任务特定模型在400个公开ARC任务中,超过一半实现接近100%的测试解题率,仅通过监督学习即可完成。这凸显了为强大(视觉)变换器注入恰当归纳偏置的重要性,尤其是在数据充足且映射无噪声时。因此,ViTARC为未来基于变换器的视觉推理研究提供了坚实基础。
原文摘要 · Abstract (English)
The Abstraction and Reasoning Corpus (ARC) is a popular benchmark focused on visual reasoning in the evaluation of Artificial Intelligence systems. In its original framing, an ARC task requires solving a program synthesis problem over small 2D images using a few input-output training pairs. In this work, we adopt the recently popular data-driven approach to the ARC and ask whether a Vision Transformer (ViT) can learn the implicit mapping, from input image to output image, that underlies the task. We show that a ViT -- otherwise a state-of-the-art model for images -- fails dramatically on most ARC tasks even when trained on one million examples per task. This points to an inherent representational deficiency of the ViT architecture that makes it incapable of uncovering the simple structured mappings underlying the ARC tasks. Building on these insights, we propose ViTARC, a ViT-style architecture that unlocks some of the visual reasoning capabilities required by the ARC. Specifically, we use a pixel-level input representation, design a spatially-aware tokenization scheme, and introduce a novel object-based positional encoding that leverages automatic segmentation, among other enhancements. Our task-specific ViTARC models achieve a test solve rate close to 100% on more than half of the 400 public ARC tasks strictly through supervised learning from input-output grids. This calls attention to the importance of imbuing the powerful (Vision) Transformer with the correct inductive biases for abstract visual reasoning that are critical even when the training data is plentiful and the mapping is noise-free. Hence, ViTARC provides a strong foundation for future research in visual reasoning using transformer-based architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。