arrow
返回

EATFormer: Improving Vision Transformer Inspired by Evolutionary Algorithm

delete2024-04-02
delete3
PRE
AI
J
Jiangning Zhang
X
Xiangtai Li
Y
Yabiao Wang
C
Chengjie Wang
Y
Yang, Yibo
刘
刘勇 (Yong Liu) *
D
Dacheng Tao
DOI:10.1007/s11263-024-02034-6delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Motivated by biological evolution, this paper explains the rationality of Vision Transformer by analogy with the proven practical evolutionary algorithm (EA) and derives that both have consistent mathematical formulation. Then inspired by effective EA variants, we propose a novel pyramid EATFormer backbone that only contains the proposed EA-based transformer (EAT) block, which consists of three residual parts, i.e., Multi-scale region aggregation, global and local interaction, and feed-forward network modules, to model multi-scale, interactive, and individual information separately. Moreover, we design a task-related head docked with transformer backbone to complete final information fusion more flexibly and improve a modulated deformable MSA to dynamically model irregular locations. Massive quantitative and quantitative experiments on image classification, downstream tasks, and explanatory experiments demonstrate the effectiveness and superiority of our approach over state-of-the-art methods. E.g., our Mobile (1.8 M), Tiny (6.1 M), Small (24.3 M), and Base (49.0 M) models achieve 69.4, 78.4, 83.1, and 83.9 Top-1 only trained on ImageNet-1K with naive training recipe; EATFormer-Tiny/Small/Base armed Mask-R-CNN obtain 45.4/47.4/49.0 box AP and 41.4/42.9/44.2 mask AP on COCO detection, surpassing contemporary MPViT-T, Swin-T, and Swin-S by 0.6/1.4/0.5 box AP and 0.4/1.3/0.9 mask AP separately with less FLOPs; Our EATFormer-Small/Base achieve 47.3/49.3 mIoU on ADE20K by Upernet that exceeds Swin-T/S by 2.8/1.7. Code is available at https://github.com/zhangzjn/EATFormer.
Keyword:
Computer vision
Vision transformer
Evolutionary algorithm
Image classification
Object detection
Image segmentation

期刊

International Journal of Computer Vision 封面图
International Journal of Computer Vision
IF:
9.3
论文数:
3.9K
被引数:
2.8W

机构

U
University of Sydney
学者数:
6.5W
论文数: 6.2W
被引数: 90
T
Tencent
学者数:
1.1K
论文数: 898
被引数: 5
P
peking university
学者数:
11.9W
论文数: 8.7W
被引数: 146
Z
zhejiang university
学者数:
17.7W
论文数: 12.1W
被引数: 152
学者 查看更多机构
引用论文

引用论文

NFV-Enabled Multicasting in Mobile Edge Clouds with Resource Sharing
err2019-08-05
err0
errOAAI
errZichuan Xu; Yutong Zhang; Weifa Liang; Qiufen Xia; Omer Rana; Alex Galis; Guowei Wu; Pan Zhou
err分享
err收藏
Ameliorative Effects of Resveratrol on Oxidative Stress Biomarkers in Horses
err2015-06-01
err0
PREAI
errMary Ucheagha Ememe; Wayuta Philip Mshelia; Joseph Olusegun Ayo
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收藏
学者 查看更多内容