公开(公告)号：US20210397945A1

公开(公告)日：2021-12-23

申请号：US17089492

申请日：2020-11-04

Applicant: NVIDIA CORPORATION

Inventor： Arash VAHDAT ,  Jan KAUTZ

IPC: G06N3/08 ,  G06N3/04

Abstract: One embodiment of the present invention sets forth a technique for performing machine learning. The technique includes inputting a training dataset into a variational autoencoder (VAE) comprising an encoder network, a prior network, and a decoder network. The technique also includes training the VAE by updating one or more parameters of the VAE based on a smoothness of one or more outputs produced by the VAE from the training dataset. The technique further includes producing generative output that reflects a first distribution of the training dataset by applying the decoder network to one or more values sampled from a second distribution of latent variables generated by the prior network.