用教师-评审员-学生框架提升少标注下人体姿态估计效果
A New Teacher-Reviewer-Student Framework for Semi-supervised 2D Human Pose Estimation
- 设计教师-评审员-学生架构,利用历史参数增强学习
- 多层级特征学习使关键点关系更清晰,提升定位精度
- 引入关键点混合法增强数据多样性,适合小样本场景
传统2D人体姿态估计依赖大量标注数据,成本高昂。半监督方法通过少量标注数据与大量未标注数据结合可缓解此问题。现有方法仅通过反向传播更新网络,忽视训练过程中的历史信息。为此,我们提出一种新框架——教师-评审员-学生(Teacher-Reviewer-Student),模拟人类复习巩固知识的过程:教师指导学生学习,评审员存储重要历史参数以提供额外监督信号。其次,引入多层级特征学习策略,利用骨干网络不同阶段的输出生成热图,丰富监督信息并有效捕捉关键点间关系。最后,设计关键点混合法(Keypoint-Mix),通过混合不同关键点来扰动姿态,增强模型对关键点的辨别能力。在多个公开数据集上的实验表明,该方法显著优于现有方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。