3 个月前

特征对齐作为一种生成过程

特征对齐作为一种生成过程

摘要

在人工神经网络中,可逆性使得我们能够根据输出结果恢复输入信息。本文提出了一种名为“特征对齐”(feature alignment)的方法,用于近似任意神经网络的可逆性。该方法通过最小化数据点输出与随机输入所生成的随机输出之间的距离来训练网络。我们将该技术应用于MNIST、CIFAR-10、CelebA和STL-10等图像数据集。实验结果表明,仅凭潜在表示(latent representation)即可大致恢复原始图像,而无需依赖解码器。借助变分自编码器(variational autoencoders)的框架,我们进一步证明了生成与训练数据在统计特性上具有可比性的新图像是可行的。此外,通过将生成器与判别器联合使用,我们展示了图像质量可进一步提升。最后,我们还说明,只需对方法进行少量调整,即可将其应用于本地化网络训练,从而具有显著节省计算内存资源的潜力。

代码仓库

tiago939/feature_alignment
官方
pytorch
GitHub 中提及
tiago939/feature_aligment
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-generation-on-celeba-64x64Feature Alignment
FID: 128.35
image-generation-on-mnistFeature Alignment
FID: 37.50

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
特征对齐作为一种生成过程 | 论文 | HyperAI超神经