3 个月前

基于并行解码的端到端密集视频字幕生成

基于并行解码的端到端密集视频字幕生成

摘要

密集视频字幕生成旨在从视频中自动生成多个与时间位置相对应的字幕。以往的方法通常采用复杂的“定位-描述”两阶段范式,严重依赖大量手工设计的组件。本文提出了一种简单而有效的端到端密集视频字幕框架——并行解码密集视频字幕(Parallel Decoding for Dense Video Captioning, PDVC),通过将密集字幕生成建模为集合预测任务,实现了高效统一的建模。在实践中,通过在Transformer解码器顶部堆叠一种新提出的事件计数器(event counter),PDVC能够在整体理解视频内容的基础上,精确地将视频分割为若干事件片段,显著提升了生成字幕的连贯性与可读性。相较于现有方法,PDVC具有以下显著优势:(1)无需依赖启发式非极大值抑制(non-maximum suppression)或循环事件序列选择网络来消除冗余,PDVC可直接输出规模适中的事件集合;(2)与传统的两阶段范式不同,本文将事件查询的增强表示并行输入到定位头与字幕生成头,使两个子任务在优化过程中深度耦合、相互促进;(3)在不依赖复杂组件的前提下,基于ActivityNet Captions和YouCook2数据集的大量实验表明,PDVC能够生成高质量的字幕结果,在定位准确率与现有最先进两阶段方法相当的情况下,仍能实现超越。代码已开源,地址为:https://github.com/ttengwang/PDVC。

代码仓库

ttengwang/pdvc
官方
pytorch
GitHub 中提及
aim3-ruc/youmakeup_challenge2022
pytorch
GitHub 中提及

基准测试

基准方法指标
dense-video-captioning-on-activitynetPDVC (TSP features, no SCST)
BLEU-4: 2.17
CIDEr: 31.14
METEOR: 9.03
SODA: 6.05
dense-video-captioning-on-youcook2PDVC (TSN features, no SCST)
BLEU4: 0.8
CIDEr: 22.71
METEOR: 4.74
SODA: 4.42

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于并行解码的端到端密集视频字幕生成 | 论文 | HyperAI超神经