摘要
arXiv:2502.10459v1 交叉公告类型:cross
摘要:图神经网络架构搜索(GNAS)使人们能够自动设计针对特定下游图学习任务的图神经网络(GNNs)。然而,现有的GNAS方法通常需要手动适应新的图搜索空间,这需要大量的代码优化和领域特定知识。为了解决这一挑战,我们提出了一种名为LLM4GNAS的工具包,该工具包利用了大语言模型(LLMs)的生成能力。LLM4GNAS包括基于LLMs的图神经网络架构搜索算法的算法库,通过修改LLMs提示来适应新的搜索空间,从而减少了手动干预算法适应和代码修改的需要。LLM4GNAS工具包是可扩展和稳健的,它集成了增强的图特征工程、增强的图神经网络架构搜索和增强的超参数优化。实验结果表明,LLM4GNAS在涉及同质图和异质图的任务中均优于现有GNAS方法。