Skip to content

Westlake-AI/Backbone-vs-Optimizer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

The past decade has witnessed rapid progress in vision backbones and an evolution of deep optimizers from SGD to Adam variants. This paper, for the first time, delves into the relationship between vision network design and optimizer selection. We conduct comprehensive benchmarking studies on mainstream vision backbones and widely-used optimizers, revealing an intriguing phenomenon termed backbone-optimizer coupling bias (BOCB). Notably, classical ConvNets, such as VGG and ResNet, exhibit a marked co-dependency with SGD, while modern architectures, including ViTs and ConvNeXt, demonstrate a strong coupling with optimizers with adaptive learning rates like AdamW. More importantly, we uncover the adverse impacts of BOCB on popular backbones in real-world practice, such as additional tuning time and resource overhead, which indicates the remaining challenges and even potential risks. Through in-depth analysis and apples-to-apples comparisons, however, we surprisingly observe that specific types of network architecture can significantly mitigate BOCB, which might serve as promising guidelines for future backbone design. We hope this work as a kick-start can inspire the community to further question the long-held assumptions on vision backbones and optimizers, consider BOCB in future studies, and thus contribute to more robust, efficient, and effective vision systems. It is time to go beyond those usual choices and confront the elephant in the room. The source code and models are publicly available.

Catalog

This repo is mainly based on OpenMixup to implement image classification tasks while using MMDetection for transfer learning tasks. The manuscript and repo are updating, please watch us for the latest release!

  • Image Classification on CIFAR-100 and ImageNet-1K in OpenMixup. [configs]
  • Object Detection and Segmentation on COCO. [code]
  • 2D Pose Estimation on COCO. [code]
  • Trained Models on CIFAR-100, ImageNet-1K, and COCO.

Installation

Please check INSTALL.md for installation instructions.

Experimental Results

(back to top)

License

This project is released under the Apache 2.0 license.

Acknowledgement

Our implementation is mainly based on the following codebases. We gratefully thank the authors for their wonderful works.

  • OpenMixup: Open-source toolbox for visual representation learning.
  • MMDetection: OpenMMLab Detection Toolbox and Benchmark.

Citation

If you find this repository helpful, please consider citing:

@article{li2024battle,
  title={A Decade’s Battle on Bias of Visual Backbone and Optimizer},
  author={Siyuan Li and Juanxi Tian and Zedong Wang and Luyuan Zhang and Zicheng Liu and Cheng Tan and Weiyang Jin and Lei Xin and Yang Liu and Baigui Sun and Stan Z. Li},
  year={2024},
}

(back to top)