Supervised autoencoder variants for end to end anomaly detection

Author: M. Lübbering, M. Gebauer, R. Ramamurthy, R. Sifa, C. Bauckhage
Journal: ICPR 2021: Pattern Recognition. ICPR International Workshops and Challenges
Year: 2021

Citation information

M. Lübbering, M. Gebauer, R. Ramamurthy, R. Sifa, C. Bauckhage,
ICPR 2021: Pattern Recognition. ICPR International Workshops and Challenges,
2021,
566–581,
Springer, Cham,
https://doi.org/10.1007/978-3-030-68790-8_44

Despite the success of deep learning in various domains such as natural language processing, speech recognition, and computer vision, learning from a limited amount of samples and generalizing to unseen data still pose challenges. Notably, in the tasks of outlier detection and imbalanced dataset classification, the label of interest is either scarce or its distribution is skewed, causing aggravated generalization problems. In this work, we pursue the direction of multi-task learning, specifically the idea of using supervised autoencoders (SAE), which allows us to combine unsupervised and supervised objectives in an end to end fashion. We extend this approach by introducing an adversarial supervised objective to enrich the representations which are learned for the classification task. We conduct thorough experiments on a broad range of tasks, including outlier detection, novelty detection, and imbalanced classification, and study the efficacy of our method against standard baselines using autoencoders. Our work empirically shows that the SAE methods outperform one class autoencoders, adversarially trained autoencoders and multi layer perceptrons in terms of AUPR score comparison. Additionally, our analysis of the obtained representations suggests that the adversarial reconstruction loss functions enforce the encodings to separate into class-specific clusters, which was not observed for non-adversarial reconstruction loss functions.