Redian新闻
>
多模态再次统一!Meta发布自监督算法data2vec 2.0:训练效率最高提升16倍!

多模态再次统一!Meta发布自监督算法data2vec 2.0:训练效率最高提升16倍!

公众号新闻



  新智元报道  

编辑:LRS
【新智元导读】效率更高的多模态自监督学习框架data2vec 2.0来啦!

近几年人工智能领域的突破大多由自监督学习推动,比如BERT中提出的MLM (Masked Language Model) ,通过将文本中的部分单词遮盖后重新预测,使得海量无标记文本数据也能用来训练模型,自此开启了大规模预训练模型的新时代。


但自监督学习算法也有明显的局限性,通常只适用于单一模态(如图像、文本、语音等)的数据,并且需要大量的算力从海量数据中进行学习。


相比之下,人类的学习效率要显著高于当前的AI模型,并且可以从不同类型的数据中进行学习。


2022年1月,Meta AI发布了自监督学习框架data2vec,将三个模态的数据(语音、视觉和文本)通过一个框架整合起来,大有一统多模态的趋势。


最近Meta AI发布了data2cec 2.0版本,主要在性能方面对上一代进行了改进:在精度相同的情况下,训练速度相比其他算法最高提升了16倍!

论文链接:https://ai.facebook.com/research/publications/efficient-self-supervised-learning-with-contextualized-target-representations-for-vision-speech-and-language

代码链接:https://github.com/facebookresearch/fairseq/tree/main/examples/data2vec


data2vec 1.0


目前来说,大部分机器学习模型仍然是基于有监督学习的模式,需要有专门的标注人员对目标数据打标签,但对于某些任务来说(比如地球上的几千种人类语言),收集标注数据是不可行的。


相比之下,自监督学习不需要告诉模型正确和错误,而是让机器通过观察世界来学习图像、语音和文本的结构。


相关的研究成果促进了语音(如,wave2vec 2.0)、计算机视觉(例如,掩码自编码器)和自然语言处理(例如,BERT)等领域的发展。


data2vec的主要思路就是先建立一个教师网络,首先计算来自图像、文本或语音的目标表征。然后对数据进行掩码遮盖掉部分输入,并用一个学生网络重复该过程预测教师模型得到的表征。



也就是说,学生模型只能在接受「不完整输入信息」的同时预测「完整输入数据」的表示。


为了保证两个模型的一致性,二者的参数时共享的,但在训练初期会让Teacher模型的参数更新更快。


在实验结果上,data2vec在语音、视觉、文本等任务上对比baseline模型性能提升明显。


data2vec 2.0


data2vec提出了一个通用的自监督学习框架统一了语音、视觉和语言三个模态数据的学习,而data2vec2.0主要解决的痛点就是构建自监督模型需要大量的GPU做算力支撑才能完成训练。


与最初的 data2vec 算法类似,data2vec 2.0预测数据的上下文化的表征(contextualized representations),或是神经网络的层次,而非预测图像的像素、文本段中的词或语音。



与常见的其他算法不同,这些所谓的目标表征是上下文化的,这意味着算法需要将整个训练示例考虑在内。


比如说,模型学习单词 bank 的表征是基于包含bank的整个句子,从而更容易推算出单词的正确含义,比如区分具体指代「金融机构」还是「河边的土地」。

研究人员认为上下文化的目标会促进更丰富的学习任务,并使 data2vec 2.0比其他算法学习得更快。


data2vec 2.0通过以下三种方式提高了原始 data2vec 算法的效率:


1、为特定训练样例构建目标表征,并将该表征重用在掩码版本上。


在掩码版本中,训练样例中的不同部分会被随机隐藏。随后两个版本学到的表征都会输入到学生模型中,为不同的掩码版本预测相同的上下文化的目标表征,从而有效地分摊了创建目标表征所需的计算量。


2、类似于掩码自编码器(masked autoencoder, MAE),学生模型中的编码器网络并不运训练样例中的空白部分(blanked out)。


在图像实验中,大约80%的部分都是空白,从而显著节省了计算周期。


3、使用了一个更有效的解码器模型,不再依赖于Transformer网络,而是依赖于一个多层卷积网络。


实验部分


为了更直观地理解 data2vec 2.0 比 data2vec 和其他同类算法的效率要高多少,研究人员在计算机视觉、语音和文本任务相关的基准测试中进行了广泛的实验。


实验中主要考虑最终的精确度以及预训练模型所需的时间,实验环境都是在相同的硬件上(GPU 的型号、数量等)来测量算法的运行速度。



在计算机视觉任务上,研究人员在标准 ImageNet-1K 图像分类基准上评估了 data2vec 2.0,模型通过该数据集可以学习图像表征。


实验结果显示,data2vec 2.0可以等同于掩码自编码器(MAE)的准确性,但是速度要快16倍。


如果继续给data2vec 2.0算法更多的运行时间,它可以达到更高的精度,并且仍然会比MAE的速度快。



在语音任务上,研究人员在 LibriLanguage 语音识别基准上进行了测试,它的准确性是 wave2vec 2.0的11倍以上。



对于自然语言处理任务,研究人员在通用语言理解评估(GLUE)基准上评估了 data2vec 2.0,仅需一半的训练时间即可达到与 BERT 的重新实现 RoBERTa 相同的精度。


参考资料:
https://ai.facebook.com/blog/ai-self-supervised-learning-data2vec/


微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。
相关阅读
重磅!一张RTX 2080Ti搞定大模型训练!算力节省136倍!ECCV 2022 | 半监督学习的隐秘角落: 无监督的数据选择性标注 USLFastTrack Universität 2023莱比锡大学公立语言项目招生简章换工记,在911呼叫中心工作日投降后毛泽东赴重庆国共谈判阿里「杀手锏」级语音识别模型来了!推理效率较传统模型提升10倍,已开源心肌炎比例高16倍!阳康后不可急于大负荷运动自监督为何有效?243页普林斯顿博士论文「理解自监督表征学习」,全面阐述对比学习、语言模型和自我预测三类方法如何提升深度学习算法效率,谷歌有这些绝招​WWW 2023 | 自监督多模态推荐系统OFA-Chinese:中文多模态统一预训练模型多模态 | 视觉-语言预训练入门指南基于自监督学习的多模态推荐算法用热传导方程来指导自监督学习NTU教授:多模态传感器融合算法|收获一作论文与导师推荐信!AAAI 2023 | 多模态对话的SPRING来了!无需标注就能做多模态问答预训练奇特的病人案例VideoMAE:简单高效的视频自监督预训练新范式|NeurIPS 2022NeurIPS 2022 | 基于结构聚类的异质图自监督学习MySQL 批量操作,一次插入多少行数据效率最高?语言模型自己学会用搜索引擎了?Meta AI提出API调用自监督学习方法Toolformer2022 Top10自监督学习模型发布!美中两国8项成果霸榜西电 NeurIPS 2022 | 基于结构聚类的异质图自监督学习北京:昨日发热门诊患者2.2万人次,是一周前的16倍!国家卫健委发布感染者居家指引:非必要不外出ECCV 2022 Oral | ​Language Matters:面向场景文字检测和端到端识别的弱监督的视觉-语言预训练方法美国婚前财产、婚后财产及个人财产的认定家长这样提问,孩子思维提升6倍!Asus ROG Strix XG32V 32in Curved Gaming Monitor 2560x1440 144hz100亿参数的语言模型跑不动?MIT华人博士提出SmoothQuant量化,内存需求直降一半,速度提升1.56倍!直播课预告:多模态预训练大模型OmModel及视觉应用开发德国指挥家的中国妻子北京内推 | 北京智源人工智能研究院招聘NLP/多模态算法研究员全面碾压AdamW!谷歌新出优化器内存小、效率高,网友:训练GPT 2果然快硬件预算最高直降46倍!低成本上手AIGC和千亿大模型,一行代码自动并行,Colossal-AI再升级Btrfs性能得到优化,最高提升10倍
logo
联系我们隐私协议©2024 redian.news
Redian新闻
Redian.news刊载任何文章,不代表同意其说法或描述,仅为提供更多信息,也不构成任何建议。文章信息的合法性及真实性由其作者负责,与Redian.news及其运营公司无关。欢迎投稿,如发现稿件侵权,或作者不愿在本网发表文章,请版权拥有者通知本网处理。