Simple yet Effective Node Property Prediction on Edge Streams under Distribution Shifts

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Jongha, Kwon, Taehyung, Moon, Heechan, Shin, Kijung
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913769287843840
author Lee, Jongha
Kwon, Taehyung
Moon, Heechan
Shin, Kijung
author_facet Lee, Jongha
Kwon, Taehyung
Moon, Heechan
Shin, Kijung
contents The problem of predicting node properties (e.g., node classes) in graphs has received significant attention due to its broad range of applications. Graphs from real-world datasets often evolve over time, with newly emerging edges and dynamically changing node properties, posing a significant challenge for this problem. In response, temporal graph neural networks (TGNNs) have been developed to predict dynamic node properties from a stream of emerging edges. However, our analysis reveals that most TGNN-based methods are (a) far less effective without proper node features and, due to their complex model architectures, (b) vulnerable to distribution shifts. In this paper, we propose SPLASH, a simple yet powerful method for predicting node properties on edge streams under distribution shifts. Our key contributions are as follows: (1) we propose feature augmentation methods and an automatic feature selection method for edge streams, which improve the effectiveness of TGNNs, (2) we propose a lightweight MLP-based TGNN architecture that is highly efficient and robust under distribution shifts, and (3) we conduct extensive experiments to evaluate the accuracy, efficiency, generalization, and qualitative performance of the proposed method and its competitors on dynamic node classification, dynamic anomaly detection, and node affinity prediction tasks across seven real-world datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2504_00328
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Simple yet Effective Node Property Prediction on Edge Streams under Distribution Shifts
Lee, Jongha
Kwon, Taehyung
Moon, Heechan
Shin, Kijung
Machine Learning
H.2.8; I.2.6
The problem of predicting node properties (e.g., node classes) in graphs has received significant attention due to its broad range of applications. Graphs from real-world datasets often evolve over time, with newly emerging edges and dynamically changing node properties, posing a significant challenge for this problem. In response, temporal graph neural networks (TGNNs) have been developed to predict dynamic node properties from a stream of emerging edges. However, our analysis reveals that most TGNN-based methods are (a) far less effective without proper node features and, due to their complex model architectures, (b) vulnerable to distribution shifts. In this paper, we propose SPLASH, a simple yet powerful method for predicting node properties on edge streams under distribution shifts. Our key contributions are as follows: (1) we propose feature augmentation methods and an automatic feature selection method for edge streams, which improve the effectiveness of TGNNs, (2) we propose a lightweight MLP-based TGNN architecture that is highly efficient and robust under distribution shifts, and (3) we conduct extensive experiments to evaluate the accuracy, efficiency, generalization, and qualitative performance of the proposed method and its competitors on dynamic node classification, dynamic anomaly detection, and node affinity prediction tasks across seven real-world datasets.
title Simple yet Effective Node Property Prediction on Edge Streams under Distribution Shifts
topic Machine Learning
H.2.8; I.2.6
url https://arxiv.org/abs/2504.00328