Self-Supervised Random Forest on Transformed Distribution for Anomaly Detection

Jiabin Liu, Huadong Wang, Hanyuan Hang, Shumin Ma, Xin Shen, Yong Shi

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

Anomaly detection, the task of differentiating abnormal data points from normal ones, presents a significant challenge in the realm of machine learning. Numerous strategies have been proposed to tackle this task, with classification-based methods, specifically those utilizing a self-supervised approach via random affine transformations (RATs), demonstrating remarkable performance on both image and non-image data. However, these methods encounter a notable bottleneck, the overlap of constructed labeled datasets across categories, which hampers the subsequent classifiers’ ability to detect anomalies. Consequently, the creation of an effective data distribution becomes the pivotal factor for success. In this article, we introduce a model called “self-supervised forest (sForest)”, which leverages the random Fourier transform (RFT) and random orthogonal rotations to craft a controlled data distribution. Our model utilizes the RFT to map input data into a new feature space. With this transformed data, we create a self-labeled training dataset using random orthogonal rotations. We theoretically prove that the data distribution formulated by our methodology is more stable compared to one derived from RATs. We then use the self-labeled dataset in a random forest (RF) classifier to distinguish between normal and anomalous data points. Comprehensive experiments conducted on both real and artificial datasets illustrate that sForest outperforms other anomaly detection methods, including distance-based, kernel-based, forest-based, and network-based benchmarks.

Original languageEnglish
Pages (from-to)1-15
Number of pages15
JournalIEEE Transactions on Neural Networks and Learning Systems
DOIs
Publication statusAccepted/In press - 2024

Keywords

  • Anomaly detection
  • Anomaly detection
  • Classification tree analysis
  • Forestry
  • Fourier transforms
  • Random forests
  • Self-supervised learning
  • Task analysis
  • data distribution
  • random Fourier transform (RFT)
  • random forest (RF) classifier
  • random orthogonal rotations
  • self-supervised learning

Fingerprint

Dive into the research topics of 'Self-Supervised Random Forest on Transformed Distribution for Anomaly Detection'. Together they form a unique fingerprint.

Cite this