如何将多模态数据融入到BERT架构中-多模态BERT的两类预训练任务

news/2024/10/23 9:28:03/

大家好,我是DASOU;

回到2018年BERT刚刚横空出世,如果想快速搞一篇BERT的多模态预训练论文,应该从哪些方面去考虑呢?

本文讲两个问题,把多模态BERT知识点串起来【绝对原创,至少我还没看到这么讲过的博文】:

  1. 如何将MLM和多模态数据融合
  2. 如何将NSP任务和多模态数据融合

BERT中的大部分模块都是已经有的,它最大的作用就是证明了可以通过文本重建的方式从大量的无监督语料中获取到知识;

那么我们现在思考的问题就是如何从多模态数据中,使用BERT的架构,学习到有用的知识;

BERT有两个任务,一个是MLM。一个是NSP;

MLM是做文本重建,NSP是做句间关系;

1. 如何将MLM和多模态数据融合

MLM我们需要从三个方面去考虑:

  1. MLM输入形式是什么?
  2. mask的时候需要注意什么?
  3. 输出形式是什么,损失函数是什么?

在多模态场景下,对MLM任务,需要分为两个方向,一个是对文本的重建,称之为Masked Language Modeling (MLM),一个是对图像的重建,称之为Masked Region Modeling(MRM);

文本这边的MLM很简单,和BERT原始本身没区别,就不赘述了;

有意思的是图像重建:MRM;

首先拿到一张图片,要想把这个图片送入到TRM中去,需要的是多个图片tokens;

有几种方式可以做到这一点,首先第一个就是将图片分为一个个的patch,这个老生常谈了,TRM在CV中的应用大部分都是这种方式;

还有一种就是使用Faster-RCNN对图片做目标检测,获取到一个个的含有物体的regions,那么这个regions就是可以认为是一个个的tokens;

这个时候会出现一个问题,我们思考BERT中的文本tokens的输入,不仅仅是embeddings,而且还有position embeddings;

这是因为TRM中tokens之间是无序的,需要使用position embeddings来标明顺序;

那么回到图像这里,用什么来标明顺序呢?一般来说使用的是Faster-RCNN中输出的regions的locations信息【5维或者是7维度】;

仿照文本,我们需要把图片regions的表征和地理位置的表征加起来,由于维度不一致,所以加起来之前需要各自过一个全链接层;

那么【mask】怎么去操作呢,在操作的时候需要注意什么呢?

文本这边还是直接使用【mask】符号去mask掉子词就可以;

那么在图片这边,直接使用全零向量替代掉mask掉的图片regions就可以了;

这里有一个细节很有意思,在mask的时候我们有两种选择,就是文本和图片是混合mask的或者文本和图片是conditional masking;

文本和图片是混合的,就是说明我们在mask的时候不区分图片或者文本,随机mask;

文本和图片是conditional mask,就是说我在mask文本的时候,保持图片是完整的,在mask图片的时候,保持文本是完整的;

这两方式哪种好呢?

我们这么来想:

假如你的句子中存在【苹果】这个单词,而且图片中有【苹果】这个region,那么在mask的时候,会不会存在在mask掉【苹果】这个词汇的时候,同时mask掉了【苹果】这个区域图像呢?

肯定有概率存在这种情况。

所以conditional mask一般来说会更好一点。

我们在来说MLM的第三个问题,输出形式是什么或者说损失函数是什么?

文本这边就是softmax之后找是哪一个单词,从而进行更新梯度;

图片这边会更复杂一点,一般来说分为三种形式,这主要是对于一个图片我们可以使用三种方式描述它;

首先第一种就是使用Faster-RCNN的ROI pooled feature去描述这个图片区域,那么我们就可以使用mask的图片区域的TRM输出的向量接一个全连接打到相同维度,和ROI pooled feature进行一个L2

第二个就是,比如说我现在有图片中物体类别有50个类别,那么当前图片区域的输出就可以是一个50个类别软标签(做了softmax的归一化到概率),这样可以和TRM的输出做KL散度;

第三个是承接第二个,我们可以使用概率最大的那个作为当前区域的类别,也就是得到了一个one-hot,也就是要给硬标签,这个直接做交叉熵就可以

2. 多模态数据如何做NSP任务呢?

其实很简单,NSP任务本质上是做句子间的关系,那么我们只需要类比的做一个图片和文本之间是否匹配的任务就可以了,也就是ITM任务;

ITM本质上是从文本整体和图片整体来做关系,还有的会从字和单个图片区域做关系学习,比如Word-Region Alignment (WRA) ;

多模态这块有点乱,但是大体上就是按照MLM和NSP任务扩展到多模态数据上,这么理解会更容易一些;


http://www.ppmy.cn/news/1204089.html

相关文章

nanodet训练自己的数据集、NCNN部署到Android

nanodet训练自己的数据集、NCNN部署到Android 一、介绍二、训练自己的数据集1. 运行环境2. 数据集3. 配置文件4. 训练5. 训练可视化6. 测试 三、部署到android1. 使用官方权重文件部署1.1 下载权重文件1.2 使用Android Studio部署apk 2. 部署自己的模型【暂时存在问题】2.1 生成…

【机器学习】梯度下降预测波士顿房价

文章目录 前言一、数据集介绍二、预测房价代码1.引入库2.数据3.梯度下降 总结 前言 梯度下降算法学习。 一、数据集介绍 波士顿房价数据集:波士顿房价数据集,用于线性回归预测 二、预测房价代码 1.引入库 from sklearn.linear_model import Linear…

Spring Cloud - 手写 Gateway 源码,实现自定义局部 FilterFactory

目录 一、FilterFactory 分析 1.1、前置知识 1.2、分析源码 1.2.1、整体分析 1.2.2、源码分析 1.3、手写源码 1.3.1、基础框架 1.3.2、实现自定义局部过滤器 1.3.3、加参数的自定义局部过滤器器 一、FilterFactory 分析 1.1、前置知识 前面的学习我们知道&#xff0c…

两数之和(哈希解法)

题目描述 给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。 你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。 你可以按任…

面包屑实现

背景:面包屑根据菜单内容显示不同内容。首页永远存在,后面的活动管理及多级菜单的面包屑展示。 实现原理: 通过this.$route.matched获取所有匹配路由,将处理首页外的其他路由设置到一个数组中,再通过数组循环方式显示…

界面组件Telerik UI for WinForms中文教程 - 如何自定义应用程序文件窗口?

Telerik UI for WinForms包含了一个高度可定制的组件,它取代了.NET中默认的OpenFileDialog。在下一个更新版本中,会发布一个向对话框浏览器提那家自定义位置的请求功能,本文演示了这个和另一个自定义功能,它可以帮助用户在浏览文件…

美容美发行业小程序源码系统+在线预约+会员卡 带完整的搭建教程

大家好啊,今天源码小编又来给大家分享了,今天要分享的是一款美容美发行业小程序源码系统。系统采用的是PHPMySQL组合开发。现如今,美容美发行业的发展迅速,市场竞争也日趋激烈。为了满足市场需求,提高自身竞争力&#…

第八章《搞懂算法:逻辑回归是怎么回事》笔记

8.1 如何理解逻辑回归 逻辑回归根据给定的自变量数据集来估计事件的发生概率,由于结果是一个概率,因此因变量的范围在 0 和 1 之间。 逻辑回归的与线性回归一样,也是以线性函数为基础的;而与线性回归不同的是,逻辑回…