MNASNet

Last updated on Feb 14, 2021

mnasnet_100

Parameters 4 Million
FLOPs 416 Million
File Size 16.91 MB
Training Data ImageNet
Training Resources
Training Time

Training Techniques RMSProp, Weight Decay
Architecture 1x1 Convolution, Batch Normalization, Convolution, Depthwise Separable Convolution, Dropout, Global Average Pooling, Inverted Residual Block, Residual Connection, ReLU, Max Pooling, Softmax
ID mnasnet_100
Layers 100
Dropout 0.2
Crop Pct 0.875
Momentum 0.9
Batch Size 4000
Image Size 224
Interpolation bicubic
RMSProp Decay 0.9
SHOW MORE
SHOW LESS
semnasnet_100

Parameters 4 Million
FLOPs 415 Million
File Size 15.00 MB
Training Data ImageNet
Training Resources
Training Time

Architecture 1x1 Convolution, Batch Normalization, Convolution, Depthwise Separable Convolution, Dropout, Global Average Pooling, Inverted Residual Block, Residual Connection, ReLU, Max Pooling, Softmax, Squeeze-and-Excitation Block
ID semnasnet_100
Crop Pct 0.875
Image Size 224
Interpolation bicubic
SHOW MORE
SHOW LESS
README.md

Summary

MnasNet is a type of convolutional neural network optimized for mobile devices that is discovered through mobile neural architecture search, which explicitly incorporates model latency into the main objective so that the search can identify a model that achieves a good trade-off between accuracy and latency. The main building block is an inverted residual block (from MobileNetV2).

How do I load this model?

To load a pretrained model:

import timm
m = timm.create_model('mnasnet_100', pretrained=True)
m.eval()

Replace the model name with the variant you want to use, e.g. mnasnet_100. You can find the IDs in the model summaries at the top of this page.

How do I train this model?

You can follow the timm recipe scripts for training a new model afresh.

Citation

@misc{tan2019mnasnet,
      title={MnasNet: Platform-Aware Neural Architecture Search for Mobile}, 
      author={Mingxing Tan and Bo Chen and Ruoming Pang and Vijay Vasudevan and Mark Sandler and Andrew Howard and Quoc V. Le},
      year={2019},
      eprint={1807.11626},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Results

Image Classification on ImageNet

Image Classification
BENCHMARK MODEL METRIC NAME METRIC VALUE GLOBAL RANK
ImageNet semnasnet_100 Top 1 Accuracy 75.45% # 223
Top 5 Accuracy 92.61% # 223
ImageNet mnasnet_100 Top 1 Accuracy 74.67% # 235
Top 5 Accuracy 92.1% # 235