Skip to content

Add large-radius jet regression (bJR10v00Ext and bJR10v01) scores to PHYS derivations

Andrius Vaitkus requested to merge avaitkus/athena:bJR10v00Ext_bJR10v01 into main

Adds large-radius jet regression model scores to PHYS derivation format.

  • bJR10v00Ext: same architecture as bJR10v00 (from ATL-PHYS-PUB-2024-015), but with an extended training dataset pT range to 2.1TeV.
  • bJR10v01: new model with an updated truth definition, auxiliary track classification task (outputs not exported). Trained on mixed mc20+mc23 dataset.

The model predictions are called bJR10v00Ext_mass,bJR10v00Ext_pt and bJR10v01_mass , bJR10v01_pt , respectively.

Also moved bJR10 scores to AntiKt10UFOCSSKSoftDropBeta100Zcut10JetsCPContent

Note: to read the models properly, I had to add a special case for models that are stored in JetCalibTools, since the naming convention is different, the model name extraction from nnFile is different.

Below are slides outlining both models' performance

2025-02-11-bJR10v01.pdf

Edited by Andrius Vaitkus

Merge request reports