Saved in:
Bibliographic Details
Main Authors: Wang, Yi, Wang, Zhenghong, Zhang, Fan, Kang, Chaogui, Ruan, Sijie, Zhu, Di, Tang, Chengling, Ma, Zhongfu, Zhang, Weiyu, Zheng, Yu, Yu, Philip S., Liu, Yu
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2506.13678
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909868160450560
author Wang, Yi
Wang, Zhenghong
Zhang, Fan
Kang, Chaogui
Ruan, Sijie
Zhu, Di
Tang, Chengling
Ma, Zhongfu
Zhang, Weiyu
Zheng, Yu
Yu, Philip S.
Liu, Yu
author_facet Wang, Yi
Wang, Zhenghong
Zhang, Fan
Kang, Chaogui
Ruan, Sijie
Zhu, Di
Tang, Chengling
Ma, Zhongfu
Zhang, Weiyu
Zheng, Yu
Yu, Philip S.
Liu, Yu
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2506_13678
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Gravity-informed Spatiotemporal Transformer for Human Activity Intensity Prediction
Wang, Yi
Wang, Zhenghong
Zhang, Fan
Kang, Chaogui
Ruan, Sijie
Zhu, Di
Tang, Chengling
Ma, Zhongfu
Zhang, Weiyu
Zheng, Yu
Yu, Philip S.
Liu, Yu
Machine Learning
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.
title A Gravity-informed Spatiotemporal Transformer for Human Activity Intensity Prediction
topic Machine Learning
url https://arxiv.org/abs/2506.13678