/ECRT

Supercharging Imbalanced Data Learning WithCausal Representation Transfer

Primary LanguagePython

ECRT: Energy-based Causal Representation Transfer

Code for Supercharging Imbalanced Data Learning With Energy-basedContrastive Representation Transfer (NeurIPS 2021)

Model

Prerequisites

The algorithm is built with:

  • Python (version 3.7 or higher)
  • Numpy (version 1.16 or higher)
  • PyTorch (version 1.3.1)

Clone the repository, e.g.:

git clone https://github.com/ZidiXiu/ECRT.git

Running the Toy Dataset

Here we present a toy synthetic dataset

python train train_CRT_toy.py

When building the CRT framework, we referenced the following sources: