articlePeerJ Computer ScienceJan 31, 2024GOLD OA

Lightweight transformer image feature extraction network

University of Electronic Science and Technology of China · Guizhou University · +1 more institution

PubMed
Indexed incrossrefdoajpubmed

Abstract

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…

No related works found for this paper.