UCSC-VLAA/RobustCNN

This is stars and forks stats for /UCSC-VLAA/RobustCNN repository. As of 04 May, 2024 this repository has 136 stars and 11 forks.

Can CNNs Be More Robust Than Transformers? This is a PyTorch/PyTorch-XLA implementation of the ICLR'23 paper Can CNNs Be More Robust Than Transformers?. It can run on both GPU or TPU devices (see TRAIN/TEST instructions). Our implementation is heavily based on the bits_and_tpu branch of TIMM. Introduction The recent success of Vision Transformers is shaking the long dominance of Convolutional Neural Networks (CNNs) in image recognition for a decade. Specifically, in terms of robustness on out-of-distribution...