Timezone: »

 
Poster
Learning Generative Vision Transformer with Energy-Based Latent Space for Saliency Prediction
Jing Zhang · Jianwen Xie · Nick Barnes · Ping Li

Tue Dec 07 08:30 AM -- 10:00 AM (PST) @

Vision transformer networks have shown superiority in many computer vision tasks. In this paper, we take a step further by proposing a novel generative vision transformer with latent variables following an informative energy-based prior for salient object detection. Both the vision transformer network and the energy-based prior model are jointly trained via Markov chain Monte Carlo-based maximum likelihood estimation, in which the sampling from the intractable posterior and prior distributions of the latent variables are performed by Langevin dynamics. Further, with the generative vision transformer, we can easily obtain a pixel-wise uncertainty map from an image, which indicates the model confidence in predicting saliency from the image. Different from the existing generative models which define the prior distribution of the latent variables as a simple isotropic Gaussian distribution, our model uses an energy-based informative prior which can be more expressive to capture the latent space of the data. We apply the proposed framework to both RGB and RGB-D salient object detection tasks. Extensive experimental results show that our framework can achieve not only accurate saliency predictions but also meaningful uncertainty maps that are consistent with the human perception.

Author Information

Jing Zhang (Australian National University)
Jianwen Xie (Baidu Research)
Nick Barnes (Australian National University)
Ping Li (Baidu Research USA)

More from the Same Authors