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

AUFormer:视觉Transformer是参数高效的面部动作单元检测器

AUFormer: Vision Transformers are Parameter-Efficient Facial Action Unit Detectors

Tianjin University(天津大学);Great Bay University(大湾区大学);Lappeenranta-Lahti University of Technology(拉彭兰塔-拉赫蒂理工大学);Shenzhen University(深圳大学)

Kaishen Yuan, Zitong Yu, Xin Liu, Weicheng Xie, Huanjing Yue, Jingyu Yang

AI总结:

针对面部AU检测现有方法过拟合或依赖额外数据的问题,提出参数高效的AUFormer模型,结合MoKE协作机制与MDWA-Loss,在多场景下实现最优性能与强泛化。

摘要:

面部动作单元(Facial Action Units, AU)是情感计算领域的核心概念,AU检测一直是研究热点。现有方法要么在标注稀缺的AU数据集上使用大量可学习参数导致过拟合,要么严重依赖大量额外相关数据。参数高效迁移学习(Parameter-Efficient Transfer Learning, PETL)为解决这些问题提供了可行范式,但现有PETL方法缺乏针对AU特性的设计。为此,我们创新性地将PETL范式引入AU检测,提出AUFormer,并设计了新型的知识混合专家(Mixture-of-Knowledge Expert, MoKE)协作机制。每个对应特定AU的MoKE仅含极少可学习参数,先整合个性化的多尺度知识与关联知识,再在专家组内与其他MoKE协作获取聚合信息,将其注入冻结的视觉Transformer(Vision Transformer, ViT)中,实现参数高效的AU检测。此外,我们设计了边截断难度感知加权非对称损失(Margin-truncated Difficulty-aware Weighted Asymmetric Loss, MDWA-Loss),可促使模型更关注激活的AU,区分未激活AU的难度,并丢弃潜在标注错误样本。从域内、跨域、数据效率、微表情域等多维度开展的大量实验表明,AUFormer无需依赖额外相关数据,即可取得当前最优性能与强泛化能力。AUFormer的代码可在https://github.com/yuankaishen2001/AUFormer获取。

英文摘要:

Facial Action Units (AU) is a vital concept in the realm of affective computing, and AU detection has always been a hot research topic. Existing methods suffer from overfitting issues due to the utilization of a large number of learnable parameters on scarce AU-annotated datasets or heavy reliance on substantial additional relevant data. Parameter-Efficient Transfer Learning (PETL) provides a promising paradigm to address these challenges, whereas its existing methods lack design for AU characteristics. Therefore, we innovatively investigate PETL paradigm to AU detection, introducing AUFormer and proposing a novel Mixture-of-Knowledge Expert (MoKE) collaboration mechanism. An individual MoKE specific to a certain AU with minimal learnable parameters first integrates personalized multi-scale and correlation knowledge. Then the MoKE collaborates with other MoKEs in the expert group to obtain aggregated information and inject it into the frozen Vision Transformer (ViT) to achieve parameter-efficient AU detection. Additionally, we design a Margin-truncated Difficulty-aware Weighted Asymmetric Loss (MDWA-Loss), which can encourage the model to focus more on activated AUs, differentiate the difficulty of unactivated AUs, and discard potential mislabeled samples. Extensive experiments from various perspectives, including within-domain, cross-domain, data efficiency, and micro-expression domain, demonstrate AUFormer's state-of-the-art performance and robust generalization abilities without relying on additional relevant data. The code for AUFormer is available at https://github.com/yuankaishen2001/AUFormer.

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

↑