因果扩散掩码自编码器

图分布外泛化

通过因果感知课程与扩散去噪,学习不变因果表征,显著提升图神经网络在分布偏移下的鲁棒性。

Jiahao, Liang · Chen, Wuxing · 余志文 · 张通 · Philip Chen, Chun Lung Philip · Yang, Kaixiang

Information Fusion 2026

技术优势

找出因果边

因果复杂度课程衡量器对边的因果重要性进行排序,引导模型优先学习关键因果结构,避免虚假关联。

逐步去噪

自步掩码调度器渐进地应用因果引导的扩散,通过逐步去噪捕获数据分布,增强泛化能力。

融入因果信息

因果感知扩散掩码自编码器将因果重要性融入编码过程,并在逐步去噪中施加因果正则化,学习不变表征。

性能显著提升

在图OOD基准上的大量实验显示,本方法相比现有方法有显著性能提升,且消融实验证实因果引导课程与扩散过程优于传统预训练方法。

应用场景