Machine Learning toolkit for calorimeter topo-cluster classification and regression using simulated data from the ATLAS experiment.

Feel free to contact joakim.olsson[at]cern.ch if you'd like to contribute!

Machine Learning is awesome :)

Images are created from ESD (Event Summary Data) files using the MLTree Athena package, which generates a root TTree that contains the images as well as some other info. Six images are saved for each cluster, corresponding to the barrels layers of the EM (EMB1, EMB2, EMB3) and HAD (TileBar0, TileBar2, TileBar3) calorimeter. Normalized cell energies are used as pixel values. The image size is 0.4x0.4 in eta-phi space.

The outputs from MLTree can be converted into numpy arrays with mltree2array.py

Train a classifier to determine which type of particle generated the parton showers in the cluster (e.g. electrons vs. charged pions or charged pions vs. neutral pions).

The following models are implemented:

1. Simple fully-connected Neural Network (flattening the images and only using the 512 pixels in the EMB1 layer).
2. Convolutional Neural Networks using only one layer (preserving the shape of the 2D images).
3. A network with multiple images as inputs, and one output (first couple of ConvNets are trained separately, then flattened and concatenated).

Everything is in the TopoClusterClassifier.ipynb notebook, so it is easy to modify and play around with!

TODO

• Implement a network of concatenated ConvNets taking all calorimeter layer images into account.
• Also compare the performance with other ML algorithms; logistic regression, SVD, Naive Bias, Gaussians, etc.

Coming soon...