跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

透過記憶回溯與拓撲歸因實現時間圖網路的可解釋性

文章摘要

本文提出MemExplainer方法,透過拓撲歸因樹和記憶回溯樹來解釋時間圖網路(TGN)的預測。該方法首次考慮TGN中的記憶模組,量化歷史事件對節點記憶向量的影響,並利用層間相關傳播(LRP)確保事件貢獻之和等於模型logits。在九個資料集上的實驗表明,該方法在節點屬性預測、連結預測和圖分類任務中均優於現有基線,提供了忠實於模型的解釋。該論文被ICML 2026接收為Spotlight論文,程式碼已開源。

來源arXiv Machine Learning作者: Yazheng Liu, Xi Zhang, Sihong Xie, Hui Xiong
透過記憶回溯與拓撲歸因實現時間圖網路的可解釋性
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

時間圖網路(TGN)在眾多現實應用中表現出色,但理解其預測背後的歷史事件對於提升可信度至關重要。然而,現有解釋方法大多忽略了TGN的核心元件——記憶模組,該模組負責記錄和更新節點歷史,導致過去事件的影響未被充分挖掘。來自研究團隊的最新論文提出了一種名為MemExplainer的方法,旨在透過拓撲歸因樹和記憶回溯樹來歸因TGN的預測,從而解決這一空白。

MemExplainer的核心由兩個部分組成:拓撲歸因樹和記憶回溯樹。前者主要捕獲鄰居節點及其記憶向量的影響,後者則量化歷史事件如何塑造節點的記憶向量。研究團隊進一步將層間相關傳播(LRP)應用於TGN,確保所有事件的貢獻之和精確等於模型的logits值。此外,由於從logits到機率的非線性對映可能導致簡單的top-k選擇不忠實,他們設計了專門的最佳化目標來識別真正重要的事件。

實驗在九個時間圖資料集上進行,涵蓋了節點屬性預測、連結預測和圖分類三種任務。結果表明,MemExplainer不僅能夠提供忠實的解釋,而且在各項指標上均超越了當前最先進的基線方法。該研究的程式碼已在GitHub上開源,便於其他研究者復現和應用。這一成果被ICML 2026接收為Spotlight論文,為時間圖網路的可解釋性研究開闢了新的方向,並有望推動TGN在金融、社交網路等高風險領域的可靠部署。

展開要點與分析

文章情報

工程師進階

要點

  • 現有TGN解釋方法忽略了記憶模組,MemExplainer首次將其納入解釋框架。
  • 提出拓撲歸因樹和記憶回溯樹,分別捕獲鄰居影響和歷史事件貢獻。
  • 應用LRP確保事件貢獻總和等於模型logits,並設計最佳化目標處理top-k不忠實問題。
  • 在九個時間圖資料集上實驗,涵蓋多種任務,效能超越當前最優基線。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。