用自动标注构建百万级视频字幕数据集,提升字幕与物体定位的精准度。
Large-scale Pre-training for Grounded Video Caption Generation
- 通过帧级字幕与框体融合,生成时序稠密的自动标注数据。
- 在5个数据集上达到最优效果,关键依赖大规模预训练+小规模精调。
- 适合做视频理解、多模态生成的研究者参考。
本文提出一种新型视频字幕生成与物体定位方法,通过时序稠密的边界框实现字幕中物体在视频中的精准定位。首先,提出一种大规模自动标注方法,将帧级带框字幕整合为时序连续且一致的标注数据,应用于HowTo100M数据集,构建了名为HowToGround1M的大规模预训练数据集;同时提出名为GROVE的接地视频字幕生成模型,并在HowToGround1M上进行预训练。其次,构建iGround数据集,包含3513个视频,配有手动标注的字幕和稠密时空定位框,用于评估进展及模型微调。第三,实验表明,该方法在iGround、VidSTG、ActivityNet-Entities、GroundingYouTube和YouCook-Interactions等数据集上均达到当前最佳性能。消融实验证明,在HowToGround1M上预训练后,再于iGround上微调的重要性,并验证了模型关键技术的有效性。数据集与代码已公开。
原文摘要 · Abstract (English)
We propose a novel approach for captioning and object grounding in video, where the objects in the caption are grounded in the video via temporally dense bounding boxes. We introduce the following contributions. First, we present a large-scale automatic annotation method that aggregates frame-level captions grounded with bounding boxes into temporally dense and consistent annotations. We apply this approach on the HowTo100M dataset to construct a large-scale pre-training dataset, named HowToGround1M. We also introduce a Grounded Video Caption Generation model, dubbed GROVE, and pre-train the model on HowToGround1M. Second, we introduce iGround--a dataset of 3513 videos with manually annotated captions and dense spatio-temporally grounded bounding boxes. This allows us to measure progress on this challenging problem, as well as to fine-tune our model on this small-scale but high-quality data. Third, we demonstrate that our approach achieves state-of-the-art results on the proposed iGround dataset, as well as on the VidSTG, ActivityNet-Entities, GroundingYouTube, and YouCook-Interactions datasets. Our ablations demonstrate the importance of pre-training on our automatically annotated HowToGround1M dataset followed by fine-tuning on the manually annotated iGround dataset and validate the key technical contributions of our model. The dataset and code are available at https://ekazakos.github.io/grounded_video_caption_generation/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。