NeuralCollapseApplications/ImbalancedLearning

[NeurIPS 2022] The official code for our NeurIPS 2022 paper "Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?".

28
/ 100
Experimental

This project helps machine learning researchers and practitioners investigate the effectiveness of simplified classification methods for imbalanced datasets. It takes standard image datasets, particularly those with an uneven distribution of examples per category, and outputs trained models using different classification heads and loss functions. The models are evaluated on their ability to accurately classify examples from both common and rare categories.

No commits in the last 6 months.

Use this if you are developing or evaluating deep learning models for classification tasks where certain categories have significantly fewer training examples than others.

Not ideal if you are looking for a general-purpose, out-of-the-box solution for balanced image classification or a tool for data preprocessing.

imbalanced-data deep-learning-research image-classification machine-learning-engineering model-evaluation
No License Stale 6m No Package No Dependents
Maintenance 0 / 25
Adoption 8 / 25
Maturity 8 / 25
Community 12 / 25

How are scores calculated?

Stars

49

Forks

6

Language

Python

License

Last pushed

Oct 12, 2022

Commits (30d)

0

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/NeuralCollapseApplications/ImbalancedLearning"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.