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

GTPT:基于分组令牌剪枝的高效人体姿态估计Transformer

GTPT: Group-based Token Pruning Transformer for Efficient Human Pose Estimation

State Key Laboratory for Novel Software Technology, Nanjing University(南京大学计算机软件新技术国家重点实验室);Cainiao Network(菜鸟网络)

Haonan Wang, Jie Liu, Jie Tang, Gangshan Wu, Bo Xu, Yanbing Chou, Yong Wang

AI总结:

提出GTPT,一种基于分组令牌剪枝的Transformer,通过从粗到细引入关键点、分组关键点令牌并剪枝视觉令牌,以及多头组注意力实现高效全身姿态估计,在COCO和COCO-WholeBody上以更低计算量取得更高性能。

摘要:

近年来,二维人体姿态估计在公开基准上取得了显著进展。然而,许多方法由于参数量大和计算开销高,在工业界的应用中面临适用性不足的挑战。高效人体姿态估计仍然是一个难题,尤其是对于包含众多关键点的全身姿态估计。虽然当前大多数高效人体姿态估计方法主要依赖CNN,我们提出了基于分组的令牌剪枝Transformer(GTPT),充分利用了Transformer的优势。GTPT通过从粗到细的方式逐步引入关键点来减轻计算负担,在确保高性能的同时最小化计算开销。此外,GTPT对关键点令牌进行分组并剪枝视觉令牌,以提高模型性能并减少冗余。我们提出了不同组之间的多头组注意力(MHGA),以较低的计算开销实现全局交互。我们在COCO和COCO-WholeBody上进行了实验。与其他方法相比,实验结果表明,GTPT可以在更少的计算量下实现更高的性能,尤其是在包含众多关键点的全身姿态估计任务中。

英文摘要:

In recent years, 2D human pose estimation has made significant progress on public benchmarks. However, many of these approaches face challenges of less applicability in the industrial community due to the large number of parametric quantities and computational overhead. Efficient human pose estimation remains a hurdle, especially for whole-body pose estimation with numerous keypoints. While most current methods for efficient human pose estimation primarily rely on CNNs, we propose the Group-based Token Pruning Transformer (GTPT) that fully harnesses the advantages of the Transformer. GTPT alleviates the computational burden by gradually introducing keypoints in a coarse-to-fine manner. It minimizes the computation overhead while ensuring high performance. Besides, GTPT groups keypoint tokens and prunes visual tokens to improve model performance while reducing redundancy. We propose the Multi-Head Group Attention (MHGA) between different groups to achieve global interaction with little computational overhead. We conducted experiments on COCO and COCO-WholeBody. Compared to other methods, the experimental results show that GTPT can achieve higher performance with less computation, especially in whole-body with numerous keypoints.

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

↑