SamsungLabs / ASAMLinks
Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.
☆145Updated 3 years ago
Alternatives and similar repositories for ASAM
Users that are interested in ASAM are comparing it to the libraries listed below
Sorting:
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆143Updated 2 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- Implementation of Visual Transformer for Small-size Datasets☆124Updated 3 years ago
- (ICML 2022) Official PyTorch implementation of “Blurs Behave Like Ensembles: Spatial Smoothings to Improve Accuracy, Uncertainty, and Rob…☆78Updated 2 years ago
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆159Updated 3 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆217Updated 4 years ago
- ☆58Updated 2 years ago
- Official Implementation of SWAD (NeurIPS 2021)☆163Updated 2 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆116Updated 3 years ago
- Recent Advances in MLP-based Models (MLP is all you need!)☆115Updated 2 years ago
- Transformers w/o Attention, based fully on MLPs☆93Updated last year
- ICLR 2021 i-Mix: A Domain-Agnostic Strategy for Contrastive Representation Learning☆78Updated last year
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆33Updated 7 months ago
- A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization☆135Updated 4 years ago
- Confidence-Aware Learning for Deep Neural Networks (ICML2020)☆74Updated 4 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆142Updated last year
- This repo is for our paper: Normalization Techniques in Training DNNs: Methodology, Analysis and Application☆84Updated 3 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- [BMVC 2022] Official repository for "How to Train Vision Transformer on Small-scale Datasets?"☆154Updated last year
- Official Pytorch implementation of "Learning Debiased Representation via Disentangled Feature Augmentation (Neurips 2021, Oral)"☆101Updated 2 years ago
- ☆35Updated 2 years ago
- Pytorch implementation of the paper Exploring Simple Siamese Representation Learning.☆74Updated 4 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆205Updated last year
- Official code for ICML 2022: Mitigating Neural Network Overconfidence with Logit Normalization☆147Updated 2 years ago
- Code for ICML 2022 paper "Out-of-distribution Detection with Deep Nearest Neighbors"☆185Updated 10 months ago
- [ICLR 2022] "Deep AutoAugment" by Yu Zheng, Zhi Zhang, Shen Yan, Mi Zhang☆64Updated 8 months ago
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆82Updated last year
- PyTorch implementation of EMAN for self-supervised and semi-supervised learning: https://arxiv.org/abs/2101.08482☆103Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- [NeurIPS 2021] Official codes for "Efficient Training of Visual Transformers with Small Datasets".☆143Updated 5 months ago