https://github.com/minitechy/betavae_variants
beta-VAE variants trained using deep neural networks on the dSprites dataset with PyTorch
https://github.com/minitechy/betavae_variants
beta-vae deep-neural-networks disentanglement dsprites mig mutual-information pytorch sap
Last synced: 2 months ago
JSON representation
beta-VAE variants trained using deep neural networks on the dSprites dataset with PyTorch
- Host: GitHub
- URL: https://github.com/minitechy/betavae_variants
- Owner: Minitechy
- Created: 2025-01-07T14:08:17.000Z (5 months ago)
- Default Branch: main
- Last Pushed: 2025-03-10T12:05:34.000Z (2 months ago)
- Last Synced: 2025-03-15T14:12:38.320Z (2 months ago)
- Topics: beta-vae, deep-neural-networks, disentanglement, dsprites, mig, mutual-information, pytorch, sap
- Language: Jupyter Notebook
- Homepage:
- Size: 22.6 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
The results for training
1. $\beta$-VAE (betaVAE_train.ipynb) can be found at the following link:
https://drive.google.com/file/d/1uThj2zB1iKGwUw2KSAvGOAIFJgR2YEPt/view?usp=sharing
3. $\beta$-VAE-v1 (betaVAE_train_v1.ipynb) can be found at the following link:
https://drive.google.com/file/d/1LNRSRvieZzuaHV4yzzEIrwBOBegUv106/view?usp=sharing
4. $\lambda\beta$-VAE (betaVAE_recon_train.ipynb) can be found at the following link:
https://drive.google.com/file/d/1BUVlm28CsBYr3fEU4FSDfLaox4TgOsBo/view?usp=sharing
6. $\gamma\beta$-VAE can be found at the following link:
- betaVAE_MC_train.ipynb:
- All $\beta$ values: https://drive.google.com/file/d/1pCecE40WWUCcgeIg9IVd86V-qrTLXBBO/view?usp=sharing
- $\beta=[1, 2, 4]$: https://drive.google.com/file/d/1WL9eWGq2NeiKS9zjjq_0SOwkh99MqTs_/view?usp=sharing
- betaVAE_VLB_train.ipynb: https://drive.google.com/file/d/1roQc5qTifdYv3YjqXjuNlkGI0ba9VjPU/view?usp=sharing