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

SCE-MAE:基于掩码自编码器的选择性对应增强用于自监督地标估计

SCE-MAE: Selective Correspondence Enhancement with Masked Autoencoder for Self-Supervised Landmark Estimation

University of Toronto(多伦多大学);ModiFace

Kejia Yin, Varshanth R. Rao, Ruowei Jiang, Xudong Liu, Parham Aarabi, David B. Lindell

AI总结:

SCE-MAE利用掩码自编码器提取区域级特征,通过对应近似与细化模块和局部约束排斥损失优化选择性对应,在地标匹配和检测任务上大幅超越现有方法。

摘要:

自监督地标估计是一项具有挑战性的任务,需要在缺乏标注数据的情况下,形成局部独特的特征表示来识别稀疏的面部地标。为了解决这一任务,现有的最先进(SOTA)方法(1)从使用实例级自监督学习(SSL)范式训练的主干网络中提取粗粒度特征,这忽略了任务的密集预测特性,(2)将这些特征聚合为内存密集型的超列结构,以及(3)监督轻量级投影网络,以朴素地建立所有空间特征对之间的完整局部对应关系。在本文中,我们提出了SCE-MAE框架,该框架(1)利用MAE(一种区域级SSL方法,天然更适合地标预测任务),(2)在原始特征图上操作,而非使用昂贵的超列,(3)采用对应近似与细化模块(CARB),该模块利用简单的密度峰值聚类算法和我们提出的局部约束排斥损失,直接优化仅选择性的局部对应关系。通过大量实验,我们证明SCE-MAE高度有效且鲁棒,在地标匹配任务上以约20%-44%的大幅优势超越现有SOTA方法,在地标检测任务上以约9%-15%的优势超越现有SOTA方法。

英文摘要:

Self-supervised landmark estimation is a challenging task that demands the formation of locally distinct feature representations to identify sparse facial landmarks in the absence of annotated data. To tackle this task, existing state-of-the-art (SOTA) methods (1) extract coarse features from backbones that are trained with instance-level self-supervised learning (SSL) paradigms, which neglect the dense prediction nature of the task, (2) aggregate them into memory-intensive hypercolumn formations, and (3) supervise lightweight projector networks to naively establish full local correspondences among all pairs of spatial features. In this paper, we introduce SCE-MAE, a framework that (1) leverages the MAE, a region-level SSL method that naturally better suits the landmark prediction task, (2) operates on the vanilla feature map instead of on expensive hypercolumns, and (3) employs a Correspondence Approximation and Refinement Block (CARB) that utilizes a simple density peak clustering algorithm and our proposed Locality-Constrained Repellence Loss to directly hone only select local correspondences. We demonstrate through extensive experiments that SCE-MAE is highly effective and robust, outperforming existing SOTA methods by large margins of approximately 20%-44% on the landmark matching and approximately 9%-15% on the landmark detection tasks.

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

↑