Skip to content

Training latent spaces in a DCNN Autoencoder network with the FMNIST dataset.

Notifications You must be signed in to change notification settings

jovanneste/Training-DCNN-Autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Training-DCNN-Autoencoder

This network combines an autoencoder with a fully connected layer to simultaneously learn a reconstruction and classify images from the FMNIST dataset. In this way, we can use the latent space representation from the autoencoder to classify images and improve accuracy by forcing the network to learn good encodings. To update the weights of this network, we can use either the classifier loss (between the predicted output and the correct label), the encoder loss (between the reconstructed image and the input) or some joint loss. We demonstrate that using a joint loss improves model classification accuracy.

About

Training latent spaces in a DCNN Autoencoder network with the FMNIST dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages