时间图网络可解释性新突破:记忆回溯与拓扑归因方法登顶ICML 2026
时间图网络在社交网络、金融交易、生物信息等动态场景中广泛应用,但其内部记忆模块——负责记录和更新节点历史状态的核心组件——长期处于“黑箱”状态。现有解释方法多聚焦于拓扑结构,却忽略了记忆模块如何整合历史事件来驱动预测。来自北京邮电大学、里海大学和香港科技大学的研究团队提出了一种名为记忆回溯与拓扑归因(Memory Backtracking and Topological Attribution) 的方法,首次系统地将TGN的记忆机制纳入可解释性框架,相关工作已被ICML 2026接收为Spotlight论文。
两大回溯树:从拓扑到记忆的完整归因
该方法的核心在于构建两棵互补的归因树:
- 拓扑归因树:捕捉邻居节点及其记忆向量对当前预测的贡献。通过层级分解,模型能够定位哪些相邻节点在特定时间步产生了关键影响。
- 记忆回溯树:进一步量化历史事件如何塑造节点记忆向量本身。这意味着,即使一个事件发生在较远的时间点,只要它对记忆向量产生持续影响,也能被准确追溯。
这种“从拓扑到记忆”的双树结构,使得归因链条能够完整覆盖事件→记忆→预测的全路径。
LRP适配与优化目标:解决概率映射失真
研究团队将分层相关性传播(LRP) 适配到TGN中,确保所有历史事件的总贡献等于模型输出的logits值,从而在数学上保证了归因的忠实性。此外,针对传统top-k选择方法因logits到概率的非线性映射而可能失真的问题,他们设计了专门的优化目标来筛选真正重要的事件,而非简单依赖阈值截断。
九大数据集验证:全面超越现有基线
实验覆盖了节点属性预测、链接预测和图分类三类任务,涉及九个公开时间图数据集。结果表明,该方法不仅在忠实度指标上显著优于现有最先进基线,还能提供更符合直觉的解释。例如,在社交网络动态链接预测中,模型能够正确识别出导致好友关系建立的关键早期互动,而非仅仅关注近期事件。
行业意义:从“能用”到“可信”的关键一步
时间图网络在金融反欺诈、交通流量预测、推荐系统等领域具有巨大潜力。然而,缺乏可解释性一直制约其在合规要求严格的行业落地。记忆回溯与拓扑归因方法首次揭示了记忆模块中的“因果链”,使开发者能够验证模型是否依赖了合理的业务逻辑,而非偶然的统计关联。随着ICML 2026的认可,这一思路有望成为TGN可解释性的标准范式。