alexlioralexli/noncontrastive-ssl

Analyzing partial dimensional collapse in non-contrastive self-supervised learning. "Understanding Collapse in Non-Contrastive Siamese Representation Learning." In ECCV, 2022.

27
/ 100
Experimental

This repository helps AI researchers and machine learning engineers analyze how self-supervised learning models, specifically those using a 'siamese' architecture, might develop redundant or 'collapsed' internal representations. It takes an ImageNet dataset and a pre-trained SimSiam model as input to train and evaluate the model's representations. The output provides metrics like k-NN accuracy and collapse scores, along with plotting capabilities to visualize the results.

No commits in the last 6 months.

Use this if you are a machine learning researcher or engineer studying the behavior of self-supervised learning models and want to understand or mitigate representation collapse in non-contrastive methods.

Not ideal if you are looking for a pre-built model for immediate application in tasks like image classification, as this tool focuses on the research and analysis of model training dynamics rather than deployment.

Machine-Learning-Research Self-Supervised-Learning Computer-Vision Representation-Learning AI-Model-Analysis
Stale 6m No Package No Dependents
Maintenance 0 / 25
Adoption 6 / 25
Maturity 16 / 25
Community 5 / 25

How are scores calculated?

Stars

16

Forks

1

Language

Jupyter Notebook

License

MIT

Last pushed

Nov 12, 2023

Commits (30d)

0

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/alexlioralexli/noncontrastive-ssl"

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