Han Xinyu, Zhao Yi
IEEE Trans Neural Netw Learn Syst. 2024 Jul;35(7):9198-9212. doi: 10.1109/TNNLS.2022.3231620. Epub 2024 Jul 8.
Graph reservoir computing (GraphRC) gains increasing attention by virtue of its high training efficiency. However, since GraphRC is developed without knowledge of its internal mechanism, it cannot be fully trusted to deploy in practice. Although there are some existing approaches that can be extended to interpret GraphRC, the specific role played by each neuron (i.e., reservoir node) of GraphRC is far less explored. To address this issue, the latent short-term memory property of each reservoir node of GraphRC is qualitatively characterized to unravel its role in predicting the graph signal, thereby enabling an interpretable GraphRC. Specifically, we first deduce the equivalence between the GraphRC and conventional reservoir computing (RC). Then, the underlying memory properties of the GraphRC and its reservoir nodes can be characterized in theory by the multisource reachability among the reservoir nodes in the transformed RC. Moreover, the distinct temporal patterns hidden in reservoir nodes are identified, and then, an attention mechanism based on the identified temporal patterns is deployed in the GraphRC to improve its performance. In addition, the effectiveness of the interpretability for GraphRC and improved GraphRC is verified on the Lorenz-96 spatiotemporal dynamical system. The experimental results of the Lorenz-96 spatiotemporal chaotic system and three real-world traffic datasets demonstrate that the improved GraphRC is superior to original GraphRC and can achieve prediction performance comparable to the state-of-the-art baseline models, but with much less training cost.
图储层计算(GraphRC)凭借其高训练效率而越来越受到关注。然而,由于GraphRC是在不了解其内部机制的情况下开发的,因此在实际部署中不能完全信任它。尽管有一些现有方法可以扩展用于解释GraphRC,但GraphRC的每个神经元(即储层节点)所起的具体作用却很少被探索。为了解决这个问题,对GraphRC的每个储层节点的潜在短期记忆特性进行了定性表征,以揭示其在预测图信号中的作用,从而实现可解释的GraphRC。具体来说,我们首先推导了GraphRC与传统储层计算(RC)之间的等价关系。然后,通过变换后的RC中储层节点之间的多源可达性,从理论上表征GraphRC及其储层节点的潜在记忆特性。此外,识别出隐藏在储层节点中的不同时间模式,然后在GraphRC中部署基于所识别时间模式的注意力机制以提高其性能。另外,在Lorenz-96时空动力学系统上验证了GraphRC和改进后的GraphRC的可解释性的有效性。Lorenz-96时空混沌系统和三个真实世界交通数据集的实验结果表明,改进后的GraphRC优于原始GraphRC,并且可以实现与最先进的基线模型相当的预测性能,但训练成本要低得多。