juntang-zhuang / GSAM
PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)
☆138Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for GSAM
- ☆58Updated last year
- Whitening for Self-Supervised Representation Learning | Official repository☆125Updated last year
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 5 months ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- Metrics for "Beyond neural scaling laws: beating power law scaling via data pruning " (NeurIPS 2022 Outstanding Paper Award)☆53Updated last year
- Official implementation of paper Gradient Matching for Domain Generalization☆116Updated 2 years ago
- ☆32Updated last year
- [CVPR 2021] "The Lottery Tickets Hypothesis for Supervised and Self-supervised Pre-training in Computer Vision Models" Tianlong Chen, Jon…☆68Updated last year
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 3 years ago
- Official PyTorch implementation of "Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversity" (ICLR'21 Oral)☆104Updated 2 years ago
- Sharpness-Aware Minimization Leads to Low-Rank Features [NeurIPS 2023]☆25Updated last year
- Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch☆71Updated last year
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆75Updated 2 years ago
- ☆58Updated 2 years ago
- Transformers trained on Tiny ImageNet☆47Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Official PyTorch implementation of the Fishr regularization for out-of-distribution generalization☆83Updated 2 years ago
- Official Pytorch's implementation of RotoGrad☆73Updated last year
- Visual Representation Learning Benchmark for Self-Supervised Models☆35Updated 6 months ago
- ☆102Updated last year
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆172Updated last year
- PixMix: Dreamlike Pictures Comprehensively Improve Safety Measures (CVPR 2022)☆100Updated 2 years ago
- Implementation of Barlow Twins paper☆99Updated 2 years ago
- PyTorch implementation of "Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning" with DDP and Apex AMP☆81Updated 4 years ago
- Code release for REPAIR: REnormalizing Permuted Activations for Interpolation Repair☆45Updated 9 months ago
- (ICML 2022) Official PyTorch implementation of “Blurs Behave Like Ensembles: Spatial Smoothings to Improve Accuracy, Uncertainty, and Rob…☆77Updated 2 years ago
- ☆105Updated last year
- [NeurIPS 2022] Make Sharpness-Aware Minimization Stronger: A Sparsified Perturbation Approach -- Official Implementation☆42Updated last year
- ☆63Updated 2 years ago
- [NeurIPS'21] "AugMax: Adversarial Composition of Random Augmentations for Robust Training" by Haotao Wang, Chaowei Xiao, Jean Kossaifi, Z…☆124Updated 2 years ago