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.
-
Notifications
You must be signed in to change notification settings - Fork 0
jovanneste/Training-DCNN-Autoencoder
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Training latent spaces in a DCNN Autoencoder network with the FMNIST dataset.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published