arXiv:2505.02027v1 交叉类型：公告 摘要：图上下文学习（Graph In-Context Learning）具备在不更新任何参数的情况下将预训练的图模型适应于新颖且多样的下游图的优势，已在研究界引起了广泛关注。图上下文学习的关键在于根据选定的提示示例对下游图进行条件化处理。现有的方法随机选择子图或边作为提示，这导致了提示图的噪声，并且模型性能较差。此外，由于预训练图和测试图之间存在差距，当测试图中的类的数量远大于训练图中的类的数量时，上下文学习的能力也会显著下降。为了应对上述挑战，我们开发了一种多阶段自适应提示优化方法——GraphPrompter，该方法优化了生成、选择和使用图提示的整个过程，以提高上下文学习的能力。首先，提示生成器引入了一个重构层，以突出最具有信息性的边并减少图提示构建过程中的无关噪声。此外，在选择阶段，提示选择器使用 $k$-最近邻算法和预训练的选择层来动态选择合适的样本并最小化无关提示的影响。最后，我们利用具有缓存替换策略的提示增强器来增强预训练模型在新数据集上的泛化能力。广泛的实验表明，GraphPrompter 有效地增强了图模型的上下文学习能力。在所有设置的平均性能上，我们的方法比最新的基线高出超过 8%。我们的代码已发布于 https://github.com/karin0018/GraphPrompter。