Publication Type
Conference Proceeding Article
Version
publishedVersion
Publication Date
6-2022
Abstract
One-class classification (OCC) aims to learn an effective data description to enclose all normal training samples and detect anomalies based on the deviation from the data description. Current state-of-the-art OCC models learn a compact normality description by hyper-sphere minimisation, but they often suffer from overfitting the training data, especially when the training set is small or contaminated with anomalous samples. To address this issue, we introduce the interpolated Gaussian descriptor (IGD) method, a novel OCC model that learns a one-class Gaussian anomaly classifier trained with adversarially interpolated training samples. The Gaussian anomaly classifier differentiates the training samples based on their distance to the Gaussian centre and the standard deviation of these distances, offering the model a discriminability w.r.t. the given samples during training. The adversarial interpolation is enforced to consistently learn a smooth Gaussian descriptor, even when the training data is small or contaminated with anomalous samples. This enables our model to learn the data description based on the representative normal samples rather than fringe or anomalous samples, resulting in significantly improved normality description. In extensive experiments on diverse popular benchmarks, including MNIST, Fashion MNIST, CIFAR10, MVTec AD and two medical datasets, IGD achieves better detection accuracy than current state-of-the-art models. IGD also shows better robustness in problems with small or contaminated training sets.
Discipline
Artificial Intelligence and Robotics
Research Areas
Intelligent Systems and Optimization
Publication
Proceedings of the 36th AAAI Conference on Artificial Intelligence, Virtual Conference, 2022 February 22- March 1
Volume
36
First Page
383
Last Page
392
ISBN
9781577358763
Identifier
10.1609/aaai.v36i1.19915
Publisher
AAAI
City or Country
Virtual Conference
Citation
CHEN, Yuanhong; TIAN, Yu; PANG, Guansong; and CARNEIRO, Gustavo.
Deep one-class classification via interpolated Gaussian descriptor. (2022). Proceedings of the 36th AAAI Conference on Artificial Intelligence, Virtual Conference, 2022 February 22- March 1. 36, 383-392.
Available at: https://ink.library.smu.edu.sg/sis_research/7034
Creative Commons License
This work is licensed under a Creative Commons Attribution-NonCommercial-No Derivative Works 4.0 International License.
Additional URL
http://doi.org/10.1609/aaai.v36i1.19915