#419 · Primary category: Computer Vision

DynamicViT

computer-vision deep-learning image-classification vision-transformers

[NeurIPS 2021] [T-PAMI] DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsification

Project last updated:07/11/23

GitHub Stars

668

Forks

80

Contributors

6

License

MIT

Why we included this project

Vision transformers burn a lot of compute, and DynamicViT tackles that directly by learning to drop redundant image tokens as they move through the network, so later layers only process the patches that matter for a given input. The authors report cutting FLOPs by over 30% and raising throughput by more than 40% while keeping accuracy loss under 0.5%, and the repo ships pretrained ImageNet models for DeiT and LVViT backbones. The same sparsification idea carries over to ConvNeXt and Swin transformers, plus object detection and semantic segmentation, in the T-PAMI version of the paper. This is research code rather than a packaged library, so expect to adapt the training scripts to your own pipeline. If you are exploring efficient vision backbones or want to build on token sparsification, the PyTorch implementation is a solid reference to start from.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category