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)
☆139Updated 2 years ago
Alternatives and similar repositories for GSAM:
Users that are interested in GSAM are comparing it to the libraries listed below
- ☆57Updated 2 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆139Updated 9 months ago
- ☆35Updated 2 years ago
- Official implementation of paper Gradient Matching for Domain Generalization☆119Updated 3 years ago
- [CVPR 2021] "The Lottery Tickets Hypothesis for Supervised and Self-supervised Pre-training in Computer Vision Models" Tianlong Chen, Jon…☆68Updated 2 years ago
- Metrics for "Beyond neural scaling laws: beating power law scaling via data pruning " (NeurIPS 2022 Outstanding Paper Award)☆55Updated last year
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- Official Pytorch's implementation of RotoGrad☆72Updated last year
- ☆104Updated last year
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 4 years ago
- Code release for REPAIR: REnormalizing Permuted Activations for Interpolation Repair☆47Updated last year
- Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch☆74Updated 2 years ago
- Official PyTorch implementation of "Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversity" (ICLR'21 Oral)☆103Updated 3 years ago
- Official PyTorch implementation of the Fishr regularization for out-of-distribution generalization☆85Updated 2 years ago
- The official PyTorch implementation - Can Neural Nets Learn the Same Model Twice? Investigating Reproducibility and Double Descent from t…☆77Updated 2 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆35Updated 2 years ago
- Weight-Averaged Sharpness-Aware Minimization (NeurIPS 2022)☆28Updated 2 years ago
- Offical Repo for Firefly Neural Architecture Descent: a General Approach for Growing Neural Networks. Accepted by Neurips 2020.☆31Updated 4 years ago
- Sharpness-Aware Minimization Leads to Low-Rank Features [NeurIPS 2023]☆27Updated last year
- Official implementation of Auxiliary Learning by Implicit Differentiation [ICLR 2021]☆84Updated 7 months ago
- ☆46Updated 2 years ago
- Transformers trained on Tiny ImageNet☆53Updated 2 years ago
- ImageNet Testbed, associated with the paper "Measuring Robustness to Natural Distribution Shifts in Image Classification."☆117Updated last year
- [TPAMI2022 & NeurIPS2020] Official implementation of Self-Adaptive Training☆129Updated 3 years ago
- ☆38Updated 3 months ago
- Visual Representation Learning Benchmark for Self-Supervised Models☆35Updated 10 months ago
- Code for "Just Train Twice: Improving Group Robustness without Training Group Information"☆70Updated 9 months ago
- PyTorch implementation of "Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning" with DDP and Apex AMP☆80Updated 4 years ago
- Code for the paper "Efficient Dataset Distillation using Random Feature Approximation"☆37Updated 2 years ago