Test-Time Search for Automated GFM Fine-Tuning
Wenji Hu, Xianan Wang, Chunyu Wei, Senhao Liu, Kuien Liu, Yunhai Wang, Yueguo Chen
Abstract
Graph Foundation Models (GFMs) have emerged as a powerful paradigm for learning transferable graph representations, yet adapting them to downstream tasks requires navigating an exponentially large decision space, traditionally demanding heavy expert effort. We propose GFMTuner, a framework that automates GFM fine-tuning by combining Large Language Model (LLM) agents with Monte Carlo Tree Search. GFMTuner accepts natural language task descriptions and generates effective fine-tuning strategies through test-time search. We introduce the Graph-Instructed Actor, which equips the LLM with graph analysis tools to ground action generation in structural insights, and Gradient Consistency, a self-supervised reward that measures gradient alignment across perturbed executions for efficient strategy evaluation. Experiments across diverse graph domains demonstrate that GFMTuner matches or exceeds human expert designs while reducing effort from weeks to a single natural language query. The code is available in https://github.com/GFMTuner/GFMTuner
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Related papers
- LLaMo: Large Language Model-based Molecular Graph AssistantJinyoung Park, Minseong Bae, Dohwan Ko, Hyunwoo J. KimNeurIPS 2024 · 33 citations
- GraphTranslator: Aligning Graph Model to Large Language Model for Open-ended TasksMengmei Zhang, Mingwei Sun, Peng Wang, Shen Fan et al.WWW 2024 · 99 citations
- GraphChain: Large Language Models for Large-scale Graph Analysis via Tool ChainingChunyu Wei, Wenji Hu, Xingjia Hao, Xin Wang et al.NeurIPS 2025 · 7 citations
- UniGTE: Unified Graph-Text Encoding for Zero-Shot Generalization across Graph Tasks and DomainsDuo Wang, Yuan Zuo, Guangyue Lu, Junjie WuNeurIPS 2025 · 9 citations
- Plan Then Retrieve: Reinforcement Learning-Guided Complex Reasoning over Knowledge GraphsYanlin Song, Ben Liu, Víctor Gutiérrez-Basulto, Zhiwei Hu et al.WWW 2026
