Jan, 2024

分支可变自动编码器分类器

TL;DR本研究介绍了一种改进的变分自动编码器(VAEs),该编码器包含了一个额外的神经网络分支。这个分支 VAE(BVAE)通过将分类标签基于总损失中的类别信息加入到潜在表示中,从而提供了分类信息。结果上,输入类别的潜在空间分布被分离和排序,从而提高了分类准确性。通过使用基准 MNIST 数据集对未旋转和旋转数字进行数值计算来量化提升程度。然后将所提出的技术与具有固定输出分布的 VAE 进行比较并整合。研究发现,这种方法在广泛的输出分布范围下可以获得改进性能。