arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.00635cs.CVcs.AI

S$^3$POT: 通过自监督提示学习实现对比驱动的人脸遮挡分割

S$^3$POT: Contrast-Driven Face Occlusion Segmentation via Self-Supervised Prompt Learning

ShanghaiTech University(上海科技大学);United Imaging Intelligence(联影智能)

Lingsong Wang, Mancheng Meng, Ziyan Wu, Terrence Chen, Fan Yang, Dinggang Shen

AI总结:

S$^3$POT通过自监督提示学习实现对比驱动的人脸遮挡分割,结合人脸生成与空间提示,提升遮挡区域分割精度。

摘要:

现有的人脸解析方法通常将遮挡误分类为面部组件。这是因为遮挡是一个高层概念,它不指代一个具体的物体类别。因此,构建一个涵盖所有遮挡物体类别的现实人脸数据集几乎不可能,且准确的掩码标注是劳动密集型的。为了解决这些问题,我们提出了S$^3$POT,一个结合人脸生成与自监督空间提示的对比驱动框架,以实现遮挡分割。该框架受到两个见解的启发:1)现代人脸生成器能够真实地重建遮挡区域,生成一幅保持面部几何结构但消除遮挡的图像;2)基础分割模型(如SAM)在提供适当提示时能够提取精确的掩码。特别是,S$^3$POT由三个模块组成:参考生成(RF)、特征增强(FE)和提示选择(PS)。首先,通过解析掩码的结构引导生成参考图像。其次,FE在原始图像和参考图像之间进行令牌对比,以获得初始提示,然后通过交叉注意力修改图像特征。第三,基于增强的特征,PS构建一组正负提示并将其通过自注意力网络筛选,以供掩码解码器使用。该网络在没有遮挡地面真实掩码的指导下,通过三种新颖且互补的目标函数进行学习。在专门收集的数据集上的广泛实验表明,S$^3$POT的优越性能以及每个模块的有效性。

英文摘要:

Existing face parsing methods usually misclassify occlusions as facial components. This is because occlusion is a high-level concept, it does not refer to a concrete category of object. Thus, constructing a real-world face dataset covering all categories of occlusion object is almost impossible and accurate mask annotation is labor-intensive. To deal with the problems, we present S$^3$POT, a contrast-driven framework synergizing face generation with self-supervised spatial prompting, to achieve occlusion segmentation. The framework is inspired by the insights: 1) Modern face generators' ability to realistically reconstruct occluded regions, creating an image that preserve facial geometry while eliminating occlusion, and 2) Foundation segmentation models' (e.g., SAM) capacity to extract precise mask when provided with appropriate prompts. In particular, S$^3$POT consists of three modules: Reference Generation (RF), Feature enhancement (FE), and Prompt Selection (PS). First, a reference image is produced by RF using structural guidance from parsed mask. Second, FE performs contrast of tokens between raw and reference images to obtain an initial prompt, then modifies image features with the prompt by cross-attention. Third, based on the enhanced features, PS constructs a set of positive and negative prompts and screens them with a self-attention network for a mask decoder. The network is learned under the guidance of three novel and complementary objective functions without occlusion ground truth mask involved. Extensive experiments on a dedicatedly collected dataset demonstrate S$^3$POT's superior performance and the effectiveness of each module.

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

↑