English
Related papers

Related papers: TGL: A General Framework for Temporal GNN Training…

200 papers

Graph neural networks (GNNs) are among the most powerful tools in deep learning. They routinely solve complex problems on unstructured networks, such as node classification, graph classification, or link prediction, with high accuracy.…

Machine Learning · Computer Science 2023-08-21 Maciej Besta , Torsten Hoefler

We present distributed algorithms for training dynamic Graph Neural Networks (GNN) on large scale graphs spanning multi-node, multi-GPU systems. To the best of our knowledge, this is the first scaling study on dynamic GNN. We devise…

Distributed, Parallel, and Cluster Computing · Computer Science 2021-09-17 Venkatesan T. Chakaravarthy , Shivmaran S. Pandian , Saurabh Raje , Yogish Sabharwal , Toyotaro Suzumura , Shashanka Ubaru

Large-scale graph training is a notoriously challenging problem for graph neural networks (GNNs). Due to the nature of evolving graph structures into the training process, vanilla GNNs usually fail to scale up, limited by the GPU memory…

Machine Learning · Computer Science 2023-03-02 Keyu Duan , Zirui Liu , Peihao Wang , Wenqing Zheng , Kaixiong Zhou , Tianlong Chen , Xia Hu , Zhangyang Wang

Temporal heterogeneous networks (THNs) are evolving networks that characterize many real-world applications such as citation and events networks, recommender systems, and knowledge graphs. Although different Graph Neural Networks (GNNs)…

Machine Learning · Computer Science 2026-03-10 Manuel Dileo , Matteo Zignani , Sabrina Gaito

Graph representation learning (also known as network embedding) has been extensively researched with varying levels of granularity, ranging from nodes to graphs. While most prior work in this area focuses on node-level representation,…

Machine Learning · Computer Science 2023-06-05 Lili Wang , Chenghan Huang , Weicheng Ma , Xinyuan Cao , Soroush Vosoughi

In recent years, spatio-temporal graph neural networks (GNNs) have attracted considerable interest in the field of time series analysis, due to their ability to capture, at once, dependencies among variables and across time points. The…

Machine Learning · Computer Science 2025-12-01 Flavio Corradini , Flavio Gerosa , Marco Gori , Carlo Lucheroni , Marco Piangerelli , Martina Zannotti

Graph convolutional networks (GCNs) allow us to learn topologically-aware node embeddings, which can be useful for classification or link prediction. However, they are unable to capture long-range dependencies between nodes without adding…

Machine Learning · Computer Science 2023-08-17 Reza Namazi , Elahe Ghalebi , Sinead Williamson , Hamidreza Mahyar

Training graph neural networks on large datasets has long been a challenge. Traditional approaches include efficiently representing the whole graph in-memory, designing parameter efficient and sampling-based models, and graph partitioning…

Machine Learning · Computer Science 2024-11-19 Dmytro Lopushanskyy , Borun Shi

The quality of graph-structured data is fundamental to the success of modern graph analysis techniques such as Graph Neural Networks (GNNs). However, real-world graph data is often suboptimal, suffering from issues such as noise and…

Machine Learning · Computer Science 2026-05-19 Shen Han , Zhiyao Zhou , Jiawei Chen , Sheng Zhou , Canghong Jin , Hai Lin , Da Zhong Li , Bingde Hu , Can Wang

Future link prediction is a fundamental challenge in various real-world dynamic systems. To address this, numerous temporal graph neural networks (temporal GNNs) and benchmark datasets have been developed. However, these datasets often…

Machine Learning · Computer Science 2025-03-18 Lu Yi , Jie Peng , Yanping Zheng , Fengran Mo , Zhewei Wei , Yuhang Ye , Yue Zixuan , Zengfeng Huang

Time series are the primary data type used to record dynamic system measurements and generated in great volume by both physical sensors and online processes (virtual sensors). Time series analytics is therefore crucial to unlocking the…

Machine Learning · Computer Science 2024-08-12 Ming Jin , Huan Yee Koh , Qingsong Wen , Daniele Zambon , Cesare Alippi , Geoffrey I. Webb , Irwin King , Shirui Pan

Large language models (LLMs) have demonstrated strong performance in natural language generation but remain limited in knowle- dge-intensive tasks due to outdated or incomplete internal knowledge. Retrieval-Augmented Generation (RAG)…

Artificial Intelligence · Computer Science 2025-08-05 Dong Li , Yichen Niu , Ying Ai , Xiang Zou , Biqing Qi , Jianxing Liu

Dynamic Graph Neural Networks (DGNNs) have emerged as the predominant approach for processing dynamic graph-structured data. However, the influence of temporal information on model performance and robustness remains insufficiently explored,…

Machine Learning · Computer Science 2023-11-27 Xiangjian Jiang , Yanyi Pu

Modern machine learning frameworks can train neural networks using multiple nodes in parallel, each computing parameter updates with stochastic gradient descent (SGD) and sharing them asynchronously through a central parameter server. Due…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-03-02 Zhuojin Li , Wumo Yan , Marco Paolieri , Leana Golubchik

Temporal Graph Neural Networks (TGNNs) have gained growing attention for modeling and predicting structures in temporal graphs. However, existing TGNNs primarily focus on pairwise interactions while overlooking higher-order structures that…

Machine Learning · Computer Science 2025-05-22 Jingzhe Liu , Zhigang Hua , Yan Xie , Bingheng Li , Harry Shomer , Yu Song , Kaveh Hassani , Jiliang Tang

Graph convolutional network (GCN) has been successfully applied to many graph-based applications; however, training a large-scale GCN remains challenging. Current SGD-based algorithms suffer from either a high computational cost that…

Machine Learning · Computer Science 2019-08-09 Wei-Lin Chiang , Xuanqing Liu , Si Si , Yang Li , Samy Bengio , Cho-Jui Hsieh

In our recent research, we have developed a framework called GraphSnapShot, which has been proven an useful tool for graph learning acceleration. GraphSnapShot is a framework for fast cache, storage, retrieval and computation for graph…

Machine Learning · Computer Science 2025-01-14 Dong Liu , Roger Waleffe , Meng Jiang , Shivaram Venkataraman

Heterogeneous graph neural networks (HGNNs) can learn from typed and relational graph data more effectively than conventional GNNs. With larger parameter spaces, HGNNs may require more training data, which is often scarce in real-world…

Machine Learning · Computer Science 2023-05-18 Xinyu Fu , Irwin King

Graph neural networks (GNNs) are widely used for learning on graph datasets derived from various real-world scenarios. Learning from extremely large graphs requires distributed training, and mini-batching with sampling is a popular approach…

Temporal knowledge graph question answering (TKGQA) poses a significant challenge task, due to the temporal constraints hidden in questions and the answers sought from dynamic structured knowledge. Although large language models (LLMs) have…

Computation and Language · Computer Science 2024-07-25 Yifu Gao , Linbo Qiao , Zhigang Kan , Zhihua Wen , Yongquan He , Dongsheng Li