Towards Graph Foundation Models: Learning Generalities Across Graphs via Task-Trees
Created by
Haebom
저자
Zehong Wang, Zheyuan Zhang, Tianyi Ma, Nitesh V Chawla, Chuxu Zhang, Yanfang Ye
개요
본 논문은 이종 그래프 작업 전반에서 일반화 가능한 패턴을 발견하는 문제를 해결하기 위해, 작업 트리를 이용한 그래프 내 크로스-태스크 일반화에 대한 새로운 접근 방식을 제안합니다. 작업 트리는 노드, 에지, 그래프 수준 작업을 정렬하는 통합된 학습 인스턴스 역할을 합니다. 이론적으로 작업 트리의 안정성, 전이성 및 일반화 특성을 분석하여 재구성 목표를 가진 다양한 작업 트리에서 GNN(그래프 신경망)을 사전 훈련하면 전이 가능한 지식이 유도됨을 보여줍니다. 이를 통해 최소한의 미세 조정으로 다운스트림 작업에 효율적으로 적응할 수 있습니다. 제안된 프레임워크를 검증하기 위해, 5개 도메인에 걸쳐 30개 이상의 그래프에서 미세 조정, 문맥 내 학습 및 제로샷 일반화를 통해 강력한 성능을 보여주는 그래프 기반 모델인 GIT(Graph Generality Identifier on Task-Trees)를 소개합니다. 코드와 데이터는 https://github.com/Zehong-Wang/GIT 에서 이용 가능합니다.