用时序关系知识提升弱监督动态场景图生成的检测精度
TRKT: Weakly Supervised Dynamic Scene Graph Generation with Temporal-enhanced Relation-aware Knowledge Transferring
- 通过关系感知注意力与光流增强,挖掘动态场景中的交互区域
- 在Action Genome上达到当前最优,显著提升物体定位与置信度
- 适合关注视频理解中关系建模与弱监督学习的研究者
动态场景图生成(DSGG)旨在通过检测物体并预测其关系,为每帧视频构建场景图。弱监督DSGG(WS-DSGG)通过每视频仅用一帧的未定位场景图进行训练,以减少标注负担。现有方法依赖预训练外部检测器生成伪标签,但这些检测器基于静态、以物体为中心的图像训练,在动态、关系感知场景中表现不佳,导致定位不准、置信度低。为此,本文提出时序增强的关系感知知识迁移(TRKT)方法,包含两个核心组件:(1) 关系感知知识挖掘:使用物体和关系类别解码器生成类别特定注意力图,突出物体区域与交互区域;提出跨帧注意力增强策略,利用相邻帧的光流信息提升注意力图,使其具备运动感知能力,增强对运动模糊的鲁棒性,实现关系与运动感知的知识挖掘。(2) 双流融合模块:将类别特定注意力图融入外部检测结果,优化物体定位并提升提案置信度。大量实验表明,TRKT在Action Genome数据集上达到当前最优性能。代码已开源:https://github.com/XZPKU/TRKT.git。
原文摘要 · Abstract (English)
Dynamic Scene Graph Generation (DSGG) aims to create a scene graph for each video frame by detecting objects and predicting their relationships. Weakly Supervised DSGG (WS-DSGG) reduces annotation workload by using an unlocalized scene graph from a single frame per video for training. Existing WS-DSGG methods depend on an off-the-shelf external object detector to generate pseudo labels for subsequent DSGG training. However, detectors trained on static, object-centric images struggle in dynamic, relation-aware scenarios required for DSGG, leading to inaccurate localization and low-confidence proposals. To address the challenges posed by external object detectors in WS-DSGG, we propose a Temporal-enhanced Relation-aware Knowledge Transferring (TRKT) method, which leverages knowledge to enhance detection in relation-aware dynamic scenarios. TRKT is built on two key components:(1)Relation-aware knowledge mining: we first employ object and relation class decoders that generate category-specific attention maps to highlight both object regions and interactive areas. Then we propose an Inter-frame Attention Augmentation strategy that exploits optical flow for neighboring frames to enhance the attention maps, making them motion-aware and robust to motion blur. This step yields relation- and motion-aware knowledge mining for WS-DSGG. (2) we introduce a Dual-stream Fusion Module that integrates category-specific attention maps into external detections to refine object localization and boost confidence scores for object proposals. Extensive experiments demonstrate that TRKT achieves state-of-the-art performance on Action Genome dataset. Our code is avaliable at https://github.com/XZPKU/TRKT.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。