Few-Shot One-Class Classification via Meta-Learning

Authors

  • Ahmed Frikha Siemens Technology Siemens AI Lab Ludwig Maximilian University of Munich
  • Denis Krompaß Siemens Technology Siemens AI Lab
  • Hans-Georg Köpken Siemens Digital Industries
  • Volker Tresp Siemens Technology Ludwig Maximilian University of Munich

Keywords:

Transfer/Adaptation/Multi-task/Meta/Automated Learning, Anomaly/Outlier Detection, Time-Series/Data Streams, Applications

Abstract

Although few-shot learning and one-class classification (OCC), i.e., learning a binary classifier with data from only one class, have been separately well studied, their intersection remains rather unexplored. Our work addresses the few-shot OCC problem and presents a method to modify the episodic data sampling strategy of the model-agnostic meta-learning (MAML) algorithm to learn a model initialization particularly suited for learning few-shot OCC tasks. This is done by explicitly optimizing for an initialization which only requires few gradient steps with one-class minibatches to yield a performance increase on class-balanced test data. We provide a theoretical analysis that explains why our approach works in the few-shot OCC scenario, while other meta-learning algorithms fail, including the unmodified MAML. Our experiments on eight datasets from the image and time-series domains show that our method leads to better results than classical OCC and few-shot classification approaches, and demonstrate the ability to learn unseen tasks from only few normal class samples. Moreover, we successfully train anomaly detectors for a real-world application on sensor readings recorded during industrial manufacturing of workpieces with a CNC milling machine, by using few normal examples. Finally, we empirically demonstrate that the proposed data sampling technique increases the performance of more recent meta-learning algorithms in few-shot OCC and yields state-of-the-art results in this problem setting.

Downloads

Published

2021-05-18

How to Cite

Frikha, A., Krompaß, D., Köpken, H.-G., & Tresp, V. (2021). Few-Shot One-Class Classification via Meta-Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 35(8), 7448-7456. Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/16913

Issue

Section

AAAI Technical Track on Machine Learning I