用万有引力定律改进Transformer,让人类活动预测更准确可信
A Gravity-informed Spatiotemporal Transformer for Human Activity Intensity Prediction
- 将万有引力模型融入注意力机制,学习物理约束下的空间交互
- 在6个真实数据集上超越现有方法,零样本跨区域预测效果提升明显
- 可解释性强,学到的引力矩阵符合地理规律,适合城市规划等场景
人类活动强度预测对位置服务至关重要。尽管现有方法在建模活动动态方面取得进展,但大多忽略空间交互的物理约束,导致空间相关性不可解释且出现过度平滑现象。为此,本文提出一种物理信息深度学习框架——引力感知时空Transformer(Gravityformer),通过引入万有引力定律优化Transformer注意力。具体包括:(1) 基于时空嵌入特征估计两个显式空间质量参数;(2) 在端到端神经网络中使用自适应引力模型建模空间交互,学习物理约束;(3) 利用学习到的空间交互引导并缓解Transformer注意力中的过度平滑问题。此外,提出并行时空图卷积Transformer以平衡时空联合学习。在六个真实世界大规模活动数据集上的系统实验表明,本模型在定量和定性上均优于当前最优基准。同时,学习到的引力注意力矩阵可基于地理规律解耦与解释,并显著提升零样本跨区域推理的泛化能力。该工作为融合物理定律与深度学习进行时空预测提供了新思路。
原文摘要 · Abstract (English)
Human activity intensity prediction is crucial to many location-based services. Despite tremendous progress in modeling dynamics of human activity, most existing methods overlook physical constraints of spatial interaction, leading to uninterpretable spatial correlations and over-smoothing phenomenon. To address these limitations, this work proposes a physics-informed deep learning framework, namely Gravity-informed Spatiotemporal Transformer (Gravityformer) by integrating the universal law of gravitation to refine transformer attention. Specifically, it (1) estimates two spatially explicit mass parameters based on spatiotemporal embedding feature, (2) models the spatial interaction in end-to-end neural network using proposed adaptive gravity model to learn the physical constraint, and (3) utilizes the learned spatial interaction to guide and mitigate the over-smoothing phenomenon in transformer attention. Moreover, a parallel spatiotemporal graph convolution transformer is proposed for achieving a balance between coupled spatial and temporal learning. Systematic experiments on six real-world large-scale activity datasets demonstrate the quantitative and qualitative superiority of our model over state-of-the-art benchmarks. Additionally, the learned gravity attention matrix can be not only disentangled and interpreted based on geographical laws, but also improved the generalization in zero-shot cross-region inference. This work provides a novel insight into integrating physical laws with deep learning for spatiotemporal prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。