arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.14700cs.CVcs.AI

基于深度引导的跨模态蒸馏自监督人体关键点检测

Depth-Guided Self-Supervised Human Keypoint Detection via Cross-Modal Distillation

Queen’s University Ontario Canada(皇后大学)

Aman Anand, Elyas Rashno, Amir Eskandari, Farhana Zulkernine

AI总结:

提出跨模态知识蒸馏框架Distill-DKP,利用深度图指导RGB图像进行自监督人体关键点检测,有效减少背景误检并在多个数据集上显著降低误差。

摘要:

现有的无监督关键点检测方法对图像应用人工变形(如遮挡图像的大部分区域),并使用原始图像重建作为学习目标来检测关键点。然而,这种方法缺乏图像中的深度信息,且经常在背景上检测到关键点。为了解决这一问题,我们提出了Distill-DKP,这是一种新颖的跨模态知识蒸馏框架,它利用深度图和RGB图像在自监督环境下进行关键点检测。在训练期间,Distill-DKP从基于深度的教师模型中提取嵌入级知识,以指导基于图像的学生模型,且推理仅限于学生模型。实验表明,Distill-DKP显著优于以往的无监督方法:在Human3.6M上将平均L2误差降低了47.15%,在Taichi上将平均误差降低了5.67%,并在DeepFashion数据集上将关键点准确率提高了1.3%。详细的消融实验证明了知识蒸馏对网络不同层的敏感性。项目页面:https://23wm13.github.io/distill-dkp/

英文摘要:

Existing unsupervised keypoint detection methods apply artificial deformations to images such as masking a significant portion of images and using reconstruction of original image as a learning objective to detect keypoints. However, this approach lacks depth information in the image and often detects keypoints on the background. To address this, we propose Distill-DKP, a novel cross-modal knowledge distillation framework that leverages depth maps and RGB images for keypoint detection in a self-supervised setting. During training, Distill-DKP extracts embedding-level knowledge from a depth-based teacher model to guide an image-based student model with inference restricted to the student. Experiments show that Distill-DKP significantly outperforms previous unsupervised methods by reducing mean L2 error by 47.15% on Human3.6M, mean average error by 5.67% on Taichi, and improving keypoints accuracy by 1.3% on DeepFashion dataset. Detailed ablation studies demonstrate the sensitivity of knowledge distillation across different layers of the network. Project Page: https://23wm13.github.io/distill-dkp/

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

↑