arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.07978cs.CV

通过弥合伪标签中的训练-测试差距改进弱监督时序动作定位

Improving Weakly Supervised Temporal Action Localization by Bridging Train-Test Gap in Pseudo Labels

The Chinese University of Hong Kong(香港中文大学);Centre for Perceptual and Interactive Intelligence(香港感知与交互智能研究中心);Beihang University(北京航空航天大学)

Jingqiu Zhou, Linjiang Huang, Liang Wang, Si Liu, Hongsheng Li

AI总结:

本文针对弱监督时序动作定位中训练与测试不一致的问题,提出高斯加权融合、约束优化伪标签生成和Δ伪标签自我校正机制,在THUMOS14和ActivityNet1.3上显著提升了定位性能。

摘要:

弱监督时序动作定位任务旨在为感兴趣的动作生成时间边界,同时还需要对动作类别进行分类。基于伪标签的方法作为一种有效的解决方案,近年来得到了广泛研究。然而,现有方法在训练阶段生成伪标签,在测试阶段则采用不同的流程或设置进行预测,导致训练与测试之间存在差距。本文提出从预测的动作边界中生成高质量的伪标签。然而,我们注意到现有的后处理方法(如NMS)会导致信息损失,不足以生成高质量的动作边界。更重要的是,将动作边界转换为伪标签相当具有挑战性,因为预测出的动作实例通常相互重叠且具有不同的置信度分数。此外,在训练早期阶段,生成的伪标签可能波动且不准确。如果没有自我校正机制,它可能会反复强化错误预测。为了解决这些问题,我们提出了一种学习更好伪标签的有效流程。首先,我们提出了一个高斯加权融合模块,以保留动作实例的信息并获得高质量的动作边界。其次,我们将伪标签生成形式化为一个在动作实例置信度分数约束下的优化问题。最后,我们引入了Δ伪标签的思想,使模型具备自我校正能力。我们的方法在两个基准数据集THUMOS14和ActivityNet1.3上取得了优于现有方法的性能,在平均mAP指标上分别在THUMOS14上提升了1.9%,在ActivityNet1.3上提升了3.7%。

英文摘要:

The task of weakly supervised temporal action localization targets at generating temporal boundaries for actions of interest, meanwhile the action category should also be classified. Pseudo-label-based methods, which serve as an effective solution, have been widely studied recently. However, existing methods generate pseudo labels during training and make predictions during testing under different pipelines or settings, resulting in a gap between training and testing. In this paper, we propose to generate high-quality pseudo labels from the predicted action boundaries. Nevertheless, we note that existing post-processing, like NMS, would lead to information loss, which is insufficient to generate high-quality action boundaries. More importantly, transforming action boundaries into pseudo labels is quite challenging, since the predicted action instances are generally overlapped and have different confidence scores. Besides, the generated pseudo-labels can be fluctuating and inaccurate at the early stage of training. It might repeatedly strengthen the false predictions if there is no mechanism to conduct self-correction. To tackle these issues, we come up with an effective pipeline for learning better pseudo labels. Firstly, we propose a Gaussian weighted fusion module to preserve information of action instances and obtain high-quality action boundaries. Second, we formulate the pseudo-label generation as an optimization problem under the constraints in terms of the confidence scores of action instances. Finally, we introduce the idea of $Δ$ pseudo labels, which enables the model with the ability of self-correction. Our method achieves superior performance to existing methods on two benchmarks, THUMOS14 and ActivityNet1.3, achieving gains of 1.9\% on THUMOS14 and 3.7\% on ActivityNet1.3 in terms of average mAP.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑