ssi-research / BayesAgg_MTL
Code that accompanies the paper Bayesian Uncertainty for Gradient Aggregation in Multi-Task Learning - Accepted to ICML2024
☆12Updated 3 months ago
Alternatives and similar repositories for BayesAgg_MTL:
Users that are interested in BayesAgg_MTL are comparing it to the libraries listed below
- Code for paper "Orthogonal Convolutional Neural Networks".☆116Updated 3 years ago
- Recent Advances in MLP-based Models (MLP is all you need!)☆113Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- Official repository for "Orthogonal Projection Loss" (ICCV'21)☆119Updated 3 years ago
- PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)☆83Updated 3 years ago
- Code for "Unsupervised Hyperbolic Metric Learning" in CVPR 2021.☆19Updated 2 years ago
- Code for ICML2020 paper - CLUB: A Contrastive Log-ratio Upper Bound of Mutual Information☆323Updated 9 months ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆79Updated last year
- Reproduce CKA: Similarity of Neural Network Representations Revisited☆297Updated 4 years ago
- 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
- ICLR 2021 i-Mix: A Domain-Agnostic Strategy for Contrastive Representation Learning☆77Updated last year
- Code for our ICLR'2022 paper "Generalizing Few-Shot NAS with Gradient Matching"☆21Updated 2 years ago
- When Does Label Smoothing Help?_pytorch_implementationimp☆126Updated 5 years ago
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆82Updated last year
- A pytorch implementation of paper 'Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillation', …☆169Updated 3 years ago
- This repository is an implementation for the loss function proposed in https://arxiv.org/pdf/2110.06848.pdf.☆112Updated 3 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- Domain Generalization with MixStyle (ICLR'21)☆286Updated 2 years ago
- Pytorch implementation of Learning Disentangled Representations via Mutual Information Estimation (ECCV 2020)☆76Updated 3 years ago
- Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆131Updated 2 years ago
- Transformer based Self-Attention for Complex Numbers☆12Updated 3 years ago
- ☆34Updated 2 years ago
- Official implementation of paper "Knowledge Distillation from A Stronger Teacher", NeurIPS 2022☆140Updated 2 years ago
- [ECCV 2022] Multi-Domain Long-Tailed Recognition, Imbalanced Domain Generalization, and Beyond☆131Updated 2 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆317Updated 3 years ago
- Pytorch implementation for t-SNE with cuda to accelerate☆329Updated last year
- [ICLR 2021 Spotlight] Code release for "Long-tailed Recognition by Routing Diverse Distribution-Aware Experts."☆267Updated last year
- Code for ICML 2022 paper — Efficient Test-Time Model Adaptation without Forgetting☆113Updated last year
- An implementation of Maximum Mean Discrepancy (MMD) as a differentiable loss in PyTorch.☆31Updated 2 years ago
- ☆16Updated 2 years ago