LLM2D
基于大型语言模型的图神经架构搜索工具包
LLM4GNAS: A Large Language Model Based Toolkit for Graph Neural Architecture Search
作者: Yang Gao, Hong Yang, Yizhi Chen, Junxian Wu, Peng Zhang, Haishuai Wang
发布日期: 2/18/2025
arXiv ID: oai:arXiv.org:2502.10459v1

摘要

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