References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Ripple Attention for Visual Perception with Sub-quadratic Complexity
(Submitted on 6 Oct 2021 (v1), last revised 15 Jun 2022 (this version, v2))
Abstract: Transformer architectures are now central to sequence modeling tasks. At its heart is the attention mechanism, which enables effective modeling of long-term dependencies in a sequence. Recently, transformers have been successfully applied in the computer vision domain, where 2D images are first segmented into patches and then treated as 1D sequences. Such linearization, however, impairs the notion of spatial locality in images, which bears important visual clues. To bridge the gap, we propose ripple attention, a sub-quadratic attention mechanism for vision transformers. Built upon the recent kernel-based efficient attention mechanisms, we design a novel dynamic programming algorithm that weights contributions of different tokens to a query with respect to their relative spatial distances in the 2D space in linear observed time. Extensive experiments and analyses demonstrate the effectiveness of ripple attention on various visual tasks.
Submission history
From: Lin Zheng [view email][v1] Wed, 6 Oct 2021 02:00:38 GMT (118kb,D)
[v2] Wed, 15 Jun 2022 13:59:31 GMT (219kb,D)
Link back to: arXiv, form interface, contact.