Skip to main navigation Skip to search Skip to main content

Instruction-based Hypergraph Pretraining

  • Mingdai Yang
  • , Zhiwei Liu
  • , Liangwei Yang
  • , Xiaolong Liu
  • , Chen Wang
  • , Hao Peng*
  • , Philip S. Yu
  • *Corresponding author for this work
  • University of Illinois at Chicago
  • Salesforce AI Research
  • Alpha Innovation Institute
  • Shenzhen Institute of Computing Sciences

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Pretraining has been widely explored to augment the adaptability of graph learning models to transfer knowledge from large datasets to a downstream task, such as link prediction or classification. However, the gap between training objectives and the discrepancy between data distributions in pretraining and downstream tasks hinders the transfer of the pre-trained knowledge. Inspired by instruction-based prompts widely used in pre-trained language models, we introduce instructions into graph pertaining. In this paper, we propose a novel pretraining framework named Instruction-based Hypergraph Pretraining. To overcome the discrepancy between pretraining and downstream tasks, text-based instructions provide explicit guidance on specific tasks for representation learning. Compared to learnable prompts, whose effectiveness depends on the quality and diversity of training data, text-based instructions intrinsically encapsulate task information and support the model's generalization beyond the structure seen during pretraining. To capture high-order relations with task information in a context-aware manner, a novel prompting hypergraph convolution layer is devised to integrate instructions into information propagation in hypergraphs. Extensive experiments conducted on three public datasets verify the superiority of IHP in various scenarios.

Original languageEnglish
Title of host publicationSIGIR 2024 - Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval
PublisherAssociation for Computing Machinery, Inc
Pages501-511
Number of pages11
ISBN (Electronic)9798400704314
DOIs
StatePublished - 11 Jul 2024
Event47th International ACM SIGIR Conference on Research and Development in Information Retrieval, SIGIR 2024 - Washington, United States
Duration: 14 Jul 202418 Jul 2024

Publication series

NameSIGIR 2024 - Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval

Conference

Conference47th International ACM SIGIR Conference on Research and Development in Information Retrieval, SIGIR 2024
Country/TerritoryUnited States
CityWashington
Period14/07/2418/07/24

Keywords

  • graph pretraining
  • hypergraph learning

Fingerprint

Dive into the research topics of 'Instruction-based Hypergraph Pretraining'. Together they form a unique fingerprint.

Cite this