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

一种用于半监督二维人体姿态估计的新型 Teacher-Reviewer-Student 框架

A New Teacher-Reviewer-Student Framework for Semi-supervised 2D Human Pose Estimation

Wulian Yun, Mengshi Qi, Fei Peng, Huadong Ma

AI总结:

针对半监督二维人体姿态估计忽视历史训练信息的问题,提出 Teacher-Reviewer-Student 框架,并结合多层次特征学习与 Keypoint-Mix 增强,显著提升性能。

摘要:

传统的二维人体姿态估计方法通常需要大量带标注数据,既耗费人力又成本高昂。相比之下,半监督二维人体姿态估计能够利用大量无标注数据和一小部分带标注数据来缓解上述问题。现有的半监督二维人体姿态估计方法通过反向传播更新网络,忽视了先前训练过程中至关重要的历史信息。因此,我们提出一种新颖的半监督二维人体姿态估计方法,该方法采用新设计的 Teacher-Reviewer-Student 框架。具体而言,我们首先模仿人类不断复习已有知识以加以巩固的现象来设计框架:教师网络预测结果以指导学生网络学习,审阅者网络则存储重要的历史参数,以提供额外的监督信号。其次,我们引入多层次特征学习策略,利用骨干网络不同阶段的输出来估计热图,以指导网络训练,在丰富监督信息的同时有效捕捉关键点之间的关系。最后,我们设计了一种数据增强策略,即 Keypoint-Mix,通过混合不同关键点来扰动姿态信息,从而增强网络辨别关键点的能力。在公开可用数据集上进行的大量实验表明,与现有方法相比,我们的方法取得了显著提升。

英文摘要:

Conventional 2D human pose estimation methods typically require extensive labeled annotations, which are both labor-intensive and expensive. In contrast, semi-supervised 2D human pose estimation can alleviate the above problems by leveraging a large amount of unlabeled data along with a small portion of labeled data. Existing semi-supervised 2D human pose estimation methods update the network through backpropagation, ignoring crucial historical information from the previous training process. Therefore, we propose a novel semi-supervised 2D human pose estimation method by utilizing a newly designed Teacher-Reviewer-Student framework. Specifically, we first mimic the phenomenon that human beings constantly review previous knowledge for consolidation to design our framework, in which the teacher predicts results to guide the student's learning and the reviewer stores important historical parameters to provide additional supervision signals. Secondly, we introduce a Multi-level Feature Learning strategy, which utilizes the outputs from different stages of the backbone to estimate the heatmap to guide network training, enriching the supervisory information while effectively capturing keypoint relationships. Finally, we design a data augmentation strategy, i.e., Keypoint-Mix, to perturb pose information by mixing different keypoints, thus enhancing the network's ability to discern keypoints. Extensive experiments on publicly available datasets, demonstrate our method achieves significant improvements compared to the existing methods.

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

↑