用柯尔莫戈洛夫算子增强图卷积,提升不规则空间动态预测精度
Transformer with Koopman-Enhanced Graph Convolutional Network for Spatiotemporal Dynamics Forecasting
- 引入柯尔莫戈洛夫算子理论,将非线性动态映射到近似线性的潜在空间
- 两阶段设计:先降维再用Transformer捕捉长时依赖,预测误差显著降低
- 适用于心脏等复杂系统建模,适合研究时空动态的科研人员
时空动态预测在不规则几何域上尤为困难,需同时捕捉复杂的空间相关性和非线性时间动态。为此,我们提出TK-GCN,一种两阶段框架,结合几何感知的空间编码与长程时间建模。第一阶段,构建柯尔莫戈洛夫增强图卷积网络(K-GCN),将分布在空间不规则域上的高维动态投影至潜在空间,使系统状态演化近似线性;利用柯尔莫戈洛夫算子理论,增强潜在学习中的时间一致性。第二阶段,采用Transformer模块在柯尔莫戈洛夫编码的潜在空间中建模时间演进,通过自注意力机制捕获长程时间依赖,实现长时间跨度的精准预测。我们在心电时空动态预测任务中评估了TK-GCN,并与多个先进基线模型对比。实验结果和消融研究显示,TK-GCN在多种预测时长远超基线,充分展现了其对复杂空间结构和非线性时间动态的建模能力。
原文摘要 · Abstract (English)
Spatiotemporal dynamics forecasting is inherently challenging, particularly in systems defined over irregular geometric domains, due to the need to jointly capture complex spatial correlations and nonlinear temporal dynamics. To tackle these challenges, we propose TK-GCN, a two-stage framework that integrates geometry-aware spatial encoding with long-range temporal modeling. In the first stage, a Koopman-enhanced Graph Convolutional Network (K-GCN) is developed to embed the high-dimensional dynamics distributed on spatially irregular domains into a latent space where the evolution of system states is approximately linear. By leveraging Koopman operator theory, this stage enhances the temporal consistency during the latent learning. In the second stage, a Transformer module is employed to model the temporal progression within the Koopman-encoded latent space. Through the self-attention mechanism, the Transformer captures long-range temporal dependencies, enabling accurate forecasting over extended horizons. We evaluate TK-GCN in spatiotemporal cardiac dynamics forecasting and benchmark its performance against several state-of-the-art baselines. Experimental results and ablation studies show that TK-GCN consistently delivers superior predictive accuracy across a range of forecast horizons, demonstrating its capability to effectively model complex spatial structures and nonlinear temporal dynamics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。