Lightweight transformer image feature extraction network

doi:10.7717/peerj-cs.1755

Lightweight transformer image feature extraction network

2024-01-31

84

OA

AI

DOI：10.7717/peerj-cs.1755

原文链接

查看原文

摘要

En 中文

In recent years, the image feature extraction method based on Transformer has become a research hotspot. However, when using Transformer for image feature extraction, the model's complexity increases quadratically with the number of tokens entered. The quadratic complexity prevents vision transformer-based backbone networks from modelling high-resolution images and is computationally expensive. To address this issue, this study proposes two approaches to speed up Transformer models. Firstly, the self-attention mechanism's quadratic complexity is reduced to linear, enhancing the model's internal processing speed. Next, a parameter-less lightweight pruning method is introduced, which adaptively samples input images to filter out unimportant tokens, effectively reducing irrelevant input. Finally, these two methods are combined to create an efficient attention mechanism. Experimental results demonstrate that the combined methods can reduce the computation of the original Transformer model by 30%-50%, while the efficient attention mechanism achieves an impressive 60%-70% reduction in computation.

Keyword:

Transformer

Quadratic complexity

Image feature extraction

Self-attention mechanism

Pruning

Efficient attention

Lightweight transformer image feature extraction network

摘要

AI总结

期刊

机构