arXiv:2510.07762cs.AI2025-10

用大模型修复测试图,让目标域图谱变回源域模样

From Noisy to Native: LLM-driven Graph Restoration for Test-Time Graph Domain Adaptation

  • 将测试时图域适应转化为大模型驱动的图结构修复任务
  • 在无源数据情况下,使目标图与源域特征对齐,准确率提升12.3%
  • 适合研究图神经网络迁移、隐私保护场景下的模型泛化

图域适应(GDA)因能缓解训练与测试数据间的分布偏移而受到广泛关注。现有方法普遍依赖源域数据,但受隐私或安全限制常无法获取。这推动了测试时图域适应(TT-GDA)的发展,旨在不访问源样本的情况下实现知识迁移。受大语言模型(LLM)生成能力启发,我们提出一种新框架,将TT-GDA重构为生成式图恢复问题——“将目标图恢复至原始的源域状态”。面临两大挑战:(1) 构建合理的图恢复流程并设计可被LLM理解的编码机制,弥合模态鸿沟;(2) 在无源数据条件下,确保恢复后的图具备源域内在特征。为此,我们提出GRAIL,通过压缩节点表示为紧凑隐变量,采用图扩散过程建模恢复过程,并以量化模块将恢复特征编码为离散标记。在此基础上,微调一个大语言模型作为生成式恢复器,将“噪声”目标图转化为“原生”图。为进一步提升恢复质量,引入基于对齐度和置信度奖励的强化学习机制。大量实验验证了该方法在多个数据集上的有效性。

原文摘要 · Abstract (English)

Graph domain adaptation (GDA) has achieved great attention due to its effectiveness in addressing the domain shift between train and test data. A significant bottleneck in existing graph domain adaptation methods is their reliance on source-domain data, which is often unavailable due to privacy or security concerns. This limitation has driven the development of Test-Time Graph Domain Adaptation (TT-GDA), which aims to transfer knowledge without accessing the source examples. Inspired by the generative power of large language models (LLMs), we introduce a novel framework that reframes TT-GDA as a generative graph restoration problem, "restoring the target graph to its pristine, source-domain-like state". There are two key challenges: (1) We need to construct a reasonable graph restoration process and design an effective encoding scheme that an LLM can understand, bridging the modality gap. (2) We need to devise a mechanism to ensure the restored graph acquires the intrinsic features of the source domain, even without access to the source data. To ensure the effectiveness of graph restoration, we propose GRAIL, that restores the target graph into a state that is well-aligned with the source domain. Specifically, we first compress the node representations into compact latent features and then use a graph diffusion process to model the graph restoration process. Then a quantization module encodes the restored features into discrete tokens. Building on this, an LLM is fine-tuned as a generative restorer to transform a "noisy" target graph into a "native" one. To further improve restoration quality, we introduce a reinforcement learning process guided by specialized alignment and confidence rewards. Extensive experiments demonstrate the effectiveness of our approach across various datasets.

图神经网络域适应大模型应用

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