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

基於路徑的梯度提升用於圖級別預測

文章摘要

PathBoost是一種新的梯度樹提升方法,用於圖級別的分類和迴歸。它直接從輸入圖結構中學習基於路徑的判別特徵,擴充套件了先前的工作,支援二分類、多節點和邊屬性,並自動選擇錨點節點。在多個基準資料集上,PathBoost在半數測試中優於圖神經網路和圖核方法,其餘結果相當,尤其在大圖上表現更佳。

來源arXiv Machine Learning作者: Claudio Meggio, Johan Pensar, Riccardo De Bin
基於路徑的梯度提升用於圖級別預測
回報錯誤

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

查看更正說明
直接讀正文

研究人員Claudio Meggio及其合作者提出了一種名為PathBoost的新型梯度樹提升方法,專門用於圖級別的分類與迴歸任務。與傳統的黑盒模型不同,PathBoost能夠直接從輸入圖結構中學習具有判別能力的路徑特徵,從而顯著提升模型的可解釋性。該方法建立在先前針對特定化學應用的研究基礎上,但引入了三個關鍵擴充套件:首先,透過邏輯損失函式的梯度提升來適應二分類問題;其次,採用基於字首的分解技術,將多種節點和邊屬性整合到路徑特徵空間中;第三,基於類別屬性的多樣性自動選擇錨點節點,無需使用者手動指定路徑起點。

為了評估PathBoost的效能,研究團隊在多個基準資料集上將其與圖神經網路(GNN)和圖核方法進行了對比。實驗結果顯示,PathBoost在半數資料集上取得了更優的結果,在其餘資料集上則實現了相當的效能。特別值得注意的是,PathBoost在平均節點數較大的圖上表現尤為出色,這表明該方法具有良好的可擴充套件性。總體而言,這些結果證明基於路徑的梯度提升方法能夠與更為複雜的黑盒方法相競爭,為可解釋的圖級別預測提供了一條有前景的路徑。

展開要點與分析

文章情報

研究者進階

要點

  • PathBoost是一種基於路徑特徵的梯度提升方法,適用於圖級別預測。
  • 它透過字首分解整合多種節點和邊屬性,並自動選擇錨點節點。
  • 在多個基準上,PathBoost與GNN和核方法相比,半數更優,其餘持平。
  • 該方法在節點數較大的圖上表現更好,證明了路徑提升的競爭力。

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

基於路徑的梯度提升用於圖級別預測 | AI News Hub