29 January 2024 to 2 February 2024
CERN
Europe/Zurich timezone

Particle Transformer for tau lepton pair invariant mass reconstruction for the HH→bbtautau CMS analysis

31 Jan 2024, 16:25
5m
61/1-201 - Pas perdus - Not a meeting room - (CERN)

61/1-201 - Pas perdus - Not a meeting room -

CERN

10
Show room on map
Poster 2 ML for analysis : event classification, statistical analysis and inference, including anomaly detection Poster Session

Speaker

Valentina Camagni (Universita & INFN, Milano-Bicocca (IT))

Description

One of the most interesting channels to probe theories beyond the Standard Model at LHC, is the production of a new massive particle, that decays into pairs of Higgs Bosons which, in turn, decay into a pair of b-quarks and a pair of tau leptons. A fundamental discriminant variable to separate HH signal from the backgrounds is the invariant mass of the di-tau system, which can be reconstructed starting from the decay products of each tau lepton. In order to reconstruct the di-tau system in the best possible way, special techniques are needed as the presence of neutrinos from tau decay do not allow for a complete reconstruction of the event. To this end, a Transformer-based architecture (Particle Transformer - ParT) has been implemented to estimate the four-momentum of the neutrinos involved in the decay for a high-resolution reconstruction of the corresponding invariant mass. The regression has been subsequently combined with the classification task for the discrimination of signal events from background ones. ParT can be seen as a graph neural network in which each node consists of a visible particle of the decay’s final state while the links represent the connections between them, modeled through pair-wise features. This last input help to enhance the attention mechanism (core of the Transformer class models), which has the purpose of learning the interactions between particles, and improve the model’s explainability. ParT showed better results with respect to the most common used algorithm in CMS (SVFit), which is, in addition, extremely CPU consuming.

Author

Valentina Camagni (Universita & INFN, Milano-Bicocca (IT))

Co-authors

Pietro Govoni (Universita & INFN, Milano-Bicocca (IT)) Simone Gennai (Universita & INFN, Milano-Bicocca (IT)) Marco Paganoni (Universita & INFN, Milano-Bicocca (IT))

Presentation materials