Variational autoencoder (VAE) is an established generative model but is notorious for its blurriness. In this work, we investigate the blurry output problem of VAE and resolve it, exploiting the variance of Gaussian decoder and \(\beta \) of beta-VAE [14]. Specifically, we reveal that the indistinguishability of decoder variance and \(\beta \) hinders appropriate analysis of the model by random likelihood value, and limits performance improvement by omitting the gain from \(\beta \) . To address the problem, we propose Beta-Sigma VAE (BS-VAE) that explicitly separates \(\beta \) and decoder variance \(\sigma ^2_x\) in the model. Our method demonstrates not only superior performance in natural image synthesis but also controllable parameters and predictable analysis compared to conventional VAE. In our experimental evaluation, we employ the analysis of rate-distortion curve and proxy metrics on computer vision datasets. The code is available on https://github.com/overnap/BS-VAE .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Beta-Sigma VAE: Separating Beta and Decoder Variance in Gaussian Variational Autoencoder

  • Seunghwan Kim,
  • Seungkyu Lee

摘要

Variational autoencoder (VAE) is an established generative model but is notorious for its blurriness. In this work, we investigate the blurry output problem of VAE and resolve it, exploiting the variance of Gaussian decoder and \(\beta \) of beta-VAE [14]. Specifically, we reveal that the indistinguishability of decoder variance and \(\beta \) hinders appropriate analysis of the model by random likelihood value, and limits performance improvement by omitting the gain from \(\beta \) . To address the problem, we propose Beta-Sigma VAE (BS-VAE) that explicitly separates \(\beta \) and decoder variance \(\sigma ^2_x\) in the model. Our method demonstrates not only superior performance in natural image synthesis but also controllable parameters and predictable analysis compared to conventional VAE. In our experimental evaluation, we employ the analysis of rate-distortion curve and proxy metrics on computer vision datasets. The code is available on https://github.com/overnap/BS-VAE .