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

批量 Transformer:在批量中寻找注意力

Batch Transformer: Look for Attention in Batch

Department of Computer Science and Engineering, Seoul National University of Science and Technology(计算机科学与工程系,首尔科学技术大学)

Myung Beom Her, Jisu Jeong, Hojoon Song, Ji-Hyeong Han

AI总结:

针对面部表情识别中单张噪声图像信息不可信的问题,本文提出批量 Transformer 网络,通过类别批量注意力和多级注意力从批量图像中提取可信特征,在多个基准上取得领先性能。

摘要:

面部表情识别(FER)在计算机视觉领域受到了广泛关注,尤其是在人机交互等“真实场景”环境中。然而,FER 图像包含遮挡、低分辨率、姿态变化、光照变化以及主观性等不确定性,其中包括一些与目标标签不匹配的表情。因此,从带有噪声的单张图像中获得的信息很少,并且不可信。这可能显著降低 FER 任务的性能。为了解决这一问题,我们提出了一种批量 Transformer(BT),其中包含所提出的类别批量注意力(CBA)模块,通过训练从批量中若干图像反映出的特征,而不是依赖单张图像的信息,来防止在噪声数据上过拟合并提取可信信息。我们还提出了多级注意力(MLA),通过捕捉各层级之间的相关性来防止对特定特征的过拟合。在本文中,我们提出了一个结合上述方案的批量 Transformer 网络(BTN)。在多个 FER 基准数据集上的实验结果表明,所提出的 BTN 在 FER 数据集上持续优于当前最先进的方法。代表性结果证明了所提出的 BTN 在 FER 中的潜力。

英文摘要:

Facial expression recognition (FER) has received considerable attention in computer vision, with "in-the-wild" environments such as human-computer interaction. However, FER images contain uncertainties such as occlusion, low resolution, pose variation, illumination variation, and subjectivity, which includes some expressions that do not match the target label. Consequently, little information is obtained from a noisy single image and it is not trusted. This could significantly degrade the performance of the FER task. To address this issue, we propose a batch transformer (BT), which consists of the proposed class batch attention (CBA) module, to prevent overfitting in noisy data and extract trustworthy information by training on features reflected from several images in a batch, rather than information from a single image. We also propose multi-level attention (MLA) to prevent overfitting the specific features by capturing correlations between each level. In this paper, we present a batch transformer network (BTN) that combines the above proposals. Experimental results on various FER benchmark datasets show that the proposed BTN consistently outperforms the state-ofthe-art in FER datasets. Representative results demonstrate the promise of the proposed BTN for FER.

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

↑