Unsupervised Anomaly Detection via Deep Metric Learning with End-to-End Optimization

• Main Authors: Yilmaz, Selim F., Kozat, Suleyman S.
• Format: Article in Journal/Newspaper
• Language: unknown
• Published: arXiv 2020
• Subjects: Machine Learning cs.LG; Machine Learning stat.ML; FOS Computer and information sciences; DML
• Online Access: https://dx.doi.org/10.48550/arxiv.2005.05865; https://arxiv.org/abs/2005.05865

We investigate unsupervised anomaly detection for high-dimensional data and introduce a deep metric learning (DML) based framework. In particular, we learn a distance metric through a deep neural network. Through this metric, we project the data into the metric space that better separates the anomalies from the normal data and reduces the effect of the curse of dimensionality for high-dimensional data. We present a novel data distillation method through self-supervision to remedy the conventional practice of assuming all data as normal. We also employ the hard mining technique from the DML literature. We show these components improve the performance of our model and significantly reduce the running time. Through an extensive set of experiments on the 14 real-world datasets, our method demonstrates significant performance gains compared to the state-of-the-art unsupervised anomaly detection methods, e.g., an absolute improvement between 4.44% and 11.74% on the average over the 14 datasets. Furthermore, we share the source code of our method on Github to facilitate further research. : 11 pages, 3 figures