← Search

Xiao Sheng

3 accepted papers

2024

Enhancing Vision Transformer: Amplifying Non-Linearity in Feedforward Network Module

ICML 2024poster

Transformer models have been gaining substantial interest in the field of computer vision tasks nowadays. Although a vision transformer contains two important components which are self-attention module and feedforward network (FFN) module, the majority of research tends to concentrate on modifying t…

Cited by 3SourcePDFScholar
2024

QT-ViT: Improving Linear Attention in ViT with Quadratic Taylor Expansion

NeurIPS 2024poster

Vision transformer model (ViT) is widely used and performs well in vision tasks due to its ability to capture long-range dependencies. However, the time complexity and memory consumption increase quadratically with the number of input patches which limits the usage of ViT in real-world applications.…

Cited by 0SourcePDFScholar
2023

FDViT: Improve the Hierarchical Architecture of Vision Transformer

ICCV 2023poster

Despite the fact that transformer-based models have yielded great success in computer vision tasks, they suffer from the challenge of high computational costs that limits their use on resource-constrained devices. One major reason is that vision transformers have redundant calculations since the sel…

Cited by 18PDFScholar