3 个月前

双对立生成自编码器

双对立生成自编码器

摘要

我们提出一种新型生成式自编码器模型——双对立损失生成自编码器(Dual Contradistinctive Generative Autoencoder, DC-VAE),通过引入双重对立损失机制,显著提升生成式自编码器在同时进行推理(重构)与合成(采样)任务时的性能。DC-VAE 模型融合了实例级判别损失(用于保持重构/合成结果的实例级保真度)与集合级对抗损失(用于增强重构/合成结果的集合级保真度),二者均具备“对立相异”特性,协同作用以优化生成质量。我们在多个分辨率下(包括 32×32、64×64、128×128 和 512×512)进行了大量实验,结果表明,DC-VAE 中的两种对立损失在无需改变网络结构的前提下,能够和谐协同,显著提升模型在定性和定量指标上的表现,优于基准变分自编码器(VAE)。在图像重构、图像合成、图像插值以及表征学习等任务中,DC-VAE 均取得了当前最优或具有竞争力的性能表现。该模型为通用型变分自编码器架构,可广泛适用于计算机视觉与机器学习领域的多种下游任务。

基准测试

基准方法指标
image-generation-on-celeba-128x128DC-VAE
FID: 19.9
image-generation-on-celeba-hq-256x256DC-VAE
FID: 15.81
image-generation-on-cifar-10DC-VAE
FID: 17.9
image-generation-on-lsun-bedroom-128-x-128DC-VAE
FID: 14.3
image-generation-on-stl-10DC-VAE
FID: 41.9
Inception score: 8.1

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
双对立生成自编码器 | 论文 | HyperAI超神经