arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2209.13620cs.CVcs.AIcs.LGq-bio.NC
收藏

重建引导注意力提高了神经网络的鲁棒性和形状处理能力

Reconstruction-guided attention improves the robustness and shape processing of neural networks

Stony Brook University(石溪大学)

Seoyoung Ahn, Hossein Adeli, Gregory J. Zelinsky

摘要:

许多视觉现象表明,人类使用自上而下的生成或重建过程来创建视觉感知(例如,图像,对象完成,pareidolia),但很少有人知道的作用重建中发挥强大的对象识别。我们构建了一个迭代的编码器-解码器网络,它生成对象重建,并将其用作自上而下的注意力反馈,将最相关的空间和特征信息路由到前馈对象识别过程。我们使用具有挑战性的分布外数字识别数据集MNIST-C测试了该模型,其中将15种不同类型的转换和损坏应用于手写数字图像。我们的模型对各种图像扰动表现出很强的泛化性能,平均表现优于所有其他模型,包括前馈CNN和逆向训练的网络。我们的模型是特别强大的模糊,噪声和遮挡腐败,其中形状感知起着重要的作用。消融研究进一步揭示了空间注意力和基于特征的注意力在鲁棒物体识别中的两个互补作用,前者在很大程度上与注意力文献中的空间掩蔽益处一致(重建充当掩模),后者主要有助于模型的推理速度(即,达到某个置信度阈值的时间步数)。我们还观察到,该模型有时会从噪声中产生不存在的模式,导致高度可解释的类人错误。我们的研究表明,基于建模重构的反馈赋予AI系统强大的注意力机制,这可以帮助我们理解感知生成在人类视觉处理中的作用。

英文摘要:

Many visual phenomena suggest that humans use top-down generative or reconstructive processes to create visual percepts (e.g., imagery, object completion, pareidolia), but little is known about the role reconstruction plays in robust object recognition. We built an iterative encoder-decoder network that generates an object reconstruction and used it as top-down attentional feedback to route the most relevant spatial and feature information to feed-forward object recognition processes. We tested this model using the challenging out-of-distribution digit recognition dataset, MNIST-C, where 15 different types of transformation and corruption are applied to handwritten digit images. Our model showed strong generalization performance against various image perturbations, on average outperforming all other models including feedforward CNNs and adversarially trained networks. Our model is particularly robust to blur, noise, and occlusion corruptions, where shape perception plays an important role. Ablation studies further reveal two complementary roles of spatial and feature-based attention in robust object recognition, with the former largely consistent with spatial masking benefits in the attention literature (the reconstruction serves as a mask) and the latter mainly contributing to the model's inference speed (i.e., number of time steps to reach a certain confidence threshold) by reducing the space of possible object hypotheses. We also observed that the model sometimes hallucinates a non-existing pattern out of noise, leading to highly interpretable human-like errors. Our study shows that modeling reconstruction-based feedback endows AI systems with a powerful attention mechanism, which can help us understand the role of generating perception in human visual processing.

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

↑