Cranial-XIX / FAMOLinks
Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)
☆116Updated last year
Alternatives and similar repositories for FAMO
Users that are interested in FAMO are comparing it to the libraries listed below
Sorting:
- Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]☆237Updated 7 months ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆144Updated 2 years ago
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆140Updated 3 years ago
- Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)☆51Updated 2 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆394Updated 4 years ago
- PyTorch implementation of the GradNorm☆117Updated last year
- Code release of paper "ForkMerge: Mitigating Negative Transfer in Auxiliary-Task Learning" (NeurIPS 2023)☆17Updated 2 years ago
- [ICLR 24 Oral/Outstanding Paper Honorable Mention Award 🎉]☆39Updated last year
- ☆72Updated last year
- C-Mixup for NeurIPS 2022☆73Updated 2 years ago
- [NeurIPS 2023] The PyTorch Implementation of Scheduled (Stable) Weight Decay.☆62Updated 2 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)☆146Updated 3 years ago
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆69Updated 2 years ago
- Official repository of "Pareto Manifold Learning: Tackling multiple tasks via ensembles of single-task models" [ICML 2023]☆23Updated last year
- Code for "Gradient Surgery for Multi-Task Learning"☆345Updated 5 years ago
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆85Updated 2 years ago
- ☆74Updated 2 years ago
- Using pre-trained Diffusion models as priors for inference tasks☆210Updated 3 years ago
- Bi-level Optimization for Advanced Deep Learning☆47Updated 4 years ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆236Updated 2 years ago
- The official PyTorch implementation of the paper: Xili Dai, Shengbang Tong, et al. "Closed-Loop Data Transcription to an LDR via Minimaxi…☆64Updated 3 years ago
- ICLR 2023 Paper submission analysis from https://openreview.net/group?id=ICLR.cc/2023/Conference☆107Updated 3 years ago
- Official implementation of Auxiliary learning as an Bargaining Game.☆32Updated 2 years ago
- Official implementation of ORCA proposed in the paper "Cross-Modal Fine-Tuning: Align then Refine"☆73Updated last year
- ☆47Updated 3 years ago
- Crawl & Visualize ICLR 2023 Data from OpenReview☆86Updated 3 years ago
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆80Updated 3 years ago
- Some thoughts about writing scientific papers☆21Updated last year
- About Code release for "Flowformer: Linearizing Transformers with Conservation Flows" (ICML 2022), https://arxiv.org/pdf/2202.06258.pdf☆331Updated last year
- Official Pytorch's implementation of RotoGrad☆80Updated 2 years ago