Variational Autoencoder for Deep Learning of Images, Labels and Captions
About
A novel variational autoencoder is developed to model images, as well as associated labels or captions. The Deep Generative Deconvolutional Network (DGDN) is used as a decoder of the latent image features, and a deep Convolutional Neural Network (CNN) is used as an image encoder; the CNN is used to approximate a distribution for the latent DGDN features/code. The latent code is also linked to generative models for labels (Bayesian support vector machine) or captions (recurrent neural network). When predicting a label/caption for a new image at test, averaging is performed across the distribution of latent codes; this is computationally efficient as a consequence of the learned CNN-based encoder. Since the framework is capable of modeling the image in the presence/absence of associated labels/captions, a new semi-supervised setting is manifested for CNN learning with images; the framework even allows unsupervised CNN learning, based on images alone.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | ImageNet 2012 (val) | -- | 205 | |
| Image Classification | ILSVRC 2012 (test) | Top-1 Acc48.41 | 117 | |
| Adversarial Detection | ImageNet BLIP-2 | Detection Rate72 | 33 | |
| Adversarial Detection | ImageNet BLIP | Detection Rate80 | 24 | |
| Adversarial Detection | ImageNet Img2Prompt | Detection Rate78 | 23 | |
| Image Classification | ImageNet 10% labels 1K (val) | Top-5 Error35.24 | 18 | |
| Adversarial Detection | ImageNet UniDiffuser | Detection Rate70 | 12 | |
| Adversarial Detection | ImageNet MiniGPT-4 | Detection Rate53 | 12 | |
| Adversarial Detection | ImageNet UniDiffuser (test) | Detection Rate51 | 12 |