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

EM-Net:基于期望最大化算法的视线估计

EM-Net: Gaze Estimation with Expectation Maximization Algorithm

School of Computer and Information Science, Chongqing Normal University(重庆师范大学计算机与信息科学学院)

Zhang Cheng, Yanxia Wang, Guoyu Xia

AI总结:

针对现有视线估计方法依赖大模型与大数据、计算资源需求高的问题,提出结合期望最大化算法的轻量级模型EM-Net,通过GAM与分层特征学习提升性能与泛化性,在仅用半量训练数据时仍优于对比方法且抗噪性良好。

摘要:

近年来,视线估计技术的准确率逐步提升,但现有方法往往依赖大规模数据集或大模型来提高性能,导致对计算资源的需求较高。针对这一问题,本文结合深度学习与传统机器学习算法期望最大化(Expectation Maximization)算法,提出了一种轻量级视线估计模型EM-Net。首先,该模型加入了所提出的全局注意力机制(Global Attention Mechanism,GAM)来提取与视线估计相关的特征,以提升模型捕捉全局依赖关系的能力,进而改善其性能。其次,通过EM模块学习分层特征表示,模型具备了较强的泛化能力,从而降低了对样本量的需求。实验证实,在仅使用50%训练数据的前提下,EM-Net在Gaze360、MPIIFaceGaze和RT-Gene数据集上的性能相较于GazeNAS-ETH分别提升了2.2%、2.02%和2.03%,且在面对高斯噪声干扰时也表现出良好的鲁棒性。

英文摘要:

In recent years, the accuracy of gaze estimation techniques has gradually improved, but existing methods often rely on large datasets or large models to improve performance, which leads to high demands on computational resources. In terms of this issue, this paper proposes a lightweight gaze estimation model EM-Net based on deep learning and traditional machine learning algorithms Expectation Maximization algorithm. First, the proposed Global Attention Mechanism(GAM) is added to extract features related to gaze estimation to improve the model's ability to capture global dependencies and thus improve its performance. Second, by learning hierarchical feature representations through the EM module, the model has strong generalization ability, which reduces the need for sample size. Experiments have confirmed that, on the premise of using only 50% of the training data, EM-Net improves the performance of Gaze360, MPIIFaceGaze, and RT-Gene datasets by 2.2%, 2.02%, and 2.03%, respectively, compared with GazeNAS-ETH. It also shows good robustness in the face of Gaussian noise interference.

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

↑