search

wpeebles / hessian_penalty

Official Implementation of "The Hessian Penalty: A Weak Prior for Unsupervised Disentanglement" (ECCV 2020 Spotlight)
BSD 2-Clause "Simplified" License
207 stars 21 forks source link

This project can be applied to VAE? #3

Closed pandsia2007 closed 2 years ago

pandsia2007 commented 2 years ago

This project can be applied to VAE?

wpeebles commented 2 years ago

Hi @pandsia2007. In principle it can be applied to VAEs. There are different ways you could go about it, but probably the most natural would be to apply the Hessian Penalty to the output of the decoder w.r.t. the input latent code. We haven't experimented much beyond applying it to GANs, though.

pandsia2007 commented 2 years ago

Thank you very much.