资源下载地址:/download/sheziqiong/85908050
资源下载地址:/download/sheziqiong/85908050
ClipCap:让计算机学会看图说话
项目简介
Image Caption即我们常说的看图说话:给定一张图片,生成该图片对应的自然语言描述。
该任务涉及到了图像与自然语言两个模态,然而图像空间与自然语言空间本就十分庞大,并且两者之间存在巨大的语义鸿沟。
如何将两个庞大的语义空间进行对齐,这是该任务的重点。本项目对ClipCap: CLIP Prefix for Image Captioning
论文进行介绍,并且对论文在Flickr30k中文数据集上进行实验复现和效果展示。
论文概述
模型总览
ClipCap提出了一种基于Mapping Network的Encoder-Decoder模型,其中Mapping Network扮演了图像空间与文本空间之间的桥梁。模型主要分为三部分:
图像编码器:采用CLIP模型,负责对输入的图像进行编码,得到一个图片向量clip_embed。Mapping Network:扮演图像空间与文本空间之间的桥梁,负责将图片向量clip_embed映射到文本空间中,得到一个文本提示向量序列prefix_embeds。文本解码器:采用GPT2模型,根据提示向量序列prefix_embeds,生成caption。
论文中直接使用了预训练好的CLIP与GPT2-Large权重,两者分别是非常优秀的图像编码器和文本生层器,
能够将图像与文本映射到较好的语义空间中。然而两个模型是使用不同的任务进行训练的,语义空间没有进行对齐。
对于该问题,作者设计了两种Mapping Network,负责将图像与文本空间进行对齐。
MLP Mapping Network
使用多层全连接层作为CLIP与GPT2模型之间的桥梁。
具体做法如下:首先将图片向量clip_embed经过两层全连接层,映射成提示向量prefix_embeds,该向量作为提示信息输入到GPT2模型中,
然后GPT2模型根据prefix_embeds生成图片对应的caption。
Transformer Mapping Network
使用Transformer作为CLIP与GPT2模型之间的桥梁。具体做法如下:
首先将图片向量clip_embed经过一层全连接层,映射成提示向量序列prefix_embeds。Transformer中还维护了一个可学习的constant_embeds,将prefix_embeds与constant_embeds进行concat操作,然后输入到Transforemer中,让prefix_embeds与constant_embeds进行充分的交互。将prefix_embeds在Transformer对应的输出作为最终的prefix_embeds,作为GPT2模型的提示信息,生成最终的caption。
在该方法中,作者引入了constant embeds,这是一个向量序列,目的是为了让其能够在多头注意力中,捕获到prefix_embeds中携带的语义信息。
项目结构
datasets:存放数据models:存放自己实现的BERT模型代码output:输出目录pretrain_models:预训练模型存放位置scripts:脚本存放位置dataset.pypredict.py:根据图片生成captionprocess_caption.py:将flickr中文caption整理成统一格式process_clickr.py:对图片进行编码,获得image-caption训练数据statistics.py:对flickr数据集的caption长度分布进行统计train.py:训练脚本使用方法
Quick Start
安装依赖包:
pip install -r requirements.txt
数据预处理(主要是对图像进行编码,得到train.pkl):
python process_flickr.py
训练MLP+GPT2 tuning:
bash scripts/train_finetune_gpt2.sh
训练Bert+GPT2 no_tuning:
bash scripts/train_no_finetune_gpt2.sh
使用MLP+GPT2 tuning进行生成:
bash scripts/predict_finetune_gpt2.sh
使用Bert+GPT2 no_tuning进行生成:
bash scripts/predict_no_finetune_gpt2.sh
实验介绍
数据集
图片与caption分别来自:Flickr30k数据集 与 机器翻译得到的中文caption数据
使用脚本,对caption进行预处理:
python process_caption.py
caption被统一处理成以下格式,其中第一列为image_id,第二列为caption,并且每张图有多个caption:
1000092795两个年轻小伙子的头发浓密的看着他们的手在院子里闲逛1000092795两只年轻的白色的男性在附近的许多灌木丛1000092795绿色衬衫的男人站在一个院子里1000092795一个穿蓝色衬衫的人站在花园里1000092795朋友们一起享受时光10002456几名在硬帽的男人是一个巨大的滑轮系统10002456工人从上面的一块设备上往下看10002456一个戴着头盔的机器的男人10002456四个男人在一个高大的结构上10002456三个男人在一个大的钻机
caption的长度分布如下图:
由于caption是使用机器翻译的,所以数据集中会存在翻译结果不佳的问题。翻译的badcase如下:
1003428081至少有四个房间的窗帘玩乐器演奏单簧管等乐器1003428081四名妇女在一个客厅其中三个是清楚地演奏一种乐器1003428081一群老年妇女发挥单簧管一起为他们读出乐谱100444898一个大的结构已经坏了正在铺设一个巷道1007205537看门人推小车在管理员工具10082348在一个咖啡杯小便一个人站100845130五人走在五彩的天空背景
除此之外,据观察数据集中存在很大一部分数据是以【一个、一群、一对】等数量为开头的caption,所以该数据集是有偏的。例如:
106691539一个外科手术的外科医生和病人1067180831一只黑白相间的狗正试图在一个低切码中捕捉到一个黄色和紫色的物体1067180831一只黑色的狗正跳起来去抓一只紫色和绿色的玩具1067180831一只黑白相间的狗一只黄色的玩具跳了起来1067180831一只狗跳起来抓玩具1067675215一个人躺在一个棕色的和皮卡之间的门黄在停车场的垫子1067675215一个赤裸上身的男子躺在一条繁忙的马路中间1067675215一个穿着蓝色短裤的人在停车场外面躺了下来1067675215一个穿着蓝色短裤的人躺在街上1067790824一条白色和黑色的狗一只棕色的狗在沙地上
基于flickr中文caption存在以上问题,后续会尝试使用更高质量的中文数据集进行训练。
实验细节
图像编码器与文本生成器分别使用CLIP与中文GPT2,预训练权重分别使用的是【ViT-B/32 】【GPT2模型分享-通用中文模型 】。Mapping Network分别使用MLP与Bert进行实现(原文的Transformer本质上也是8层多头注意力进行堆叠)训练的时候,CLIP模型的权重均被冻结。在MLP+GPT2 tuning的训练中MLP与GPT2的权重均进行finetune;而在Bert+GPT2 no_tuning的训练中,只对Bert的权重进行finetune,并且Bert没有加载预训练权重,为随机初始化。Bert为8层多头注意力的堆叠。训练时的batch size为40,epoch为40,学习率为2e-5,prefix_len为10,constant_len为10,clip_size为512,warmup_step为5000。在生成阶段,对于每张图片,使用topp采样(核采样)生成10个候选的caption, 其中p设为0.8。
训练过程分析
训练过程中,模型在验证集上的loss的变化,可以在output文件夹下查看,运行如下脚本:
tensorboard --logdir ./output
训练过程中,模型在验证集上的变化曲线如下图:
简单的总结:
仅从验证集loss的角度,可以看到,相比于Bert+GPT2 no_tuning,MLP+GPT2 tuning的收敛速度更快,并且能够达到更好的效果。
这也比较符合常理,因为在MLP+GPT2 tuning中,MLP与GPT2模型的参数是一起训练的,使得文本空间往图像空间靠拢。而在Bert+GPT2 no_tuning中,
只有Bert参数在进行训练,而GPT2是固定的,因此Bert的训练难度更高。训练30-40k步的时候,模型趋于收敛,继续训练模型loss不降反升,说明此时模型已经过拟合了。可以尝试使用更大的数据集来解决该问题。
生成效果
为了验证模型的caption生成效果,我们分别在dev和test这两个小批量图片上进行caption生成。对于每张图片,使用topp采样(核采样)生成10个候选的caption,
其中p设为0.8。dev与test数据说明如下:
Dev生成效果
下表展示了模型在Dev图片上的生成效果。其中每张图对应两个生成caption,其中第一行为MLP+GPT2 tuning生成的,第二行为Bert+GPT2 no_tuning生成的
多结果展示
下面为多结果展示,可以明显看到MLP+GPT2 tuning的生成效果好于Bert+GPT2 no_tuning。更多生成caption可查看目录output/dev。
MLP+GPT2 tuning生成的10个case如下:
一个穿红色衬衣和牛仔裤的人在木制的建筑物外面走一个穿着牛仔裤和恤衫的年轻人站在一个棕色的混凝土结构外一个穿红色衬衣的人正在走下楼梯一个穿着牛仔裤和一件红色恤衫的人在楼梯上走着一个穿着蓝色牛仔裤和红色恤衫的人在外面一个人走在一个高高的篱笆上一个人站在一个残桩上一个水桶一个穿着牛仔裤和恤衫的年轻人站在一个木门旁边一个手推车一个男人正走在一个停车场一个人带着一顶红色的帽子和一条红色的围巾看起来他正走在一个小木屋里
Bert+GPT2 no_tuning生成的10个case如下:
一个穿着红色衬衫的人在一个自行车棚一个穿红色衬衫的男人在一个空旷的街道一个年轻的男人走在一个小的院子一个穿着灰色裤子和灰色衬衫的人是在一个农庄里走的一个穿红色衬衫的年轻人走在红色衬衫外面的围栏外一个穿着灰色和黑色夹克的男人走在一个地方一个人在房间里打着什么一个人踩在一个红色的高跟鞋的门槛上一个年轻人从楼上的建筑物开始跑步穿着拖鞋在院子里一个穿着衬衫和蓝色牛仔裤的男人正在操作他的人
Test生成效果
下表展示了MLP+GPT2 tuning模型在Test图片上的生成效果。更多生成caption可查看目录output/test。
多结果展示
可以看到模型生成的前几个caption,无论语义还是流畅度都与图片相符,即使后面几个case在流畅性方面存在不足,但是却都能捕获到图片中【女人、白色衣服】这一关键信息。
一个女人穿着白色的衣服走到田野里一个穿着白衣服的女人在外面散步一个穿着白色连衣裙的女人一个穿着白色连衣裙的女人走在一个岩石的田野里一个穿着白衣服的女人走下一条土路一个穿着白色衣服的女人正走在一棵大树下一个穿着白衣服的女人正走着一棵树一个女人正走着一个白色的花在一个山谷一个女人在一个白色的衣服和一顶白色的帽子在田野里散步一个穿着白衣服的女人正走着一棵棕色的灌木
可以看到,在下列生成的10个caption里,几乎所有caption都捕获到了关键信息【电脑】,并且绝大部分caption都能较为准确地对图片进行描述
一个人在电脑上工作在一个办公室里工作的女性一个戴着耳机的人在电脑上工作一个女人坐在一个苹果笔记本电脑前一个女人用一台笔记本电脑在电脑上做了一些工作一个年轻人用一台笔记本电脑做演示一个人坐在一台笔记本电脑上一个穿着黑色衬衫的男人在他的电脑上工作一个人在电脑上的数据一个女人在看电脑屏幕而在她的笔记本电脑
可以看到,模型倾向于以【一个】作为开头去生成caption,这是由于训练集本身就是有偏的,大部分数据都是以数量词作为开头。
对于该图片,虽然模型生成的大部分caption都没能百分百准确地描述语义信息,但是却生成了【湖、敲、夕阳、日落、水】等于图片语义相关的关键词,
所以模型在一定程度上还是能够将图片与文本的语义信息联系起来。
一艘船在湖上钓鱼一个人在一座桥上的湖上看日落有一艘船在一个码头上看到一个水船上的夕阳和船的背景一个人在水里钓鱼船上有很多水的背景在日落时一艘船在水中航行一个人站在码头旁的一个水上看着一条船上的夕阳日落时分一个人站在一个湖上的船上一个人正站在一个码头上看着水和帆船
模型与数据权重分享
资源下载地址:/download/sheziqiong/85908050
资源下载地址:/download/sheziqiong/85908050