Official Implementation of "LUNAR: Unifying Local Outlier Detection Methods via Graph Neural Networks"

Last update: Dec 28, 2022

Related tags

Deep Learning LUNAR

Overview

LUNAR

Official Implementation of "LUNAR: Unifying Local Outlier Detection Methods via Graph Neural Networks"

Adam Goodge, Bryan Hooi, Ng See Kiong and Ng Wee Siong (AAAI2022)

Files

main.py
variables.py : hyperparameters
utils.py : functions for loading datasets, pre-processing, graph construction, negative-sampling
LUNAR.py : GNN model and training procedure
requirements.txt : packages for virtualenv
data.zip : files for the HRSS dataset
saved_models.zip : pretrained LUNAR models for HRSS with neighbour count k = 100 and "Mixed" negative sampling

Data

MI-F/MI-V: https://www.kaggle.com/shasun/tool-wear-detection-in-cnc-mill
OPTDIGITS/PENDIGITS/SATELLITE/SHUTTLE/THYROID: http://odds.cs.stonybrook.edu

Experiments

Firstly, extract data.zip

To replicate the results on the HRSS dataset with neighbour count k = 100 and "Mixed" negative sampling scheme

Extract saved_models.zip
Run:

python3 main.py --dataset HRSS --samples MIXED --k 100

To train a new model:

Run:

python3 main.py --dataset HRSS --samples MIXED --k 100 --train_new_model

Citation

TBC

Official Implementation of "LUNAR: Unifying Local Outlier Detection Methods via Graph Neural Networks"

Related tags

Overview

LUNAR

Official Implementation of "LUNAR: Unifying Local Outlier Detection Methods via Graph Neural Networks"

Files

Data

Experiments

Citation

Owner

Adam Goodge

Unrestricted Facial Geometry Reconstruction Using Image-to-Image Translation

A graphical Semi-automatic annotation tool based on labelImg and Yolov5

Official Pytorch implementation of 'GOCor: Bringing Globally Optimized Correspondence Volumes into Your Neural Network' (NeurIPS 2020)

Recursive Bayesian Networks

Intrinsic Image Harmonization

Scripts for training an AI to play the endless runner Subway Surfers using a supervised machine learning approach by imitation and a convolutional neural network (CNN) for image classification

Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks

Supervised forecasting of sequential data in Python.

Incremental Transformer Structure Enhanced Image Inpainting with Masking Positional Encoding (CVPR2022)

STEAL - Learning Semantic Boundaries from Noisy Annotations (CVPR 2019)

DeepSTD: Mining Spatio-temporal Disturbances of Multiple Context Factors for Citywide Traffic Flow Prediction

Official implementation of the network presented in the paper "M4Depth: A motion-based approach for monocular depth estimation on video sequences"

Bolt Online Learning Toolbox

pytorch implementation of trDesign

source code for 'Finding Valid Adjustments under Non-ignorability with Minimal DAG Knowledge' by A. Shah, K. Shanmugam, K. Ahuja

Pytorch Lightning Distributed Accelerators using Ray

Training, generation, and analysis code for Learning Particle Physics by Example: Location-Aware Generative Adversarial Networks for Physics

Code for BMVC2021 "MOS: A Low Latency and Lightweight Framework for Face Detection, Landmark Localization, and Head Pose Estimation"

A transformer model to predict pathogenic mutations

EASY - Ensemble Augmented-Shot Y-shaped Learning: State-Of-The-Art Few-Shot Classification with Simple Ingredients.