4 个月前

L-Verse:图像与文本之间的双向生成

L-Verse:图像与文本之间的双向生成

摘要

超越了对自然语言长距离交互的学习,变压器(Transformers)正因其强大的能力和可扩展性成为许多视觉任务的事实标准。特别是在图像与文本之间的跨模态任务中,向量量化变分自编码器(VQ-VAEs)被广泛用于将原始RGB图像转换为特征向量序列。为了更好地利用图像与文本之间的相关性,我们提出了一种新的架构——L-Verse,该架构由特征增强变分自编码器(AugVAE)和双向自回归变压器(BiART)组成,用于图像到文本和文本到图像的生成。我们的AugVAE在ImageNet1K验证集上展示了最先进的重建性能,并且对野外未见过的图像具有鲁棒性。与其他模型不同,BiART可以区分图像(或文本)作为条件参考和生成目标。L-Verse可以直接用于图像到文本或文本到图像的生成,无需任何微调或额外的目标检测框架。在定量和定性实验中,L-Verse在MS-COCO Captions数据集上的图像到文本和文本到图像生成任务中均表现出色,优于以往的方法。此外,我们还评估了L-Verse架构在Conceptual Captions数据集上的可扩展性,并展示了其在通用领域双向视觉-语言表示学习方面的初步结果。

代码仓库

tgisaturday/L-Verse
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-captioning-on-coco-captionsL-Verse
BLEU-4: 39.9
METEOR: 31.4
ROUGE-L: 60.4
SPICE: 23.3
image-reconstruction-on-imagenet-256x256AugVAE-SL
FID: 3.28
image-reconstruction-on-imagenet-256x256AugVAE-ML
FID: 1.04
text-to-image-generation-on-cocoL-Verse
FID: 45.8
FID-1: 41.9
FID-2: 35.5
FID-4: 30.2
FID-8: 29.83
text-to-image-generation-on-cocoL-Verse-CC
FID: 37.2
FID-1: 31.6
FID-2: 25.7
FID-4: 21.4
FID-8: 21.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
L-Verse:图像与文本之间的双向生成 | 论文 | HyperAI超神经