跳到正文
原文
arXiv 机器学习· Yancheng Wang, Ping Li, Dongfang Sun, Yingzhen Yang·· 14 小时前AI 评分34

RGDM:鲁棒图扩散模型

RGDM: Robust Graph Diffusion Model

AI 导读

研究人员提出鲁棒图扩散模型 RGDM,该模型由鲁棒图自编码器(RGAE)与潜在扩散模型(LDM)组成,用于生成带标签的合成图结构并无缝融入给定图。RGDM 结合边选择机制与潜在特征低秩正则化,即使在对抗攻击图上训练也能生成高质量干净结构。实验显示,经其增强的数据可显著提升 GNN 在半监督节点分类中的抗攻击鲁棒性,代码已开源。

来源:arXiv 机器学习 · arxiv.org