Back to Search Start Over

A Fully Test-time Training Framework for Semi-supervised Node Classification on Out-of-Distribution Graphs.

Authors :
Zhang, Jiaxin
Wang, Yiqi
Yang, Xihong
Zhu, En
Source :
ACM Transactions on Knowledge Discovery from Data; Aug2024, Vol. 18 Issue 7, p1-19, 19p
Publication Year :
2024

Abstract

Graph neural networks (GNNs) have shown great potential in representation learning for various graph tasks. However, the distribution shift between the training and test sets poses a challenge to the efficiency of GNNs. To address this challenge, HomoTTT proposes a fully test-time training framework for GNNs to enhance the model's generalization capabilities for node classification tasks. Specifically, our proposed HomoTTT designs a homophily-based and parameter-free graph contrastive learning task with adaptive augmentation to guide the model's adaptation during the test-time training, allowing the model to adapt for specific target data. In the inference stage, HomoTTT proposes to integrate the original GNN model and the adapted model after TTT using a homophily-based model selection method, which prevents potential performance degradation caused by unconstrained model adaptation. Extensive experimental results on six benchmark datasets demonstrate the effectiveness of our proposed framework. Additionally, the exploratory study further validates the rationality of the homophily-based graph contrastive learning task with adaptive augmentation and the homophily-based model selection designed in HomoTTT. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
15564681
Volume :
18
Issue :
7
Database :
Complementary Index
Journal :
ACM Transactions on Knowledge Discovery from Data
Publication Type :
Academic Journal
Accession number :
178006326
Full Text :
https://doi.org/10.1145/3649507