Skip to content

Latest commit

 

History

History
30 lines (24 loc) · 1.15 KB

README.md

File metadata and controls

30 lines (24 loc) · 1.15 KB

Graph Neural Network Autoencoders Autoencoder for Jets

DOI

Overview

A graph autoencoder (GNNAE) for jets in particle physics implemented in PyTorch, mainly used as a baseline for LGAE

Data

To download data:

  1. Install JetNet:
    pip3 install jetnet; 
    
  2. Run preprocess.py
    python utils/data/preprocess.py \
    --jet-types g q t w z \
    --save-dir "./data"
    

Training

To train the model, run train.py. An example is provided in examples/train.sh.

Architecture

Both the encoder and decoder are built upon the GraphNet architecture implemented in models/graphnet.py, which is a fully connected massage passing neural network. The message passing step of GraphNet is shown in the diagram below. Here, $d$ is any distance function, and EdgeNet and NodeNet are edge and node functions at the $t$-th message passing step, respectively, both of which are MLPs with LeakyReLU activation.