通过噪声修正提升弱监督动作定位的伪标签质量
Rethinking Pseudo-Label Guided Learning for Weakly Supervised Temporal Action Localization from the Perspective of Noise Correction
- 分两阶段处理伪标签噪声:先用上下文感知去噪优化边界,再用师生框架修正遗漏与误连
- 在THUMOS14和ActivityNet v1.2上检测准确率和推理速度显著优于现有方法
- 适合关注弱监督视频理解、伪标签优化的研究者与工程师
伪标签学习广泛应用于弱监督时间动作定位。现有方法直接使用弱监督基础模型生成实例级伪标签来训练全监督检测头。我们指出,伪标签中的噪声会干扰检测头的学习,导致性能下降。噪声问题包括:(1) 边界定位不准;(2) 短动作片段漏检;(3) 多个相邻片段错误合并为一个。为此,我们提出两阶段噪声标签学习策略,充分挖掘噪声标签中的有用信号。首先,设计帧级伪标签生成模型,结合上下文感知去噪算法优化边界。其次,引入在线修订的师生框架,包含缺失实例补偿模块和模糊实例修正模块,解决短动作遗漏与多对一问题。此外,在师生框架中引入高质量伪标签挖掘损失,对不同噪声标签赋予差异化权重以更高效训练。模型在THUMOS14和ActivityNet v1.2基准上显著提升检测准确率与推理速度。
原文摘要 · Abstract (English)
Pseudo-label learning methods have been widely applied in weakly-supervised temporal action localization. Existing works directly utilize weakly-supervised base model to generate instance-level pseudo-labels for training the fully-supervised detection head. We argue that the noise in pseudo-labels would interfere with the learning of fully-supervised detection head, leading to significant performance leakage. Issues with noisy labels include:(1) inaccurate boundary localization; (2) undetected short action clips; (3) multiple adjacent segments incorrectly detected as one segment. To target these issues, we introduce a two-stage noisy label learning strategy to harness every potential useful signal in noisy labels. First, we propose a frame-level pseudo-label generation model with a context-aware denoising algorithm to refine the boundaries. Second, we introduce an online-revised teacher-student framework with a missing instance compensation module and an ambiguous instance correction module to solve the short-action-missing and many-to-one problems. Besides, we apply a high-quality pseudo-label mining loss in our online-revised teacher-student framework to add different weights to the noisy labels to train more effectively. Our model outperforms the previous state-of-the-art method in detection accuracy and inference speed greatly upon the THUMOS14 and ActivityNet v1.2 benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。