Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Yunchen Pu, Zhe Gan, Ricardo Henao, Xin Yuan, Chunyuan Li, Andrew Stevens, Lawrence Carin• 2016

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet 2012 (val)--
205
Image ClassificationILSVRC 2012 (test)
Top-1 Acc48.41
117
Adversarial DetectionImageNet BLIP-2
Detection Rate72
33
Adversarial DetectionImageNet BLIP
Detection Rate80
24
Adversarial DetectionImageNet Img2Prompt
Detection Rate78
23
Image ClassificationImageNet 10% labels 1K (val)
Top-5 Error35.24
18
Adversarial DetectionImageNet UniDiffuser
Detection Rate70
12
Adversarial DetectionImageNet MiniGPT-4
Detection Rate53
12
Adversarial DetectionImageNet UniDiffuser (test)
Detection Rate51
12
Showing 9 of 9 rows

Other info

Follow for update