Workshop: Machine Learning and the Physical Sciences

Point Cloud Generation using Transformer Encoders and Normalising Flows

Benno Käch · Dirk Krücker · Isabell Melzer


Data generation based on Machine Learning has become a major research topic in particle physics. This is due to the current Monte Carlo simulation approach being computationally challenging for future colliders, which will have a significantly higher luminosity. The generation of collider data is similar to point cloud generation, but arguably more difficult as there are complex correlations between the points which need to be modelled correctly. A refinement model consisting of normalising flows and transformer encoders is presented. The normalising flow output is corrected by a transformer encoder, which is adversarially trained against another transformer encoder. The model reaches state-of-the-art results with a lightweight model architecture which is stable to train.

Chat is not available.