arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.06110cs.CV

FGA:面向人群计数估计的傅里叶引导注意力网络

FGA: Fourier-Guided Attention Network for Crowd Count Estimation

IIIT-Delhi(德里印度信息技术学院);IIT-Bombay(孟买印度理工学院);University of Jeddah(吉达大学)

Yashwardhan Chaudhuri, Ankit Kumar, Arun Balaji Buduru, Adel Alshamrani

AI总结:

针对现有卷积注意力网络全尺度全局模式捕获效率低的问题,提出融合频域与空间信息的双路径FGA机制,应用于主流人群计数模型后在多个基准数据集上取得性能提升,并验证了可解释性。

摘要:

人群计数正日益凸显其社会价值,尤其在城市规划、人群管理与公共安全领域。本文提出傅里叶引导注意力(Fourier-guided attention,FGA),这是一种用于人群计数估计的新型注意力机制,旨在解决现有基于卷积的注意力网络在全尺度全局模式捕获方面效率低下的问题。FGA结合用于全局特征的快速傅里叶变换(Fast-Fourier Transformations,FFT)与空间注意力,以及用于半全局和局部特征的带通道注意力的卷积,可高效捕获包含全尺度全局模式的多尺度信息。FGA采用双路径架构:(1)通过FFT处理全尺度全局特征的路径,可在频域高效提取信息;(2)使用传统卷积与通道注意力处理其余特征图以获取半全局和局部特征的路径。这种双路径架构使FGA能够无缝融合频域与空间信息,增强其捕获多样人群模式的能力。我们将FGA应用于两种主流人群计数模型CSRNet和CANNet的最后几层,在ShanghaiTech-A、ShanghaiTech-B、UCF-CC-50、JHU++ crowd等基准数据集上评估该模块的性能。实验表明,在均方误差(Mean-Squared-Error,MSE)和平均绝对误差(Mean-Absolute-Error,MAE)指标上,所有数据集的性能均有显著提升,表现可与近期的最优方法相媲美。此外,我们利用Grad-CAM热力图通过定性分析阐释了模型的可解释性,以证明FGA在捕获人群模式方面的有效性。

英文摘要:

Crowd counting is gaining societal relevance, particularly in domains of Urban Planning, Crowd Management, and Public Safety. This paper introduces Fourier-guided attention (FGA), a novel attention mechanism for crowd count estimation designed to address the inefficient full-scale global pattern capture in existing works on convolution-based attention networks. FGA efficiently captures multi-scale information, including full-scale global patterns, by utilizing Fast-Fourier Transformations (FFT) along with spatial attention for global features and convolutions with channel-wise attention for semi-global and local features. The architecture of FGA involves a dual-path approach: (1) a path for processing full-scale global features through FFT, allowing for efficient extraction of information in the frequency domain, and (2) a path for processing remaining feature maps for semi-global and local features using traditional convolutions and channel-wise attention. This dual-path architecture enables FGA to seamlessly integrate frequency and spatial information, enhancing its ability to capture diverse crowd patterns. We apply FGA in the last layers of two popular crowd-counting works, CSRNet and CANNet, to evaluate the module's performance on benchmark datasets such as ShanghaiTech-A, ShanghaiTech-B, UCF-CC-50, and JHU++ crowd. The experiments demonstrate a notable improvement across all datasets based on Mean-Squared-Error (MSE) and Mean-Absolute-Error (MAE) metrics, showing comparable performance to recent state-of-the-art methods. Additionally, we illustrate the interpretability using qualitative analysis, leveraging Grad-CAM heatmaps, to show the effectiveness of FGA in capturing crowd patterns.

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

↑