adrianjav / rotograd
Official Pytorch's implementation of RotoGrad
☆72Updated last year
Alternatives and similar repositories for rotograd:
Users that are interested in rotograd are comparing it to the libraries listed below
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆114Updated last year
- 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
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- Official implementation of paper Gradient Matching for Domain Generalization☆117Updated 3 years ago
- Gradient Starvation: A Learning Proclivity in Neural Networks☆61Updated 4 years ago
- Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Training and Effective Adaptation", Haoxi…☆67Updated 3 years ago
- Winning Solution of the NeurIPS 2020 Competition on Predicting Generalization in Deep Learning☆38Updated 3 years ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- Differentiable Top-k Classification Learning☆79Updated 2 years ago
- implements optimal transport algorithms in pytorch☆92Updated 2 years ago
- Official PyTorch implementation of the Fishr regularization for out-of-distribution generalization☆85Updated 2 years ago
- Code for Which Tasks Should Be Learned Together in Multi-task Learning?☆94Updated last year
- [NeurIPS 2021] A Geometric Analysis of Neural Collapse with Unconstrained Features☆54Updated 2 years ago
- ☆57Updated last year
- Code for "Just Train Twice: Improving Group Robustness without Training Group Information"☆70Updated 8 months ago
- Official implementation of Auxiliary Learning by Implicit Differentiation [ICLR 2021]☆84Updated 6 months ago
- Official code for ICML 2022: Mitigating Neural Network Overconfidence with Logit Normalization☆146Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 3 years ago
- In Defense of the Unitary Scalarization for Deep Multi-Task Learning☆21Updated last year
- ☆107Updated last year
- AdaShare: Learning What To Share For Efficient Deep Multi-Task Learning☆112Updated 4 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆76Updated 2 years ago
- Code to reproduce experiments from 'Does Knowledge Distillation Really Work' a paper which appeared in the 2021 NeurIPS proceedings.☆33Updated last year
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆35Updated 2 years ago
- Simple data balancing baselines for worst-group-accuracy benchmarks.☆41Updated last year
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆54Updated last year
- ☆34Updated 3 years ago
- MoCo with Alignment and Uniformity Loss.☆61Updated 3 years ago
- Code for "On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertainty".☆110Updated 2 years ago
- ☆58Updated 3 years ago