Microsoft Vision Model ResNet-50 combines web-scale data and multi-task learning to achieve state of the art

Published

By , Software Engineer, Microsoft Bing Multimedia Group , Program Manager, Microsoft Bing Multimedia Group

Research Area

Two part animation: 

In part one, an animation graphs shows Microsoft Vision Model ResNet-50 is a state-of-the-art pretrained ResNet-50 model,   
measured above by the mean average score across seven popular computer vision benchmarks. 

In part two, three models are represented through visuals and text.
Microsoft Vision Model ResNet-50 is a state-of-the-art pretrained ResNet-50 model, measured above by the mean average score across seven popular computer vision benchmarks.

Pretrained vision models accelerate deep learning research and bring down the cost of performing computer vision tasks in production. By pretraining one large vision model to learn general visual representation of images, then transferring the learning across multiple downstream tasks, a team achieves competitive performance at a fraction of the cost when compared to collecting new data and training a new model for each task. Further fine-tuning of the pretrained model with task-specific training data often yields even higher performance than training specialized models.

Microsoft Vision Model ResNet-50 is a large pretrained vision model created by the Multimedia Group at Microsoft Bing. The model is built using the search engine’s web-scale image data in order to power its Image Search and Visual Search. We are excited to announce that we are making Microsoft Vision Model ResNet-50 publicly available today.

Microsoft
Vision Model
Google
Big Transfer
OpenAI
CLIP
PyTorch
ResNet-50
CIFAR-1092.6492.5187.8582.23
CIFAR-10076.0579.8467.0261.36
STL-1098.1098.7197.2096.32
SVHN72.6464.2264.3352.05
CUB82.2082.7568.3838.79
Flowers-10299.2899.3895.2377.62
ImageNet 73.8572.8357.0075.63
Average84.9784.3276.7269.14
Evaluation of Microsoft Vision Model ResNet-50 and comparable models on seven popular computer vision benchmarks

We evaluate Microsoft Vision Model ResNet-50 against the state-of-the-art pretrained ResNet-50 models and the baseline PyTorch implementation of ResNet-50, following the experiment setup of OpenAI CLIP. Linear probe is a standard evaluation protocol for representation learning in which a linear classifier is trained on frozen embeddings of the pretrained vision model for each benchmark.

To achieve the state-of-the-art performance in a cost-sensitive production setting, Microsoft Vision Model ResNet-50 leverages multi-task learning and optimizes separately for four datasets, including ImageNet-22k, Microsoft COCO, and two web-supervised datasets containing 40 million image-label pairs collected from image search engines.

Multi-task learning via hard parameter sharing

We chose to use multi-task learning with hard parameter sharing (see figure above). Single neural networks will optimize each classification problem at the same time. By using tasks of varied sizes—up to 40 million images with 100,000 different labels from web-supervised sources—Microsoft Vision Model ResNet-50 can achieve high robustness and good transferability to different domains.  

Spotlight: Webinar series

Microsoft research webinars

Lectures from Microsoft researchers with live Q&A and on-demand viewing.

During training, images from each dataset are sampled proportionally to the size of the datasets. This approach promotes larger datasets at first, but once optimization flattens, the optimizer needs to look for improvements to smaller datasets without downgrading the performance of the larger ones. As a result, final accuracy scores for each of the datasets are competitive with specialized models trained on each specific dataset.

Dive deeper into Microsoft Vision Model ResNet-50

You can get your hands on Microsoft Vision Model ResNet-50 by visiting https://aka.ms/microsoftvision. On this webpage, you will find a description of how to install and use the model to encode images into embedding vectors. We are also hosting a public webinar about our model on February 25 at 10 AM PT. Part of the webinar will be a demo of applying the model to example computer vision tasks, and there will be a live Q&A session at the end. You can learn more and register for the webinar at its registration page.

Acknowledgments

Microsoft Vision Model ResNet-50 is one in a family of world-class computer vision models we’ve built at Microsoft Bing Multimedia Group. We thank Mark Bolin, Ravi Yada, Kun Wu, Meenaz Merchant, Arun Sacheti, and Jordi Ribas for enabling this work. If the opportunity to build pioneering computer vision models excites you, visit our career page to learn about our openings.