English
Related papers

Related papers: Learning Accurate, Efficient, and Interpretable ML…

200 papers

Graph Neural Networks (GNNs) are pivotal in graph-based learning, particularly excelling in node classification. However, their scalability is hindered by the need for multi-hop data during inference, limiting their application in…

Machine Learning · Computer Science 2025-05-30 Ziang Zhou , Zhihao Ding , Jieming Shi , Qing Li , Shiqi Shen

We study the challenging problem for inference tasks on large-scale graph datasets of Graph Neural Networks: huge time and memory consumption, and try to overcome it by reducing reliance on graph structure. Even though distilling graph…

Machine Learning · Computer Science 2024-03-05 Junxian Li , Bin Shi , Erfei Cui , Hua Wei , Qinghua Zheng

Semi-supervised learning on graphs is an important problem in the machine learning area. In recent years, state-of-the-art classification methods based on graph neural networks (GNNs) have shown their superiority over traditional ones such…

Machine Learning · Computer Science 2021-03-05 Cheng Yang , Jiawei Liu , Chuan Shi

Graph Neural Networks (GNNs) have demonstrated their effectiveness in various graph learning tasks, yet their reliance on neighborhood aggregation during inference poses challenges for deployment in latency-sensitive applications, such as…

Machine Learning · Computer Science 2024-12-06 Zehong Wang , Zheyuan Zhang , Chuxu Zhang , Yanfang Ye

Graph Neural Networks (GNNs) have shown exceptional performance in the task of link prediction. Despite their effectiveness, the high latency brought by non-trivial neighborhood data dependency limits GNNs in practical deployments.…

Machine Learning · Computer Science 2023-06-06 Zhichun Guo , William Shiao , Shichang Zhang , Yozen Liu , Nitesh V. Chawla , Neil Shah , Tong Zhao

Graph neural networks (GNNs) have shown remarkable performance on diverse graph mining tasks. Although different GNNs can be unified as the same message passing framework, they learn complementary knowledge from the same graph. Knowledge…

Machine Learning · Computer Science 2023-04-06 Zhichun Guo , Chunhui Zhang , Yujie Fan , Yijun Tian , Chuxu Zhang , Nitesh Chawla

Knowledge distillation (KD) techniques have emerged as a powerful tool for transferring expertise from complex teacher models to lightweight student models, particularly beneficial for deploying high-performance models in…

Machine Learning · Computer Science 2025-10-28 Paul Agbaje , Arkajyoti Mitra , Afia Anjum , Pranali Khose , Ebelechukwu Nwafor , Habeeb Olufowobi

This paper explores the use of knowledge distillation to improve a Multi-Task Deep Neural Network (MT-DNN) (Liu et al., 2019) for learning text representations across multiple natural language understanding tasks. Although ensemble learning…

Computation and Language · Computer Science 2019-04-23 Xiaodong Liu , Pengcheng He , Weizhu Chen , Jianfeng Gao

Recent progress in Graph Neural Networks (GNNs) has greatly enhanced the ability to model complex molecular structures for predicting properties. Nevertheless, molecular data encompasses more than just graph structures, including textual…

Machine Learning · Computer Science 2024-06-04 Junjie Xu , Zongyu Wu , Minhua Lin , Xiang Zhang , Suhang Wang

Graph unlearning has emerged as a pivotal method to delete information from a pre-trained graph neural network (GNN). One may delete nodes, a class of nodes, edges, or a class of edges. An unlearning method enables the GNN model to comply…

Machine Learning · Computer Science 2024-06-11 Yash Sinha , Murari Mandal , Mohan Kankanhalli

Knowledge distillation (KD) has shown great potential for transferring knowledge from a complex teacher model to a simple student model in which the heavy learning task can be accomplished efficiently and without losing too much prediction…

Machine Learning · Computer Science 2023-07-14 Dai Shi , Zhiqi Shao , Yi Guo , Junbin Gao

Graph convolutional networks (GCNs) have been employed as a kind of significant tool on many graph-based applications recently. Inspired by convolutional neural networks (CNNs), GCNs generate the embeddings of nodes by aggregating the…

Machine Learning · Computer Science 2020-11-20 Tao Huang , Yihan Zhang , Jiajing Wu , Junyuan Fang , Zibin Zheng

Knowledge distillation (KD) has shown to be effective to boost the performance of graph neural networks (GNNs), where the typical objective is to distill knowledge from a deeper teacher GNN into a shallower student GNN. However, it is often…

Machine Learning · Computer Science 2023-11-17 Kaituo Feng , Yikun Miao , Changsheng Li , Ye Yuan , Guoren Wang

Hypergraph Neural Networks (HGNNs) have recently attracted much attention and exhibited satisfactory performance due to their superiority in high-order correlation modeling. However, it is noticed that the high-order modeling capability of…

Machine Learning · Computer Science 2024-02-20 Yifan Feng , Yihe Luo , Shihui Ying , Yue Gao

This work studies ensemble learning for graph neural networks (GNNs) under the popular semi-supervised setting. Ensemble learning has shown superiority in improving the accuracy and robustness of traditional machine learning by combining…

Machine Learning · Computer Science 2024-05-07 Xin Zhang , Daochen Zha , Qiaoyu Tan

Graph Neural Networks (GNNs) have demonstrated their effectiveness in various graph learning tasks, yet their reliance on neighborhood aggregation during inference poses challenges for deployment in latency-sensitive applications, such as…

Machine Learning · Computer Science 2024-12-09 Zehong Wang , Zheyuan Zhang , Chuxu Zhang , Yanfang Ye

Graph neural networks (GNNs) have gained considerable attention in recent years for traffic flow prediction due to their ability to learn spatio-temporal pattern representations through a graph-based message-passing framework. Although GNNs…

Machine Learning · Computer Science 2025-03-12 Qianru Zhang , Xinyi Gao , Haixin Wang , Siu-Ming Yiu , Hongzhi Yin

Graph Neural Networks (GNNs) have become a dominant approach to learning graph representations, primarily because of their message-passing mechanisms. However, GNNs typically adopt a fixed aggregator function such as Mean, Max, or Sum…

Machine Learning · Computer Science 2025-07-29 Xuanting Xie , Bingheng Li , Erlin Pan , Zhao Kang , Wenyu Chen

Accurate prediction of molecular properties is essential in drug discovery and related fields. However, existing graph neural networks (GNNs) often struggle to simultaneously capture both local and global molecular structures. In this work,…

Machine Learning · Computer Science 2025-08-26 XiaYu Liu , Chao Fan , Yang Liu , Hou-biao Li

Recent advances in graph neural networks (GNNs) have enabled more comprehensive modeling of molecules and molecular systems, thereby enhancing the precision of molecular property prediction and molecular simulations. Nonetheless, as the…

Machine Learning · Computer Science 2023-10-31 Filip Ekström Kelvinius , Dimitar Georgiev , Artur Petrov Toshev , Johannes Gasteiger