随着短视频兴起,每天有海量的短视频上传到各大视频平台,面对海量的短视频,如何提升这些短视频的智能分发效率是各大短视频平台面临的重要课题。视频的标签技术是内容理解的一种重要手段,可以对不同兴趣的用户分发属于他们感兴趣的视频,从而提升平台的用户时长和ctr。
视频标签技术难点
由于视频内容是包含文本、视觉、语音等信息的多模态表现形式,如何提取多模态特征的表征能力并进行模态之间的融合互补,综合应用在模型中并取得较好的结果是至关重要的,该领域存在以下几个技术难点。
标签量级大
由于标签的粗细力度对内容推荐和下游分发的精准性有很大的影响,所以标签的量级越大、覆盖范围越广,越能精细表示视频内容和用户习惯。网易新闻站内标签是以几十万量级计数,并且每个视频有多个标签所属,同时标签长尾分布现象严重,常规的多标签分类方式解决已不再适用。
标签覆盖率低
比如一个视频的标题为“这才是百姓小馆,特别接地气”,视频内容是在餐馆用餐,标注同学标注结果是探店美食,但是同样日常美食、美食测评也同样满足这个视频内容。但是这些可用标签没有被数据标注的情况给模型学习带来了挑战。
多模态信息理解差异化
视频是一种多模态的内容形式,包括文本、图像、音频等多种模态的信息。对于不同的视频,不同模态的重要程度也不同,因此需要对多模态的信息进行综合全面的向量表征。
技术解决方案
针对以上对视频内容标签流程的难点和挑战,我们设计了基于召回-排序的两段式算法流程,如图1所示,召回是视频和标签粗粒度匹配的过程,排序是视频和标签细粒度匹配的过程,用来做精细化标签适用性的排序。此算法可以很好地解决视频标注过程中的难点。
图1 视频标签解决方案
召回阶段
召回阶段核心目标在于召回尽可能多、尽可能全的候选标签。我们采用video-video-tag的方式召回,通过query视频向量召回相关视频,从而相关视频的原始标签可以被query视频获取。为保证召回标签尽可能全面,我们按照以下五个维度进行召回:
文本语义相似
对视频的文本采用sentence-bert[1]模型获取embedding进行相似召回。
视频语义相似
对视频进行抽帧后经过clip[2]提取图像特征后进行mean pooling得到embedding进行相似召回。
音频语义相似
对视频中背景音乐截取经过vggish[3]提取音频特征后进行mean pooling得到embedding进行相似召回。
多标签分类结果:
建模视频多模态多标签分类模型,将输出结果作为召回标签。具体结构如图2所示。因为该模型的目的是训练通用视频表征能力的embedding,所以模型的分类类别较多。与传统分类模型相比,我们添加了标签embedding和视频融合embedding的相似学习分支,增强视频向量在语义空间的鲁棒性,也取得了预期的效果。
图2 多模态多标签分类模型
多模态语义相似
对视频,文本,音频三路特征进行多模态融合后得到embedding进行相似召回。
基于以上五种召回方法,对比召回结果和原始标签标注结果计算召回率,如表1所示。其中,应用于多模态相似召回和多标签分类召回的网络模型为我们原创模型。
表1 多种召回方式召回率
文本相似召回 视频相似召回 音频相似召回 多标签分类召回 多模态相似召回 整体召回 0.255 0.532 0.332 0.667 0.733 0.874 排序阶段
排序阶段核心目标在于精准排序所有召回的标签,排序模型的结构如图3所示。
图3 对比学习排序模型模型
排序模型中,我们采用了nextvlad[4]的进行视频帧特征和音频特征的聚合降维,使得视频,音频和文本特征可以进行模态对齐。文本特征提取我们采取了轻量化的tiny-albert[5]模型,便于线上快速推理。
排序模型是基于向量对比学习,目的是拉近视频的相关标签和推远无关标签。我们实验发现在对比学习中,负样本标签的构建尤为重要。我们尝试了纯简单样本,加大负样本的negative_num,困难样本等多种思路,得到以下结论:
1、当只用简单样本时,并且负样本采样个数为10个时,排序模型可以将无关召回标签排在后面,高相关性标签排前,但相近的标签区分度不大。并且当召回源较多时候,排序准确性会降低。
2、加大负样本的negative_num到400时,极大解决了召回源多时,排序不准确的问题。但还会存在一定程度相近标签难以区分。
3、尝试进行"2跳"方法构建困难负样本进行训练,即与video1有相同标签的videok的其他标签,当做video1的困难负样本。很大程度相近标签解决精准排序问题。
以上三种尝试的排序样例如表2所示(未进行相似标签去重和个数限制)。以第二个视频为例,我们可以明显观察到视频的的ground_true 是篮球比赛,在负样本较少的情况下(negative_num=10)排序top5会出现足球运动,拳击运动等体育相关的运动,但跟篮球无关,细粒度排序效果差。当增加负样本 (negative_num=400)后,排序top5的结果有所改善,拳击运动排在靠后位置,当(negative_num=400)并且增加困难样本后,排序top5出现前四个全部描述篮球运动的,后一个体育资讯也是与篮球运动有相关性的。可以得出训练时增加负样本比例,增加困难样本,可以得到更好的排序结果。
表2 排序结果对比
标题
房子自动加热到64度,墙纸都被烧焦,地板还能炒鸡蛋" 当年梦一队爆发严重内讧,天赋一个比一个恐怖,谁都不服乔丹! 人工标签 住房问题||奇闻讲述||安全隐患 篮球比赛 负样本negative_num=10预测结果 室内装修||房屋改造||装修分享||家具||室内拍摄||新房||卧室||租房生活||房产价格||办公室||房子|||科学实验||房产实拍||房屋介绍||装修风格||毛坯房||出租房||房屋户型||甜品 足球运动||篮球比赛||篮球运动||体育资讯||拳击运动||篮球扣篮||赛事回顾||nba赛事||棋牌博弈||棋牌对局||新闻报道 负样本negative_num=400预测结果 奇闻讲述||社会奇闻||室内装修||房屋改造||房产知识||室内设计||房屋户型||创意装修||装修展示||厨房||房产实拍||国外奇闻||出租房||房屋介绍||房子||装修风格||房产展示||趣味实验||毛坯房||科学实验 篮球比赛||篮球运动||体育资讯||足球运动||篮球技巧||街头篮球||篮球扣篮||拳击运动||运动||图书馆||nba赛事||赛事回顾 负样本negative_num=400+困难样本预测结果 奇闻讲述||社会奇闻||安全隐患||住房问题||室内装修||房屋改造||房产知识||室内设计||房屋户型||创意装修||装修展示||厨房||房产实拍||国外奇闻||出租房||房屋介绍||房子||装修风格||房产展示||趣味实验 篮球比赛||篮球运动|| nba赛事||体育资讯||篮球技巧||街头篮球||篮球扣篮||赛事||足球运动||拳击运动||运动
效果评估
综合以上方案设计,我们在网易新闻客户端落地机器标签预测模型。后续通过扩展标签的embedding,无需模型训练可以直接扩充标签覆盖数目。同时,我们对比了多模态分类结果,多路召回+排序结果,和全排序的结果(即无召回阶段,在全部的标签上直接做排序)。如表3所示。多种方案的人工评测数据效果可供参考。
表3 多种标签方案指标对比
模型 标签准确率 平均标签个数 多模态分类模型 0.730 3.59 全排序模型 0.859 4.83 多路召回+排序模型 0.884 3.12
总结展望
本文提出了一种多模态视频多标签的建模方法,旨在通过引入召回排序的方式,去提升标签预测的准确性。同时落地在视频新闻场景,在下游推荐算法中取得了不错的业务效果。未来希望融入图谱知识到模型中,提升模型的泛化能力。
参考文献
[1] Reimers N, Gurevych I. Sentence-bert: Sentence embeddings using siamese bert-networks[J]. arXiv preprint arXiv:1908.10084, 2019.
[2] Radford A, Kim J W, Hallacy C, et al. Learning transferable visual models from natural language supervision[C]//International Conference on Machine Learning. PMLR, 2021: 8748-8763.
[3] Hershey S, Chaudhuri S, Ellis D P W, et al. CNN architectures for large-scale audio classification[C]//2017 ieee international conference on acoustics, speech and signal processing (icassp). IEEE, 2017: 131-135.
[4] Lin Rongcheng, Xiao Jing, and Jianping Fan. Nextvlad: An efficient neural network to aggregate frame-level features for large-scale video classification. In: Proc. of the 2nd Workshop on YouTube-8M Large-Scale Video Understanding (2018), 2018. 1, 2, 3, 5
[5] Lan Z, Chen M, Goodman S, et al. Albert: A lite bert for self-supervised learning of language representations[J]. arXiv preprint arXiv:1909.11942, 2019.
作者简介:
李文君、石佳影均来自网易新闻推荐算法团队
实习招聘信息
网易新闻推荐算法团队常年招聘算法实习生,欢迎有兴趣的同学自荐,联系方式:liwenjun01@corp.netease.com
岗位描述
1、通过对内容和用户的精准理解,提升网易新闻的阅读体验,给用户提供有态度、有用有趣的新闻资讯 ;
2、有计算机视觉、自然语言处理、推荐、搜索等算法方向背景的同学,都有对应的职位可供选择;
岗位要求
我们希望你:
1、计算机相关专业,硕士及以上学历,能实习三个月以上为佳;
2、熟悉常用的机器学习技术和深度学习算法,包括但不限于以下几个方向的经验:
推荐:召回、粗排、精排、重排等;
搜索:query理解、数据挖掘等;
自然语言处理:知识图谱、语义理解等;
计算机视觉:视频理解、多模态学习、分类检测等;
3、熟悉c/c++/python等编程语言,熟悉spark/hadoop/redis等工具,能够基于tensorflow/pytorch等框架实现算法模型;
4、聪明、踏实,善于沟通和协作;
热门跟贴