Preventing Failures by Dataset Shift Detection in Safety-Critical Graph Applications

• Main Authors: Hoseung Song, Jayaraman J. Thiagarajan, Bhavya Kailkhura
• Format: Article
• Language: English
• Published: Frontiers Media S.A. 2021-05-01
• Series: Frontiers in Artificial Intelligence
• Subjects: graph learning; dataset shift; safety; two-sample testing; random graph models
• Online Access: https://www.frontiersin.org/articles/10.3389/frai.2021.589632/full

Dataset shift refers to the problem where the input data distribution may change over time (e.g., between training and test stages). Since this can be a critical bottleneck in several safety-critical applications such as healthcare, drug-discovery, etc., dataset shift detection has become an important research issue in machine learning. Though several existing efforts have focused on image/video data, applications with graph-structured data have not received sufficient attention. Therefore, in this paper, we investigate the problem of detecting shifts in graph structured data through the lens of statistical hypothesis testing. Specifically, we propose a practical two-sample test based approach for shift detection in large-scale graph structured data. Our approach is very flexible in that it is suitable for both undirected and directed graphs, and eliminates the need for equal sample sizes. Using empirical studies, we demonstrate the effectiveness of the proposed test in detecting dataset shifts. We also corroborate these findings using real-world datasets, characterized by directed graphs and a large number of nodes.