Bootstrap

Neural Discrete Representation Learning-代码分析

VAE

VAE包括一下几部分:

1)一个encoder网络,对后验分布 q(z|x) 进行参数化,z 是离散隐随机变量,x 为输入数据;

2)先验分布 p(z);

3)一个decoder网络,它的输入数据分布是 p(x|z)。

VQ-VAE

使用离散隐变量,受向量量化的启发用一种新的方式进行训练。后验和先验分布是明确分类的,从这些分布中提取的样本可通过嵌入表进行索引。然后将这些嵌入作为解码器网络的输入。

网络结构如下:

这里写图片描述

首先输入x图像encoder得到特征 ze(x) ,之后查找Embedding Space中与特征 ze(x)

;