用扩散模型实现视频场景图的实时更新,兼顾精度与效率。
DIFFVSGG: Diffusion-Driven Online Video Scene Graph Generation
- 将视频场景图生成转为迭代去噪过程,统一物体分类、定位与关系推理。
- 在Action Genome上达到领先性能,支持连续时序推理,帧间依赖更合理。
- 适合需要实时处理视频流的应用,如智能监控与自动驾驶系统。
当前主流的视频场景图生成(VSGG)方法多采用离线流程,虽表现良好,但难以应对实时视频流,且占用大量显存。此外,这些方法在时序建模上能力有限,仅通过聚合帧级预测进行简单时间整合。为此,本文提出DIFFVSGG,一种面向在线视频场景图生成的新方法。受潜在扩散模型(LDMs)启发,该方法将物体分类、边界框回归和场景图生成任务统一于一个共享的潜在特征嵌入中。给定包含物体对联合特征的嵌入,通过在LDM中逐步去噪,获得清晰表达物体间关系的干净嵌入,再输入到特定任务头中完成最终预测。同时,该方法支持连续时序推理:后续帧的预测以历史帧结果作为条件输入,引导当前帧的反向扩散过程。在Action Genome数据集的三个设置下进行大量实验,验证了DIFFVSGG在准确率与实时性上的显著优势。
原文摘要 · Abstract (English)
Top-leading solutions for Video Scene Graph Generation (VSGG) typically adopt an offline pipeline. Though demonstrating promising performance, they remain unable to handle real-time video streams and consume large GPU memory. Moreover, these approaches fall short in temporal reasoning, merely aggregating frame-level predictions over a temporal context. In response, we introduce DIFFVSGG, an online VSGG solution that frames this task as an iterative scene graph update problem. Drawing inspiration from Latent Diffusion Models (LDMs) which generate images via denoising a latent feature embedding, we unify the decoding of object classification, bounding box regression, and graph generation three tasks using one shared feature embedding. Then, given an embedding containing unified features of object pairs, we conduct a step-wise Denoising on it within LDMs, so as to deliver a clean embedding which clearly indicates the relationships between objects. This embedding then serves as the input to task-specific heads for object classification, scene graph generation, etc. DIFFVSGG further facilitates continuous temporal reasoning, where predictions for subsequent frames leverage results of past frames as the conditional inputs of LDMs, to guide the reverse diffusion process for current frames. Extensive experiments on three setups of Action Genome demonstrate the superiority of DIFFVSGG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。