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

TORE:面向高效人体网格恢复的Transformer令牌削减方法

TORE: Token Reduction for Efficient Human Mesh Recovery with Transformer

City University of Hong Kong(香港城市大学);The University of Hong Kong(香港大学);Texas A&M University(德克萨斯农工大学);University of Oxford(牛津大学);Tencent Games(腾讯游戏);University of Cambridge(剑桥大学)

Zhiyang Dou, Qingxuan Wu, Cheng Lin, Zeyu Cao, Qiangqiang Wu, Weilin Wan, Taku Komura, Wenping Wang

AI总结:

本文提出TORE令牌削减策略,结合3D身体结构先验和2D图像特征聚类,在保持甚至提升人体网格恢复精度的同时大幅降低Transformer计算成本,并验证了其在手部网格恢复上的泛化性。

摘要:

本文提出了一组简单而有效的令牌削减(TORE)策略,用于基于Transformer的单目图像人体网格恢复。当前最先进的性能由基于Transformer的结构实现,但它们因冗余令牌而导致模型复杂度和计算成本过高。我们基于两个重要方面提出令牌削减策略,即3D几何结构和2D图像特征,其中我们利用身体结构先验分层恢复网格几何,并进行令牌聚类,以向Transformer传递更少但更具判别力的图像特征令牌。我们的方法大幅减少了Transformer中高复杂度交互所涉及的令牌数量。这显著降低了计算成本,同时在形状恢复中仍保持有竞争力甚至更高的精度。在广泛的基准上的大量实验验证了所提方法的卓越有效性。我们进一步展示了该方法在手部网格恢复上的泛化能力。请访问我们的项目页面:https://frank-zy-dou.github.io/projects/Tore/index.html。

英文摘要:

In this paper, we introduce a set of simple yet effective TOken REduction (TORE) strategies for Transformer-based Human Mesh Recovery from monocular images. Current SOTA performance is achieved by Transformer-based structures. However, they suffer from high model complexity and computation cost caused by redundant tokens. We propose token reduction strategies based on two important aspects, i.e., the 3D geometry structure and 2D image feature, where we hierarchically recover the mesh geometry with priors from body structure and conduct token clustering to pass fewer but more discriminative image feature tokens to the Transformer. Our method massively reduces the number of tokens involved in high-complexity interactions in the Transformer. This leads to a significantly reduced computational cost while still achieving competitive or even higher accuracy in shape recovery. Extensive experiments across a wide range of benchmarks validate the superior effectiveness of the proposed method. We further demonstrate the generalizability of our method on hand mesh recovery. Visit our project page at https://frank-zy-dou.github.io/projects/Tore/index.html.

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

↑