Knowledge-Graph Based Multi-Target Deep-Learning Models for Train Anomaly Detection

Abstract

The state-of-art image segmentation algorithms can be applied to accurately localize objects by using deep convolutional neural networks (CNN). In this paper, we consider the anomaly detection problem encountered in a train wheel system. We propose a progressive approach to use a multi-target network to segment each component of the considered system sequentially by decoupling the segmentation and the classification task. Moreover, we use the knowledge graph approach to establish a semantic consistency matrix by quantifying the spatial relationship between various components. We show that by establishing a knowledge graph of the normally operating systems, we are able to identify a faulty component effectively.