arXiv:2411.12913cs.LGcs.AI2024-11KDD被引 12

提出可自适应跨域泛化的图神经网络框架,提升模型在未知分布上的表现。

MLDGG: Meta-Learning for Domain Generalization on Graphs

  • 融合元学习与结构学习,动态优化图结构和语义表示
  • 在三种分布偏移场景下均超越基线方法,提升泛化性能
  • 适合处理数据分布不一致的图数据任务,如跨领域推荐

图上的域泛化旨在开发具备强鲁棒泛化能力的模型,确保在测试集上面对训练与测试分布差异时仍能有效运行。然而,现有方法常依赖静态编码器直接应用于目标域,限制了其灵活适应性。不同于传统方法聚焦于构建特定泛化模型,本框架MLDGG通过结合跨多域元学习、结构学习与语义识别,实现对多样化域的自适应泛化。首先引入通用结构学习器,缓解无关边的负面影响,增强图神经网络(GNN)表示的完整性并捕捉跨域共享结构信息;其次设计表示学习器,利用因果推理分离节点嵌入中的域不变语义与域特定变化信息,进一步提升泛化能力。在元学习框架下,两个学习器的元参数通过优化实现知识迁移,并支持在目标域内通过微调实现有效适应——目标图在训练阶段不可见。实验结果表明,MLDGG在三种不同分布偏移设置下均优于基线方法,验证了其有效性。

原文摘要 · Abstract (English)

Domain generalization on graphs aims to develop models with robust generalization capabilities, ensuring effective performance on the testing set despite disparities between testing and training distributions. However, existing methods often rely on static encoders directly applied to the target domain, constraining its flexible adaptability. In contrast to conventional methodologies, which concentrate on developing specific generalized models, our framework, MLDGG, endeavors to achieve adaptable generalization across diverse domains by integrating cross-multi-domain meta-learning with structure learning and semantic identification. Initially, it introduces a generalized structure learner to mitigate the adverse effects of task-unrelated edges, enhancing the comprehensiveness of representations learned by Graph Neural Networks (GNNs) while capturing shared structural information across domains. Subsequently, a representation learner is designed to disentangle domain-invariant semantic and domain-specific variation information in node embedding by leveraging causal reasoning for semantic identification, further enhancing generalization. In the context of meta-learning, meta-parameters for both learners are optimized to facilitate knowledge transfer and enable effective adaptation to graphs through fine-tuning within the target domains, where target graphs are inaccessible during training. Our empirical results demonstrate that MLDGG surpasses baseline methods, showcasing its effectiveness in three different distribution shift settings.

图神经网络域泛化元学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。