可扩展时空Transformer

线性复杂度

一种面向大规模城市网络的低复杂度时空预测模型,通过低维嵌入诱导的类Transformer注意力机制实现可解释的神经扩散,在保持先进性能的同时显著提升可扩展性。

Nie, Tong · 孙剑 · Ma, Wei

IEEE Transactions on Industrial Informatics 2025

技术优势

计算量随规模线性增长

模型采用线性复杂度的设计,使得在更大规模网络上进行预测在计算上可行。

注意力由低维嵌入诱导

利用低维嵌入诱导注意力层,在保持性能的同时降低了计算开销。

结构具有可解释性

模型设计遵循物理基本规律,避免架构冗余,因此更加透明,便于用户理解其预测依据。

应用场景