Diverse Imagenet Models Transfer Better

Niv Nayman, Avram Golbert, Asaf Noy, Lihi Zelnik-Manor; Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), 2024, pp. 1914-1925

Abstract


A commonly accepted hypothesis is that models with higher accuracy on Imagenet perform better on other downstream tasks, leading to much research dedicated to optimizing Imagenet accuracy. Recently this hypothesis has been challenged by evidence showing that self-supervised models transfer better than their supervised counterparts, despite their inferior Imagenet accuracy. This calls for identifying the additional factors, on top of Imagenet accuracy, that make models transferable. In this work we show that high diversity of the filters learnt by the model promotes transferability jointly with Imagenet accuracy. Encouraged by the recent transferability results of self-supervised models, we use a simple procedure to combine self-supervised and supervised pretraining and generate models with both high diversity and high accuracy, and as a result high transferability. We experiment with several architectures and multiple downstream tasks, including both single-label and multi-label classification.

Related Material


[pdf] [supp] [arXiv]
[bibtex]
@InProceedings{Nayman_2024_WACV, author = {Nayman, Niv and Golbert, Avram and Noy, Asaf and Zelnik-Manor, Lihi}, title = {Diverse Imagenet Models Transfer Better}, booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)}, month = {January}, year = {2024}, pages = {1914-1925} }