Disentanglement-In-Discriminative-Models

Course project for CS 8803 LS

We improve generalizability of CNN-based vision models to downstream tasks by disentangling learned representation space dimensions.

Our motivation, approach and results can be found in this course presentation.

Name		Name	Last commit message	Last commit date
Latest commit History 16 Commits
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
celeba_dataset.py		celeba_dataset.py
data.py		data.py
disentanglement.py		disentanglement.py
evaluation.py		evaluation.py
loss.py		loss.py
main_resnet50.py		main_resnet50.py
main_simclr.py		main_simclr.py
models.py		models.py
pets_dataset.py		pets_dataset.py
training.py		training.py
transforms.py		transforms.py
utils.py		utils.py

Provide feedback