BViT: Broad Attention-Based Vision Transformer
Nannan Li2,3; Yaran Chen2,3; Weifan Li2,3; Zixiang Ding2,3; Dongbin Zhao2,3; Shuai Nie1
刊名IEEE Transactions on Neural Networks and Learning Systems
2023-05
页码1 - 12
关键词Broad attention broad connection image classification parameter-free attention vision transformer
ISSN号2162-237X
DOI10.1109/TNNLS.2023.3264730
英文摘要

Recent works have demonstrated that transformer can achieve promising performance in computer vision, by exploiting the relationship among image patches with self-attention. They only consider the attention in a single feature layer, but ignore the complementarity of attention in different layers. In this article, we propose broad attention to improve the performance by incorporating the attention relationship of different layers for vision transformer (ViT), which is called BViT. The broad attention is implemented by broad connection and parameter-free attention. Broad connection of each transformer layer promotes the transmission and integration of information for BViT. Without introducing additional trainable parameters, parameter-free attention jointly focuses on the already available attention information in different layers for extracting useful information and building their relationship. Experiments on image classification tasks demonstrate that BViT delivers superior accuracy of 75.0%/81.6% top-1 accuracy on ImageNet with 5M/22M parameters. Moreover, we transfer BViT to downstream object recognition benchmarks to achieve 98.9% and 89.9% on CIFAR10 and CIFAR100, respectively, that exceed ViT with fewer parameters. For the generalization test, the broad attention in Swin Transformer, T2T-ViT and LVT also brings an improvement of more than 1%. To sum up, broad attention is promising to promote the performance of attention-based models. Code and pretrained models are available at https://github.com/DRL/BViT.

URL标识查看原文
语种英语
内容类型期刊论文
源URL[http://ir.ia.ac.cn/handle/173211/52190]  
专题复杂系统管理与控制国家重点实验室_深度强化学习
作者单位1.The National Laboratory of Pattern Recognition, Institute of Automation, Chinese Academy of Sciences
2.School of artificial intelligence, University of Chinese Academy of Sciences
3.The State Key Laboratory of Multimodal Artificial Intelligence Systems, Institute of Automation, Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Nannan Li,Yaran Chen,Weifan Li,et al. BViT: Broad Attention-Based Vision Transformer[J]. IEEE Transactions on Neural Networks and Learning Systems,2023:1 - 12.
APA Nannan Li,Yaran Chen,Weifan Li,Zixiang Ding,Dongbin Zhao,&Shuai Nie.(2023).BViT: Broad Attention-Based Vision Transformer.IEEE Transactions on Neural Networks and Learning Systems,1 - 12.
MLA Nannan Li,et al."BViT: Broad Attention-Based Vision Transformer".IEEE Transactions on Neural Networks and Learning Systems (2023):1 - 12.
个性服务
查看访问统计
相关权益政策
暂无数据
收藏/分享
所有评论 (0)
暂无评论
 

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。


©版权所有 ©2017 CSpace - Powered by CSpace