AI-repo

img2dataset

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

PyHealth

A Deep Learning Python Toolkit for Healthcare Applications.

openmixup

CAIRI Supervised, Semi- and Self-Supervised Visual Representation Learning Toolbox and Benchmark

BagofTricks-LT

A scientific and useful toolbox, which contains practical and effective long-tail related tricks with extensive experimental results

Parametric-Contrastive-Learning

Parametric Contrastive Learning (ICCV2021) & GPaCo (TPAMI 2023)

MIMIC-IV-Data-Pipeline

A customizable pipeline for data extraction from MIMIC-IV.

LTR-weight-balancing

CVPR 2022 - official implementation for "Long-Tailed Recognition via Weight Balancing" https://arxiv.org/abs/2203.14197

Generalized-Long-Tailed-Benchmarks.pytorch

[ECCV 2022] A generalized long-tailed challenge that incorporates both the conventional class-wise imbalance and the overlooked attribute-wise imbalance within each class. The proposed IFL together with other baselines are also included.

HAIM

This repository contains the code to replicate the data processing, modeling and reporting of our Holistic AI in Medicine (HAIM) Publication in Nature Machine Intelligence (Soenksen LR, Ma Y, Zeng C et al. 2022).

Balanced-Contrastive-Learning

Code Release for “Balanced Contrastive Learning for Long-Tailed Visual Recognition”

PageNet

Official implementation of PageNet (IJCV 2022)

CLUDA

Contrastive Learning for Domain Adaptation of Time Series

BIOT

NeurIPS2023 - A generic biosignal learning framework. Large EEG pre-trained models.

MetaSAug

[CVPR 2021] MetaSAug: Meta Semantic Augmentation for Long-Tailed Visual Recognition

ViRB

Visual Representation Learning Benchmark for Self-Supervised Models

SHIKE

Long-Tailed Visual Recognition via Self-Heterogeneous Integration with Knowledge Excavation (CVPR 2023)

OPERA

Official implementation of "OPERA: Omni-Supervised Representation Learning with Hierarchical Supervisions"

Regularized-Self-Labeling

A regularized self-labeling approach to improve the generalization and robustness of fine-tuned models

CALS

Code for our method CALS (Class Adaptive Label Smoothing) for network calibration. To Appear at CVPR 2023. Paper: https://arxiv.org/abs/2211.15088

ncl

Code of the paper "Neighborhood Contrastive Learning Applied to Online Patient Monitoring"

trex

PyTorch implementation of the paper "No reason for no supervision: Improving the generalization of supervised models"

Floss

alm-dnn

Constrained Optimization to Train Neural Networks on Critical and Under-Represented Classes [NeurIPS2021]

Unsupervised_pre-training_of_graph_transformers_on_patient_population_graphs

Official github repository for the paper "Unsupervised pre-training of graph transformers on patient population graphs"

GenSCL

Official Pytorch implementation of "Generalized Supervised Contrastive Learning Framework"

iif

Inverse Image Frequency for long-tailed image recognition. Accepted in Transactions on Image Processing - Arxiv version available at: https://arxiv.org/abs/2209.04861

H-SCL

Inmplementation of paper

AFA

PyTorch implementation for "Towards Effective Deep Transfer via Attentive Feature Alignment" (NN2021)

COIN

About This repository is the official implementation of "Improving Fine-tuning of Self-supervised Models with Contrastive Initialization".

HiPA-Framework