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

基于单向卷积的多任务视线估计

Multi-task Gaze Estimation Via Unidirectional Convolution

College of Computer;Information Science Chongqing Normal University Chongqing, China

Zhang Cheng, Yanxia Wang

AI总结:

针对轻量模型用于视线估计时性能下降的问题,提出含单向卷积等模块的Multitask-Gaze模型,在两个数据集上性能优于SUGE且大幅降低参数量与计算量。

摘要:

在视线估计任务中使用轻量模型作为骨干网络通常会导致性能显著下降,主要原因是轻量网络的特征通道数量通常较少,使得模型表达能力受限。为提升轻量模型在视线估计任务中的性能,本文提出了名为Multitask-Gaze的网络模型。该模型的主要组成部分包括单向卷积(Unidirectional Convolution,UC)、空间与通道注意力(Spatial and Channel Attention,SCA)、全局卷积模块(Global Convolution Module,GCM)以及多任务回归模块(Multi-task Regression Module,MRM)。UC不仅显著减少了参数量和浮点运算量(FLOPs),还扩展了感受野,提升了模型的长距离建模能力,进而提高模型性能;SCA可突出与视线相关的特征,抑制与视线无关的特征;GCM替代池化层,避免了信息丢失导致的性能下降;MRM提升了单个任务的精度,同时强化了任务间的关联以实现整体性能提升。实验结果表明,与当前最优方法SUGE相比,Multitask-Gaze在MPIIFaceGaze和Gaze360数据集上的性能分别提升了1.71%和2.75%,同时参数量和FLOPs分别显著降低了75.5%和86.88%。

英文摘要:

Using lightweight models as backbone networks in gaze estimation tasks often results in significant performance degradation. The main reason is that the number of feature channels in lightweight networks is usually small, which makes the model expression ability limited. In order to improve the performance of lightweight models in gaze estimation tasks, a network model named Multitask-Gaze is proposed. The main components of Multitask-Gaze include Unidirectional Convolution (UC), Spatial and Channel Attention (SCA), Global Convolution Module (GCM), and Multi-task Regression Module(MRM). UC not only significantly reduces the number of parameters and FLOPs, but also extends the receptive field and improves the long-distance modeling capability of the model, thereby improving the model performance. SCA highlights gaze-related features and suppresses gaze-irrelevant features. The GCM replaces the pooling layer and avoids the performance degradation due to information loss. MRM improves the accuracy of individual tasks and strengthens the connections between tasks for overall performance improvement. The experimental results show that compared with the State-of-the-art method SUGE, the performance of Multitask-Gaze on MPIIFaceGaze and Gaze360 datasets is improved by 1.71% and 2.75%, respectively, while the number of parameters and FLOPs are significantly reduced by 75.5% and 86.88%.

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

↑