arrow
返回

Factorization Vision Transformer: Modeling Long-Range Dependency With Local Window Cost

delete2025-02-01
delete1
delete
OA
AI
H
Haolin Qin
D
Daquan Zhou
T
Tingfa Xu
Z
Ziyang Bian
李
李佳男 (Jianan Li) *
DOI:10.1109/TNNLS.2023.3342172delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Transformers have astounding representational power but typically consume considerable computation which is quadratic with image resolution. The prevailing Swin transformer reduces computational costs through a local window strategy. However, this strategy inevitably causes two drawbacks: 1) the local window-based self-attention (WSA) hinders global dependency modeling capability and 2) recent studies point out that local windows impair robustness. To overcome these challenges, we pursue a preferable trade-off between computational cost and performance. Accordingly, we propose a novel factorization self-attention (FaSA) mechanism that enjoys both the advantages of local window cost and long-range dependency modeling capability. By factorizing the conventional attention matrix into sparse subattention matrices, FaSA captures long-range dependencies, while aggregating mixed-grained information at a computational cost equivalent to the local WSA. Leveraging FaSA, we present the factorization vision transformer (FaViT) with a hierarchical structure. FaViT achieves high performance and robustness, with linear computational complexity concerning input image spatial resolution. Extensive experiments have shown FaViT's advanced performance in classification and downstream tasks. Furthermore, it also exhibits strong model robustness to corrupted and biased data and hence demonstrates benefits in favor of practical applications. In comparison to the baseline model Swin-T, our FaViT-B2 significantly improves classification accuracy by 1% and robustness by 7%, while reducing model parameters by 14%. Our code will soon be publicly available: at https://github.com/q2479036243/FaViT.
Keyword:
Transformers
Robustness
Computational modeling
Costs
Computational efficiency
Windows
Sparse matrices
Factorization
local window
long-range dependency
model robustness
transformer

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

B
beijing institute of technology
学者数:
5.5W
论文数: 4.0W
被引数: 63
引用论文

引用论文

Stereoselective Pharmacokinetics of Methadone in Chronic Pain Patients
err1996-06-01
err0
PREAI
errKim Kristensen; Tine Blemmer; Helle R. Angelo; Lona L. Christrup; Niels E. Drenck; Søren N. Rasmussen; Per Sjøgren
err分享
err收藏
Semantic Understanding of Scenes Through the ADE20K Dataset通过ADE20K数据集对场景进行语义理解
err2018-12-07
err934
PREAI
errZhou, Bolei; Zhao, Hang; Puig, Xavier; Xiao, Tete; Fidler, Sanja; Barriuso, Adela; Torralba, Antonio
err分享
err收藏
err分享
err收藏
学者 查看更多内容