Keras 变分自动编码器示例 - 潜在输入的使用

Question

我是z Keras 的新手，并且一直在努力理解他们官方github 中变分自动编码器示例中变量的用法。我不明白为什么z没有被使用而不是变量latent_inputs。我运行了代码，它似乎有效，但我不明白是否z在幕后使用，以及 Keras 中负责它的机制是什么。这是相关的代码片段：

# VAE model = encoder + decoder
# build encoder model
inputs = Input(shape=input_shape, name='encoder_input')
x = Dense(intermediate_dim, activation='relu')(inputs)
z_mean = Dense(latent_dim, name='z_mean')(x)
z_log_var = Dense(latent_dim, name='z_log_var')(x)

# use reparameterization trick to push the sampling out as input
# note that "output_shape" isn't necessary with the TensorFlow backend
z = Lambda(sampling, output_shape=(latent_dim,), name='z')([z_mean, z_log_var])

# instantiate encoder model
encoder = Model(inputs, [z_mean, z_log_var, z], name='encoder')
encoder.summary()
plot_model(encoder, to_file='vae_mlp_encoder.png', show_shapes=True)

# build decoder model
latent_inputs = Input(shape=(latent_dim,), name='z_sampling')
x = Dense(intermediate_dim, activation='relu')(latent_inputs)
outputs = Dense(original_dim, activation='sigmoid')(x)

# instantiate decoder model
decoder = Model(latent_inputs, outputs, name='decoder')
decoder.summary()
plot_model(decoder, to_file='vae_mlp_decoder.png', show_shapes=True)

# instantiate VAE model
outputs = decoder(encoder(inputs)[2])
vae = Model(inputs, outputs, name='vae_mlp')

Answer 1

你encoder被定义为一个接受输入inputs并给出输出的模型[z_mean, z_log_var, z]。然后，您可以单独定义解码器以获取一些输入（此处称为latent_inputs和 output ）outputs。最后，您的整体模型在以下行中定义：

outputs = decoder(encoder(inputs)[2])

这意味着您将运行encoder您的inputs，它产生[z_mean, z_log_var, z]，然后它的第三个元素（调用它result[2]）作为输入参数传入decoder. 换句话说，当你实现你的网络时，你设置latent_inputs等于编码器的第三个输出，或者[z_mean, z_log_var, z][2] = z。您可以将其视为（可能不是有效代码）：

encoder_outputs = encoder(inputs)  # [z_mean, z_log_var, z]
outputs = decoder(latent_inputs=encoder_outputs[2])  # latent_inputs = z