Lightweight Mobile Vision Transformer Image Classification

via “lightweight-image-classification-inference”

image-classification model by undefined. 1,74,99,725 downloads.

Unique: Uses inverted residual blocks with squeeze-and-excitation (SE) modules and non-linear bottleneck layers, achieving state-of-the-art accuracy-to-parameter ratio (75.7% top-1 on ImageNet with 2.5M params). Trained with LAMB optimizer on ImageNet-1k, enabling faster convergence than SGD-based alternatives. Distributed via timm's unified model registry with automatic weight downloading and format conversion (PyTorch → ONNX → TensorRT).

vs others: Outperforms EfficientNet-B0 and SqueezeNet on latency-accuracy tradeoff for mobile inference; 3-5× faster than ResNet-50 on ARM devices while maintaining competitive accuracy for general-purpose classification.