Awesome Long-Tail Learning [](https://github.com/sindresorhus/awesome)

August 1, 2024 ยท View on GitHub

This repo pays special attention to the long-tailed distribution, where labels follow a long-tailed or power-law distribution in the training dataset and/or test dataset. Related papers are summarized, including its application in computer vision, in particular image classification, and extreme multi-label learning (XML), in particular text categorization.

:high_brightness: Updated 2024-07-13

Long-tailed Learning

Type of Long-Tailed Learning Methods

TypeTSTISCBSCLWNCENSDA
MeaningTwo-Stage TrainingInstance SamplingClass-Balanced SamplingClass-Level WeightingNormalized ClassifierEnsembleData Augmentation

Long-Tailed Learning Workshops

YearVenueTitleRemark
2021CVPROpen World Visionlong-tail, open-set, streaming labels
2021CVPRLearning from Limited and Imperfect Data (L2ID)label noise, SSL, long-tail

Long-Tailed Classification

YearVenueTitleRemark
2024CVPRDeiT-LT: Distillation Strikes Back for Vision Transformer Training on Long-Tailed Datasetscode
2024ICMLHarnessing Hierarchical Label Distribution Variations in Test Agnostic Long-tail Recognitioncode
2024ICMLLearning Label Shift Correction for Test-Agnostic Long-Tailed Recognitioncode
2024ICMLLong-Tail Learning with Foundation Model: Heavy Fine-Tuning Hurts๐Ÿ”ฅ๐Ÿ”ฅ๐Ÿ”ฅ code
2023TPAMIDeep Long-Tailed Learning: A Survey
2023TPAMIProbabilistic Contrastive Learning for Long-Tailed Visual Recognitioncode
2023ICLRDelving into Semantic Scale Imbalance
2023ICLRTemperature Schedules for self-supervised contrastive methods on long-tail data
2023ICLROn the Effectiveness of Out-of-Distribution Data in Self-Supervised Long-Tail Learning
2023ICLRLong-Tailed Learning Requires Feature Learning
2023ICLRDecoupled Training for Long-Tailed Classification With Stochastic Representations
2023ICLRLPT: Long-tailed Prompt Tuning for Image Classificationfine-tune ViT
2023ICLRCUDA: Curriculum of Data Augmentation for Long-tailed Recognition
2023NeurIPSA Unified Generalization Analysis of Re-Weighting and Logit-Adjustment for Imbalanced Learning code
2023NeurIPSGeneralized Logit Adjustment: Calibrating Fine-tuned Models by Removing Label Bias in Foundation Modelscode
2023arXivExploring Vision-Language Models for Imbalanced Learningpre-trained model
2023ECCVVL-LTR: Learning Class-wise Visual-Linguistic Representation for Long-Tailed Visual Recognitionfine-tune CLIP
2023AAAIMinority-Oriented Vicinity Expansion with Attentive Aggregation for Video Long-Tailed Recognitionvideo dataset, code
2022ECCVTailoring Self-Supervision for Supervised Learningvideo dataset, code
2022NeurIPSSelf-Supervised Aggregation of Diverse Experts for Test-Agnostic Long-Tailed Recognitioncode
2022arXivLearning to Re-weight Examples with Optimal Transport for Imbalanced Classification
2022TPAMIKey Point Sensitive Loss for Long-tailed Visual Recognition
2022IJCVA Survey on Long-Tailed Visual Recognitionsurvey
2022arXivNeural Collapse Inspired Attraction-Repulsion-Balanced Loss for Imbalanced Learning
2022ICLROPTIMAL TRANSPORT FOR LONG-TAILED RECOGNI- TION WITH LEARNABLE COST MATRIX
2022ICLRSELF-SUPERVISED LEARNING IS MORE ROBUST TO DATASET IMBALANCE
2022AAAICross-Domain Empirical Risk Minimization for Unbiased Long-tailed Classificationcode
2021NeurIPSImproving Contrastive Learning on Imbalanced Seed Data via Open-World Sampling
2021NeurIPSTowards Calibrated Model for Long-Tailed Visual Recognition from Prior Perspectivecode, mixup+LA
2021arXivHAR: Hardness Aware Reweighting for Imbalanced Datasets
2021arXivFeature Generation for Long-tail Classification
2021arXivLabel-Aware Distribution Calibration for Long-tailed Classification
2021arXivSelf-supervised Learning is More Robust to Dataset Imbalance
2021ArixivLong-tailed Distribution Adaptation
2021arXivLEARNING FROM LONG-TAILED DATA WITH NOISY LABELS
2021ICCVSelf Supervision to Distillation for Long-Tailed Visual Recognition
2021ICCVDistilling Virtual Examples for Long-tailed Recognition
2021CVPRContrastive Learning based Hybrid Networks for Long-Tailed Image Classification
2021CVPRMetaSAug: Meta Semantic Augmentation for Long-Tailed Visual Recognition
2021CVPRDisentangling Label Distribution for Long-tailed Visual Recognition
2021CVPRLong-Tailed Multi-Label Visual Recognition by Collaborative Training on Uniform and Re-Balanced Samplings
2021CVPRSeesaw Loss for Long-Tailed Instance Segmentation
2021ICLRExploring balanced feature spaces for representation learning
2021ICLRIS LABEL SMOOTHING TRULY INCOMPATIBLE WITH KNOWLEDGE DISTILLATION: AN EMPIRICAL STUDY
2021arXivImproving Long-Tailed Classification from Instance Level
2021arXivResLT: Residual Learning for Long-tailed Recognition
2021arXivImproving Long-Tailed Classification from Instance Level
2021arXivDisentangling Sampling and Labeling Bias for Learning in Large-Output Spacesby Google
2021arXivBreadcrumbs: Adversarial Class-Balanced Sampling for Long-tailed Recognition
2021arXivProcrustean Training for Imbalanced Deep Learning
2021arXivBalanced Knowledge Distillation for Long-tailed LearningCBS+IS, Code
2021arXivClass-Balanced Distillation for Long-Tailed Visual RecognitionENS+DA+IS, by Google Research
2021arXivDistributional Robustness Loss for Long-tail LearningTST+CBS
2021CVPRImproving Calibration for Long-Tailed RecognitionDA+TST, Code
2021CVPRDistribution Alignment: A Unified Framework for Long-tail Visual RecognitionTST
2021CVPRAdversarial Robustness under Long-Tailed Distribution
2021ICLRHETEROSKEDASTIC AND IMBALANCED DEEP LEARNING WITH ADAPTIVE REGULARIZATIONCode
2021ICLRLONG-TAILED RECOGNITION BY ROUTING DIVERSE DISTRIBUTION-AWARE EXPERTSENS+NC, Code, by Zi-Wei Liu
2021ICLRLong-Tail Learning via Logit Adjustmentby Google
2021AAAIBag of Tricks for Long-Tailed Visual Recognition with Deep Convolutional Neural Networks
2021arXivLearning From Multiple Experts: Self-paced Knowledge Distillation for Long-tailed Classification
2020arXivELF: An Early-Exiting Framework for Long-Tailed Classification
2020CVPRRethinking Class-Balanced Methods for Long-Tailed Visual Recognition from a Domain Adaptation Perspective
2020CVPREqualization Loss for Long-Tailed Object Recognition
2020CVPRDeep Representation Learning on Long-tailed Data: A Learnable Embedding Augmentation Perspective
2020ICLRDecoupling representation and classifier for long-tailed recognitionCode
2020NeurIPSBalanced Meta-Softmax for Long-Tailed Visual Recognition
2020NeurIPSRethinking the Value of Labels for Improving Class-Imbalanced LearningCode
2020CVPRBbn: Bilateral-branch network with cumulative learning for long-tailed visual recognitionCode
2019NeurIPSLearning Imbalanced Datasets with Label-Distribution-Aware Margin LossCode
2019CVPRLarge-Scale Long-Tailed Recognition in an Open WorldCode, bibtex, by CUHK
2018-iNatrualist. The inaturalist 2018 competition datasetlong-tailed dataset
2017arXivThe Devil is in the Tails: Fine-grained Classification in the Wild
2017NeurIPSLearning to model the tail

Long-Tailed Regression

YearVenueTitleRemark
2022CVPRBalanced MSE for Imbalanced Visual Regression
2021OpenReviewLIFTING IMBALANCED REGRESSION WITH SELF- SUPERVISED LEARNINGiclr rejected
2021ICMLDelving into Deep Imbalanced Regressioncode

Long-Tailed Semi-Supervised Learning

YearVenueTitleRemark
2024arXivTowards Realistic Long-tailed Semi-supervised Learning in an Open Worldcode
2024ICMLSimPro: A Simple Probabilistic Framework Towards Realistic Long-Tailed Semi-Supervised Learningcode
2023CVPRTowards Realistic Long-Tailed Semi-Supervised Learning: Consistency Is All You Need๐Ÿ”ฅ๐Ÿ”ฅ๐Ÿ”ฅcode
2023NeurIPSTowards Distribution-Agnostic Generalized Category Discoverycode
2023ICLRImbalanced Semi-supervised Learning with Bias Adaptive Classifier
2023ICLRAdaptive Robust Evidential Optimization For Open Set Detection from Imbalanced Data
2023ICLRINPL: PSEUDO-LABELING THE INLIERS FIRST FOR IMBALANCED SEMI-SUPERVISED LEARNING
2022CVPRDASO: Distribution-Aware Semantics-Oriented Pseudo-label for Imbalanced Semi-Supervised Learningcode
2022MLJTransfer and Share: Semi-Supervised Learning from Long-Tailed Datacode
2022ICMLSmoothed Adaptive Weighting for Imbalanced Semi-Supervised Learning: Improve Reliability Against Unknown Distribution Datacode
2022ICLRTHE RICH GET RICHER: DISPARATE IMPACT OF SEMI-SUPERVISED LEARNING
2022ICLRON NON-RANDOM MISSING LABELS IN SEMI-SUPERVISED LEARNING
2022OpenReviewUNIFYING DISTRIBUTION ALIGNMENT AS A LOSS FOR IMBALANCED SEMI-SUPERVISED LEARNING
2021NeurIPSABC: Auxiliary Balanced Classifier for Class-imbalanced Semi-supervised Learning
2021arXivCoSSL: Co-Learning of Representation and Classifier for Imbalanced Semi-Supervised Learning
2021CVPRCReST: A Class-Rebalancing Self-Training Framework for Imbalanced Semi-Supervised Learningby Google, Code, Tensorflow
2021arXivDISTRIBUTION-AWARE SEMANTICS-ORIENTED PSEUDO-LABEL FOR IMBALANCED SEMI-SUPERVISED LEARNINGSSL, Code
2020NeurIPSDistribution Aligning Refinery of Pseudo-label for Imbalanced Semi-supervised LearningCode

Long-Tailed Learning with Noisy Labels

YearVenueTitleRemark
2024CVPRSURE: SUrvey REcipes for building reliable and robust deep networkscode
2023ICLRLONG-TAILED PARTIAL LABEL LEARNING VIA DYNAMIC REBALANCINGcode, partial label
2023ICCVWhen Noisy Labels Meet Long Tail Dilemmas: A Representation Calibration Method
2022ECCVIdentifying Hard Noise in Long-Tailed Sample Distributioncode, large datasets
2022ICLRSAMPLE SELECTION WITH UNCERTAINTY OF LOSSES FOR LEARNING WITH NOISY LABELS
2022PAKDDPrototypical Classifier for Robust Class-Imbalanced Learningcode
2021arXivROBUST LONG-TAILED LEARNING UNDER LABEL NOISEcode

Long-Tailed OOD Detection

YearVenueTitleRemark
2024AAAIEAT: Towards Long-Tailed Out-of-Distribution Detection๐Ÿ”ฅ๐Ÿ”ฅ๐Ÿ”ฅ code

Long-Tailed Federated Learning

YearVenueTitleRemark
2022IJCAIFederated Learning on Heterogeneous and Long-Tailed Data via Classifier Re-Training with Federated Features

eXtreme Multi-label Learning

Binary Relevance

YearVenueTitleRemark
2019Machine learningData Scarcity, Robustness and Extreme Multi-label Classification
2019WSDMSlice: Scalable linear extreme classifiers trained on 100 million labels for related searches
2017KDDPPDSparse: A Parallel Primal-Dual Sparse Method for Extreme Classification
2017AISTATSLabel Filters for Large Scale Multilabel Classification
2016WSDMDiSMEC - Distributed Sparse Machines for Extreme Multi-label Classification
2016ICMLPD-Sparse: A Primal and Dual Sparse Approach to Extreme Multiclass and Multilabel Classification

Tree-based Methods

YearVenueTitleRemark
2021KDDExtreme Multi-label Learning for Semantic Matching in Product Searchby Amazon, code
2020arXivProbabilistic Label Trees for Extreme Multi-label ClassificationPLT survey, code
2020arXivOnline probabilistic label trees
2020AISTATSLdSM: Logarithm-depth Streaming Multi-label Decision TreesInstance tree,c++ code
2019NeurIPSAttentionXML: Extreme Multi-Label Text Classification with Multi-Label Attention Based Recurrent Neural NetworksLabel tree
2019arXivBonsai - Diverse and Shallow Trees for Extreme Multi-label ClassificationLabel tree
2018ICMLCRAFTML, an Efficient Clustering-based Random Forest for Extreme Multi-label LearningInstance tree
2018WWWParabel: Partitioned Label Trees for Extreme Classification with Application to Dynamic Search AdvertisingLabel tree...by Manik Varma
2016ICMLExtreme F-Measure Maximization using Sparse Probability EstimatesLabel tree
2016KDDExtreme Multi-label Loss Functions for Recommendation, Tagging, Ranking & Other Missing Label ApplicationsInstance tree
2014KDDA Fast, Accurate and Stable Tree-classifier for eXtreme Multi-label LearningInstance tree, python implementation
2013ICMLLabel Partitioning For Sublinear RankingLabel tree
2013WWWMulti-Label Learning with Millions of Labels: Recommending Advertiser Bid Phrases for Web PagesInstance tree, Random Forest, Gini Index
2011NeurIPSEfficient label tree learning for large scale object recognitionLabel tree, multi-class
2010NeurIPSLabel embedding trees for large multi-class tasksLabel tree, multi-class
2008ECML WorkshopEffective and Efficient Multilabel Classification in Domains with Large Number of LabelsLabel tree

Embedding-based Methods

YearVenueTitleRemark
2019AAAIDistributional Semantics Meets Multi-Label Learningbibtex
2019arXivRanking-Based Autoencoder for Extreme Multi-label Classification
2019NeurIPSBreaking the Glass Ceiling for Embedding-Based Classifiers for Large Ouput Spacesby Google Research
2017KDDAnnexML: Approximate Nearest Neighbor Search for Extreme Multi-label Classification
2015NeurIPSSparse Local Embeddings for Extreme Multi-label Classification
2014ICMLLarge-scale Multi-label Learning with Missing Labels
2014ICMLMulti-label Classification via Feature-aware Implicit Label Space Encoding
2013ICMLEfficient Multi-label Classification with Many Labels
2012NeurIIPSFeature-aware Label Space Dimension Reduction for Multi-label Classification
2011IJCAIWSABIE: Scaling Up To Large Vocabulary Image Annotationbibtex
2009NeurIPSMulti-Label Prediction via Compressed Sensing
2008KDDExtracting Shared Subspaces for Multi-label Classification

Speed-up and Compression

YearVenueTitleRemark
2020KDDLarge-Scale Training System for 100-Million Classification at AlibabaApplied Data Science Track
2020arXivSOLAR: Sparse Orthogonal Learned and Random Embeddings
2020ICLREXTREME CLASSIFICATION VIA ADVERSARIAL SOFTMAX APPROXIMATION
2019AISTATSStochastic Negative Mining for Learning with Large Output Spacesby Google
2019NeurIPSExtreme Classification in Log Memory using Count-Min Sketch: A Case Study of Amazon Search with 50M ProductsRice University, bibtex
2019arXivAn Embarrassingly Simple Baseline for eXtreme Multi-label Prediction
2019arXivAccelerating Extreme Classification via Adaptive Feature Agglomerationbibtex, authors from IIT
2019SDMFast Training for Large-Scale One-versus-All Linear Classifiers using Tree-Structured Initializationcode bibtex

Noval XML Settings

YearVenueTitleRemark
2020arXivExtreme Multi-label Classification from Aggregated Labelsby Inderjit Dhillon. This paper considers multi-instance learning in XML
2020arXivUnbiased Loss Functions for Extreme Classification With Missing Labelsby Rohit Babbar. Missing labels
2020ICMLDeep Streaming Label Learningcode, by Dacheng Tao, streaming multi-label learning
2016arXivStreaming Label Learning for Modeling Labels on the Flyby Dacheng Tao, streaming multi-label learning

Theoretical Studies

YearVenueTitleRemark
2019ICMLSparse Extreme Multi-label Learning with Oracle PropertyCode, by Weiwei Liu
2019NeurIPSMultilabel reductions: what is my loss optimising?bibtex, by Google

Text Classification

YearVenueTitleRemark
2022TKDEBGNN-XML: Bilateral Graph Neural Networks for Extreme Multi-label Text Classification
2021ICMLSiameseXML: Siamese Networks meet Extreme Classifiers with 100M Labels
2020KDDCorrelation Networks for Extreme Multi-label Text Classificationcode
2020arXivGNN-XML: Graph Neural Networks for Extreme Multi-label Text Classification
2020ICMLPretrained Generalized Autoregressive Model with Adaptive Probabilistic Label Clusters for Extreme Multi-label Text Classificationcode
2019ACLLarge-Scale Multi-Label Text Classification on EU LegislationEur-Lex 4.3K, bibtex
2019arXivX-BERT: eXtreme Multi-label Text Classification with BERTcode by Yiming Yang, Inderjit Dhillon
2019NeurIPSAttentionXML: Extreme Multi-Label Text Classification with Multi-Label Attention Based Recurrent Neural Networks
2018EMNLPFew-Shot and Zero-Shot Multi-Label Learning for Structured Label Spacesfew-shot, zero-shot, evaluation metric
2018NeurIPSA no-regret generalization of hierarchical softmax to extreme multi-label classificationcode, PLT code
2017SIGIRDeep Learning for Extreme Multi-label Text Classificationby Yiming Yang at CMU, bibtex

Others

Label Correlation

YearVenueTitleRemark
2019ICMLDL2: Training and Querying Neural Networks with Logic
2015KDDDiscovering and Exploiting Deterministic Label Relationships in Multi-Label Learning
2010KDDMulti-Label Learning by Exploiting Label Dependency

Long-tailed Continual Learning

YearVenueTitleRemark
2020ECCVImbalanced Continual Learning with Partitioning Reservoir Sampling

Train/Test Split

YearVenueTitleRemark
2021arXivStratified Sampling for Extreme Multi-Label Data

XML Seminar

YearVenueTitleRemark
2019Dagstuhl Seminar 18291Extreme Classification

Survey References:

  1. https://arxiv.org/pdf/1901.00248.pdf
  2. http://www.iith.ac.in/~saketha/research/AkshatMTP2018.pdf
  3. http://manikvarma.org/pubs/bengio19.pdf
  4. The Emerging Trends of Multi-Label Learning