In plain words: This attention trick approximates the usual dot-product comparison between every pair of pixels, but computes it in a way that uses far less memory and processing. Tests on labeling each pixel in an image showed it works while staying cheap.
Abstract
In this paper, to remedy this deficiency, we propose a Linear Attention Mechanism which is approximate to dot-product attention with much less memory and computational costs. The efficient design makes the incorporation between attention mechanisms and neural networks more flexible and versatile. Experiments conducted on semantic segmentation demonstrated the effectiveness of linear attention mechanism. Code is available at https://github.com/lironui/Linear-Attention-Mechanism.
Rui Li, Jianlin Su, Chenxi Duan, Shunyi Zheng
arXiv:2007.14902 · cs.CV · submitted Jul 29, 2020 · updated Aug 20, 2020
abstract · pdf