English
Related papers

Related papers: GETS: Ensemble Temperature Scaling for Calibration…

200 papers

Graph Neural Networks (GNNs) have emerged as powerful tools for learning over graph-structured data, yet recent studies have shown that their performance gains are beginning to plateau. In many cases, well-established models such as GCN and…

Machine Learning · Computer Science 2026-02-13 Mohit Meena , Yash Punjabi , Abhishek A , Vishal Sharma , Mahesh Chandran

Recent works demonstrate that early layers in a neural network contain useful information for prediction. Inspired by this, we show that extending temperature scaling across all layers improves both calibration and accuracy. We call this…

Machine Learning · Computer Science 2022-11-21 Amr Khalifa , Michael C. Mozer , Hanie Sedghi , Behnam Neyshabur , Ibrahim Alabdulmohsin

In this paper, we study the post-hoc calibration of modern neural networks, a problem that has drawn a lot of attention in recent years. Many calibration methods of varying complexity have been proposed for the task, but there is no…

Machine Learning · Computer Science 2022-08-02 Sergio A. Balanya , Juan Maroñas , Daniel Ramos

Graph Neural Networks (GNNs) are powerful tools for learning from graph-structured data, but their scalability is increasingly strained by the size of real-world graphs in domains like recommender systems, fraud detection, and molecular…

Machine Learning · Computer Science 2026-05-22 Mridul Gupta , Samyak Jain , Vansh Ramani , Hariprasad Kodamana , Sayan Ranu

Graph Neural Networks (GNNs) have emerged as powerful tools for various graph mining tasks, yet existing scalable solutions often struggle to balance execution efficiency with prediction accuracy. These difficulties stem from iterative…

Machine Learning · Computer Science 2026-04-02 Xu Cheng , Liang Yao , Feng He , Yukuo Cen , Yufei He , Chenhui Zhang , Wenzheng Feng , Hongyun Cai , Jie Tang

Graph Neural Networks (GNNs) have shown remarkable merit in performing various learning-based tasks in complex networks. The superior performance of GNNs often correlates with the availability and quality of node-level features in the input…

Social and Information Networks · Computer Science 2023-10-20 Anwar Said , Mudassir Shabbir , Tyler Derr , Waseem Abbas , Xenofon Koutsoukos

Node classifiers are required to comprehensively reduce prediction errors, training resources, and inference latency in the industry. However, most graph neural networks (GNN) concentrate only on one or two of them. The compromised aspects…

Machine Learning · Computer Science 2023-06-01 Yi Luo , Guangchun Luo , Ke Qin , Aiguo Chen

\Graph similarity computation is an essential task in many real-world graph-related applications such as retrieving the similar drugs given a query chemical compound or finding the user's potential friends from the social network database.…

Machine Learning · Computer Science 2024-12-18 Jingjing Wang , Hongjie Zhu , Haoran Xie , Fu Lee Wang , Xiaoliang Xu , Yuxiang Wang

Graph Neural Networks (GNNs) are powerful deep learning models to generate node embeddings on graphs. When applying deep GNNs on large graphs, it is still challenging to perform training in an efficient and scalable way. We propose a novel…

Machine Learning · Computer Science 2020-10-08 Hanqing Zeng , Hongkuan Zhou , Ajitesh Srivastava , Rajgopal Kannan , Viktor Prasanna

In this research, we developed a graph-based framework to represent various aspects of optimal thermal management system design, with the aim of rapidly and efficiently identifying optimal design candidates. Initially, the graph-based…

Systems and Control · Electrical Eng. & Systems 2023-11-28 Saeid Bayat , Nastaran Shahmansouri , Satya RT Peddada , Alex Tessier , Adrian Butscher , James T Allison

Network modeling is a critical component for building self-driving Software-Defined Networks, particularly to find optimal routing schemes that meet the goals set by administrators. However, existing modeling techniques do not meet the…

Networking and Internet Architecture · Computer Science 2021-06-15 Krzysztof Rusek , José Suárez-Varela , Albert Mestres , Pere Barlet-Ros , Albert Cabellos-Aparicio

Graph Neural Networks (GNNs) have experienced rapid advancements in recent years due to their ability to learn meaningful representations from graph data structures. However, in most real-world settings, such as financial transaction…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-30 Pranjal Naman , Yogesh Simmhan

Graph neural networks (GNNs) have been widely applied in safety-critical applications, such as financial and medical networks, in which compromised predictions may cause catastrophic consequences. While existing research on GNN robustness…

Machine Learning · Computer Science 2025-07-28 Wencheng Zou , Nan Wu

We consider the graph similarity computation (GSC) task based on graph edit distance (GED) estimation. State-of-the-art methods treat GSC as a learning-based prediction task using Graph Neural Networks (GNNs). To capture fine-grained…

Machine Learning · Computer Science 2024-06-24 Wei Zhuo , Guang Tan

Graph Neural Networks (GNNs) have made significant advances on several fundamental inference tasks. As a result, there is a surge of interest in using these models for making potentially important decisions in high-regret applications.…

Machine Learning · Computer Science 2020-02-26 Kaidi Xu , Sijia Liu , Pin-Yu Chen , Mengshu Sun , Caiwen Ding , Bhavya Kailkhura , Xue Lin

Graph Neural Networks (GNNs) have demonstrated impressive performance across diverse graph-based tasks by leveraging message passing to capture complex node relationships. However, on large-scale real-world graphs, GNNs face two major…

Machine Learning · Computer Science 2026-03-10 Xiang Li , Jianpeng Qi , Haobing Liu , Yuan Cao , Guoqing Chao , Zhongying Zhao , Junyu Dong , Xinwang Liu , Yanwei Yu

Matrix completion is a classic problem underlying recommender systems. It is traditionally tackled with matrix factorization. Recently, deep learning based methods, especially graph neural networks, have made impressive progress on this…

Information Retrieval · Computer Science 2021-03-01 Tieyun Qian , Yile Liang , Qing Li

Graph Neural Networks (GNNs) are playing an increasingly important role in the efficient operation and security of computing systems, with applications in workload scheduling, anomaly detection, and resource management. However, their…

Machine Learning · Computer Science 2025-03-27 Haci Ismail Aslan , Philipp Wiesner , Ping Xiong , Odej Kao

Neural network calibration is an essential task in deep learning to ensure consistency between the confidence of model prediction and the true correctness likelihood. In this paper, we propose a new post-processing calibration method called…

Machine Learning · Computer Science 2024-07-26 Yung-Chen Tang , Pin-Yu Chen , Tsung-Yi Ho

Climate change is increasing the occurrence of extreme precipitation events, threatening infrastructure, agriculture, and public safety. Ensemble prediction systems provide probabilistic forecasts but exhibit biases and difficulties in…

Machine Learning · Computer Science 2025-04-09 Christopher Bülte , Sohir Maskey , Philipp Scholl , Jonas von Berg , Gitta Kutyniok