Bridging Distribution Gaps: Invariant Pattern Discovery for Dynamic Graph Learning

Yucheng Jin,Maoyi Wang,Yun Xiong,Zhizhou Ren,Cuiying Huo,Feng Zhu,Jiawei Zhang,Guangzhong Wang,Haoran Chen
DOI: https://doi.org/10.1007/s11280-024-01283-2
2024-01-01
World Wide Web
Abstract:Temporal graph networks (TGNs) have been proposed to facilitate learning ondynamic graphs which are composed of interaction events among nodes. How-ever, existing TGNs suffer from poor generalization under distribution shifts thatoccur over time. It is vital to discover invariant patterns with stable predictivepower across various distributions to improve the generalization ability. Invari-ant pattern discovery on dynamic graphs is non-trivial, as long-term history ofinteraction events is compressed into the memory by TGNs in an entangled way,making invariant pattern discovery difficult. Furthermore, TGNs process inter-action events chronologically in batches to obtain up-to-date representations.Each batch consisting of chronologically-close events lacks diversity for identify-ing invariance under distribution shifts. To tackle these challenges, we proposea novel method called Smile, which stands for Structural teMporal InvariantLEarning. Specifically, we first propose the disentangled graph memory network,which selectively extracts pattern information from long-term history throughthe disentangled memory gating and attention network. The interaction historyapproximator is further introduced to provide diverse interaction distributionsefficiently. Smile guarantees prediction stability under diverse temporal-dynamic distributions by regularizing invariance under cross-time distribution interven-tions. Experimental results on real-world datasets demonstrate that Smileoutperforms baselines, yielding substantial performance improvements.
What problem does this paper attempt to address?