SamsungLabs / ASAM
Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.
☆140Updated 3 years ago
Alternatives and similar repositories for ASAM:
Users that are interested in ASAM are comparing it to the libraries listed below
- 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
- Implementation of Visual Transformer for Small-size Datasets☆117Updated 2 years ago
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆159Updated 3 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated last year
- Recent Advances in MLP-based Models (MLP is all you need!)☆113Updated 2 years ago
- ☆57Updated last year
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆214Updated 3 years ago
- [NeurIPS 2021] Official codes for "Efficient Training of Visual Transformers with Small Datasets".☆140Updated 2 weeks ago
- ICLR 2021 i-Mix: A Domain-Agnostic Strategy for Contrastive Representation Learning☆77Updated last year
- (ICML 2022) Official PyTorch implementation of “Blurs Behave Like Ensembles: Spatial Smoothings to Improve Accuracy, Uncertainty, and Rob…☆78Updated 2 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆114Updated 3 years ago
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆25Updated 2 months ago
- Official PyTorch implementation of "Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversity" (ICLR'21 Oral)☆104Updated 3 years ago
- Official Implementation of SWAD (NeurIPS 2021)☆158Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Pytorch implementation of the paper Exploring Simple Siamese Representation Learning.☆71Updated 4 years ago
- [BMVC 2022] Official repository for "How to Train Vision Transformer on Small-scale Datasets?"☆144Updated last year
- Official Pytorch implementation of "Learning Debiased Representation via Disentangled Feature Augmentation (Neurips 2021, Oral)"☆101Updated last year
- ☆33Updated 2 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 8 months ago
- Compare neural networks by their feature similarity☆349Updated last year
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆80Updated last year
- ☆108Updated 2 years ago
- Official implementation of paper Gradient Matching for Domain Generalization☆117Updated 3 years ago
- PyTorch implementation for Vision Transformer[Dosovitskiy, A.(ICLR'21)] modified to obtain over 90% accuracy FROM SCRATCH on CIFAR-10 wit…☆176Updated 11 months ago
- Confidence-Aware Learning for Deep Neural Networks (ICML2020)☆73Updated 4 years ago
- Official PyTorch implementation of the Fishr regularization for out-of-distribution generalization☆85Updated 2 years ago
- A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization☆136Updated 3 years ago
- Official implementation for (Show, Attend and Distill: Knowledge Distillation via Attention-based Feature Matching, AAAI-2021)☆116Updated 3 years ago
- Official repository for "Orthogonal Projection Loss" (ICCV'21)☆118Updated 3 years ago