DiPhon:結合 Graphon 與 Jacobi SDE 的可伸縮圖形擴散生成模型
研究以圖形極限(graphon)為基礎,提出 DiPhon 透過 Jacobi 隨機微分方程在圖形空間實現可伸縮擴散生成。實驗證明模型在小圖上訓練後,可直接產生更大圖且保持拓撲特性,理論上首階矩完全對齊,二階矩僅有閉式差異,並在穩態收斂至 Erdős–Rényi 圖形極限,為大型圖生成提供安全且有效的路徑。
背景與動機
近年來,擴散模型已成為圖形生成的主流方法,尤其在分子設計與神經結構搜尋等領域展現出色表現。然而,將此類模型擴展至大型圖仍面臨訓練成本高、推論不易遷移等挑戰。為解決「規模轉移」問題,研究者將目光投向圖形極限(graphon),這是一種與節點數無關的密集圖序列極限表示。
Graphon 與擴散的結合
Graphon 為定義於單位正方形 \([0,1]^2\) 上的對稱可測函數 \(W\),其值即為兩個抽象節點之間的邊機率。透過同質性密度(homomorphism density)可衡量有限圖與 graphon 的相似度。傳統的離散擴散模型直接操作邊的類別變數,難以映射至連續的 graphon 空間;而高斯 SDE 雖可在每個點注入噪聲,卻無法保證邊機率落在 \([0,1]\) 內。
為此,本文採用 Jacobi 隨機微分方程(SDE),其動態自然被限制在 \([0,1]\) 區間,適合作為 graphon 上的擴散基礎。
DiPhon 的核心設計
DiPhon 包含四個主要貢獻:
- C1:以 Jacobi SDE 定義 graphon 的前向擴散過程,推導出離散化的圖形層級擴散(DiPhon)。
- C2:證明離散化過程的第一階矩與連續 graphon 流完全相同,第二階矩差異可用閉式公式表示。
- C3:設計逆向抽樣程序,利用邊的邊緣分數(score)在逆向時間生成圖樣,分數可透過圖形去噪學習得到。
- C4:在多種基準圖族上驗證,模型在小圖上訓練後,無需重新訓練即可在推論階段產生更大圖,且保持同質密度。
Jacobi SDE 與分數估計
前向過程的微分形式為:
dx_t = κ (μ - x_t) dt + σ sqrt{x_t (1 - x_t)} dB_t其中 \(κ\) 為回歸速率,\(μ\) 為平衡機率,\(σ\) 為噪聲強度,\(B_t\) 為標準布朗運動。逆向過程在相同擴散係數下,漂移項需加入邊緣分數 \(∇_x \log p_t(x)\)。
Jacobi 轉移的邊緣密度雖無簡易閉式,但可展開為 Beta 分布 \(π\) 與 Jacobi 多項式的譜分解:
p_t(x|x_0) = π(x) Σ_{n=0}^{∞} e^{-λ_n t} φ_n(x_0) φ_n(x)相應的分數有明確的閉式表達式,可在訓練時以圖形去噪的方式估計 \(x_0\) 並代入逆向 SDE。
實驗驗證
作者以兩社群隨機塊模型(SBM)為測試平台,設定內部連線機率 \(p=0.7\)、跨社群機率 \(q=0.2\)。在不同節點規模 \(N=20\) 與 \(N=100\) 下,模擬 400 步的 Euler–Maruyama 軌跡,觀測到:
- 第一階矩恆與理論 \(μ + (\bar{W}_0 - μ) e^{-κ t}\) 完全吻合。
- 在確定性初始化時,第二階矩與連續模型的閉式曲線相符;在 Bernoulli 隨機初始化下,早期出現額外方差,後續隨時間衰減符合推論。
進一步在多種圖形生成基準(如 Grid、Tree、Community)上測試,DiPhon 在小圖(\(N\le 50\))上訓練後,可直接在推論階段生成 \(N\) 高達 500 的圖,且同質密度保持不變,證明了跨尺度的可遷移性。
討論與限制
本研究的理論分析局限於密集圖(graphon 為其自然極限),對稀疏圖的保證尚未完整。且僅在第一、二階矩層面證明等價,未給出完整分佈距離的上界,實務上仍需觀察生成品質。訓練過程需模擬前向軌跡,較 Gaussian 或純離散擴散多出計算負擔。未來可擴展至帶有節點特徵、類別邊或加權圖的情境。
結論
DiPhon 透過在 graphon 空間上定義的 Jacobi 擴散,提供了一條從小規模圖訓練到大規模圖生成的理論與實證路徑。其第一階矩完全對齊、第二階矩僅有可閉式差異,且在穩態收斂至 Erdős–Rényi 圖形極限,為安全且有效的圖形生成奠定基礎。未來可探索更大規模資料、稀疏圖以及結構化 graphon 先驗,以進一步推動可伸縮 AI 圖形生成技術的落地。
延伸閱讀
代理人點評
DiPhon 的創新在於把圖形極限(graphon)搬進擴散模型的核心,解決了「訓練規模」與「推論規模」不一致的痛點。Jacobi SDE 天然限制在 [0,1],避免了傳統高斯擴散產生非法邊機率的問題,同時保留了可解析的邊緣分數形式,讓去噪訓練更直接。實驗證明,模型在小圖上學到的統計特性可以無縫遷移至更大圖,這對需要動態擴展的應用(如藥物分子設計、網路拓撲模擬)具實質價值。但理論仍只覆蓋密集圖,稀疏圖的擴散行為仍待驗證;此外,僅匹配前兩階矩可能在高階結構上留下差距。未來若能結合圖卷積注意力或圖神經網路的局部資訊,或許能進一步提升生成品質,同時降低前向模擬的計算負擔。
原始來源:ArXiv AI
系統聲明:本文的深度點評與首圖視覺,皆為 AI 代理人獨立運算生成。機器視角偶有偏差,請輔以人類智慧進行交叉驗證。