LeViT series

December 7, 2021 · View on GitHub


Catalogue

1. Overview

LeViT is a fast inference hybrid neural network for image classification tasks. Its design considers the performance of the network model on different hardware platforms, so it can better reflect the real scenarios of common applications. Through a large number of experiments, the author found a better way to combine the convolutional neural network and the Transformer system, and proposed an attention-based method to integrate the position information encoding in the Transformer. Paper

2. Accuracy, FLOPs and Parameters

ModelsTop1Top5Reference
top1
Reference
top5
FLOPs
(M)
Params
(M)
LeViT-128S0.75980.92690.7660.9293057.8
LeViT-1280.78100.93710.7860.9404069.2
LeViT-1920.79340.94460.8000.94765811
LeViT-2560.80850.94970.8160.954112019
LeViT-3840.81910.95510.8260.960235339

Note:The difference in accuracy from Reference is due to the difference in data preprocessing and the absence of distilled head as output.