Skip to content

Add the Scouting Glo-ParT inference facility #47801

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
cmsbuild merged 1 commit into from
Apr 21, 2025
Merged

Add the Scouting Glo-ParT inference facility #47801

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions PhysicsTools/NanoAOD/python/custom_run3scouting_cff.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@
scoutingPFCandidate, # translate to reco::PFCandidate, used as input
scoutingFatPFJetRecluster, # jet clustering
scoutingFatPFJetReclusterParticleNetJetTagInfos, scoutingFatPFJetReclusterParticleNetJetTags, # jet tagging
scoutingFatPFJetReclusterGlobalParticleTransformerJetTagInfos, scoutingFatPFJetReclusterGlobalParticleTransformerJetTags, # jet tagging with Global Particle Transformer
scoutingFatPFJetReclusterSoftDrop, scoutingFatPFJetReclusterSoftDropMass, # softdrop mass
scoutingFatPFJetReclusterParticleNetJetTagInfos, scoutingFatPFJetReclusterParticleNetMassRegressionJetTags, # regressed mass
scoutingFatPFJetReclusterEcfNbeta1, scoutingFatPFJetReclusterNjettiness, # substructure variables
Expand Down
58 changes: 58 additions & 0 deletions PhysicsTools/NanoAOD/python/run3scouting_cff.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -546,6 +546,47 @@
debugMode = cms.untracked.bool(False),
)

scoutingFatPFJetReclusterGlobalParticleTransformerJetTagInfos = cms.EDProducer("DeepBoostedJetTagInfoProducer",
jet_radius = cms.double(0.8),
min_jet_pt = cms.double(50),
max_jet_eta = cms.double(2.5),
min_pt_for_track_properties = cms.double(0.95),
min_pt_for_pfcandidates = cms.double(0.1),
use_puppiP4 = cms.bool(False),
include_neutrals = cms.bool(True),
sort_by_sip2dsig = cms.bool(False),
min_puppi_wgt = cms.double(-1.0),
flip_ip_sign = cms.bool(False),
sip3dSigMax = cms.double(-1.0),
use_hlt_features = cms.bool(False),
pf_candidates = cms.InputTag("scoutingPFCandidate"),
jets = cms.InputTag("scoutingFatPFJetRecluster"),
puppi_value_map = cms.InputTag(""),
use_scouting_features = cms.bool(True),
normchi2_value_map = cms.InputTag("scoutingPFCandidate", "normchi2"),
dz_value_map = cms.InputTag("scoutingPFCandidate", "dz"),
dxy_value_map = cms.InputTag("scoutingPFCandidate", "dxy"),
dzsig_value_map = cms.InputTag("scoutingPFCandidate", "dzsig"),
dxysig_value_map = cms.InputTag("scoutingPFCandidate", "dxysig"),
lostInnerHits_value_map = cms.InputTag("scoutingPFCandidate", "lostInnerHits"),
quality_value_map = cms.InputTag("scoutingPFCandidate", "quality"),
trkPt_value_map = cms.InputTag("scoutingPFCandidate", "trkPt"),
trkEta_value_map = cms.InputTag("scoutingPFCandidate", "trkEta"),
trkPhi_value_map = cms.InputTag("scoutingPFCandidate", "trkPhi"),
)

scoutingFatPFJetReclusterGlobalParticleTransformerJetTags = cms.EDProducer("BoostedJetONNXJetTagsProducer",
jets = cms.InputTag("scoutingFatPFJetRecluster"),
produceValueMap = cms.untracked.bool(True),
src = cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTagInfos"),
preprocess_json = cms.string("RecoBTag/Combined/data/Run3Scouting/GlobalParticleTransformerAK8/General/V00/preprocess.json"),
model_path = cms.FileInPath("RecoBTag/Combined/data/Run3Scouting/GlobalParticleTransformerAK8/General/V00/global-part_2024.onnx"),
flav_names = cms.vstring([
"probQCD", "probXbb", "probXcc", "probXss", "probXqq", "probXbs", "probXgg", "probXee", "probXmm", "probXtauhtaue", "probXtauhtaum", "probXtauhtauh", "probXbc", "probXcs", "probXud", "massCorrGeneric", "massCorrGenericX2p", "massCorrGenericW2p", "massCorrResonance"
]),
debugMode = cms.untracked.bool(False),
)

# AK8 jet softdrop mass

scoutingFatPFJetReclusterSoftDrop = ak4PFJets.clone(
Expand Down Expand Up @@ -622,10 +663,27 @@
particleNet_prob_Hbb = ExtVar(cms.InputTag("scoutingFatPFJetReclusterParticleNetJetTags:probHbb"), float, doc="ParticleNet probability of Hbb", precision=10),
particleNet_prob_Hcc = ExtVar(cms.InputTag("scoutingFatPFJetReclusterParticleNetJetTags:probHcc"), float, doc="ParticleNet probability of Hcc", precision=10),
particleNet_prob_Hqq = ExtVar(cms.InputTag("scoutingFatPFJetReclusterParticleNetJetTags:probHqq"), float, doc="ParticleNet probability of Hqq", precision=10),
scoutGlobalParT_prob_QCD = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probQCD"), float, doc="Mass-decorrelated Scouting GlobalParT QCD score", precision=10),
scoutGlobalParT_prob_Xbb = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXbb"), float, doc="Mass-decorrelated Scouting GlobalParT X->bb score", precision=10),
scoutGlobalParT_prob_Xcc = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXcc"), float, doc="Mass-decorrelated Scouting GlobalParT X->cc score", precision=10),
scoutGlobalParT_prob_Xss = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXss"), float, doc="Mass-decorrelated Scouting GlobalParT X->ss score", precision=10),
scoutGlobalParT_prob_Xqq = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXqq"), float, doc="Mass-decorrelated Scouting GlobalParT X->qq score", precision=10),
scoutGlobalParT_prob_Xbc = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXbc"), float, doc="Mass-decorrelated Scouting GlobalParT X->bc score", precision=10),
scoutGlobalParT_prob_Xbs = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXbs"), float, doc="Mass-decorrelated Scouting GlobalParT X->bs score", precision=10),
scoutGlobalParT_prob_Xcs = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXcs"), float, doc="Mass-decorrelated Scouting GlobalParT X->cs score", precision=10),
scoutGlobalParT_prob_Xud = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXud"), float, doc="Mass-decorrelated Scouting GlobalParT X->ud score", precision=10),
scoutGlobalParT_prob_Xgg = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXgg"), float, doc="Mass-decorrelated Scouting GlobalParT X->gg score", precision=10),
scoutGlobalParT_prob_Xtauhtaue = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXtauhtaue"), float, doc="Mass-decorrelated Scouting GlobalParT X->tauhtaue score", precision=10),
scoutGlobalParT_prob_Xtauhtaum = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXtauhtaum"), float, doc="Mass-decorrelated Scouting GlobalParT X->tauhtaum score", precision=10),
scoutGlobalParT_prob_Xtauhtauh = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:probXtauhtauh"), float, doc="Mass-decorrelated Scouting GlobalParT X->tauhtauh score", precision=10),
# softdrop mass
msoftdrop = ExtVar(cms.InputTag("scoutingFatPFJetReclusterSoftDropMass"), float, doc="Softdrop mass", precision=10),
# regressed mass
particleNet_mass = ExtVar(cms.InputTag("scoutingFatPFJetReclusterParticleNetMassRegressionJetTags:mass"), float, doc="ParticleNet regressed mass", precision=10),
scoutGlobalParT_massCorrGeneric = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:massCorrGeneric"), float, doc="Mass-decorrelated Scouting GlobalParT mass regression corrector with respect to the original jet mass, optimised for generic jet cases. Use (massCorrGeneric * mass) to get the regressed mass", precision=10),
scoutGlobalParT_massCorrGenericX2p = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:massCorrGenericX2p"), float, doc="Mass-decorrelated Scouting GlobalParT mass regression corrector with respect to the original jet mass, optimised for generic X2p jet cases. Use (massCorrGenericX2p * mass) to get the regressed mass", precision=10),
scoutGlobalParT_massCorrGenericW2p = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:massCorrGenericW2p"), float, doc="Mass-decorrelated Scouting GlobalParT mass regression corrector with respect to the original jet mass, optimised for generic W jet cases. Use (massCorrGenericW2p * mass) to get the regressed mass", precision=10),
scoutGlobalParT_massCorrResonance = ExtVar(cms.InputTag("scoutingFatPFJetReclusterGlobalParticleTransformerJetTags:massCorrResonance"), float, doc="Scouting GlobalParT mass regression corrector with respect to the original jet mass, optimised for resonance jets. Use (massCorrResonance * mass) to get the regressed mass", precision=10),
# substructure variables
n2b1 = ExtVar(cms.InputTag("scoutingFatPFJetReclusterEcfNbeta1:ecfN2"), float, doc="N2 with beta=1", precision=10),
n3b1 = ExtVar(cms.InputTag("scoutingFatPFJetReclusterEcfNbeta1:ecfN3"), float, doc="N3 with beta=1", precision=10),
Expand Down
44 changes: 36 additions & 8 deletions RecoBTag/FeatureTools/plugins/DeepBoostedJetTagInfoProducer.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -169,14 +169,38 @@ const std::vector<std::string> DeepBoostedJetTagInfoProducer::particle_features_
"jet_pfcand_puppiw",
"pfcand_mask"};

const std::vector<std::string> DeepBoostedJetTagInfoProducer::particle_features_scouting_{
"pfcand_quality", "pfcand_charge", "pfcand_isEl", "pfcand_isMu",
"pfcand_isChargedHad", "pfcand_isGamma", "pfcand_isNeutralHad", "pfcand_phirel",
"pfcand_etarel", "pfcand_deltaR", "pfcand_abseta", "pfcand_ptrel_log",
"pfcand_erel_log", "pfcand_pt_log", "pfcand_normchi2", "pfcand_dz",
"pfcand_dxy", "pfcand_dxysig", "pfcand_btagEtaRel", "pfcand_btagPtRatio",
"pfcand_btagPParRatio", "pfcand_mask", "pfcand_pt_log_nopuppi", "pfcand_dzsig",
"pfcand_e_log_nopuppi", "pfcand_ptrel", "pfcand_erel", "pfcand_lostInnerHits"};
const std::vector<std::string> DeepBoostedJetTagInfoProducer::particle_features_scouting_{"pfcand_px",
"pfcand_py",
"pfcand_pz",
"pfcand_energy",
"pfcand_quality",
"pfcand_charge",
"pfcand_isEl",
"pfcand_isMu",
"pfcand_isChargedHad",
"pfcand_isGamma",
"pfcand_isNeutralHad",
"pfcand_phirel",
"pfcand_etarel",
"pfcand_deltaR",
"pfcand_abseta",
"pfcand_ptrel_log",
"pfcand_erel_log",
"pfcand_pt_log",
"pfcand_normchi2",
"pfcand_dz",
"pfcand_dxy",
"pfcand_dxysig",
"pfcand_btagEtaRel",
"pfcand_btagPtRatio",
"pfcand_btagPParRatio",
"pfcand_mask",
"pfcand_pt_log_nopuppi",
"pfcand_dzsig",
"pfcand_e_log_nopuppi",
"pfcand_ptrel",
"pfcand_erel",
"pfcand_lostInnerHits"};

const std::vector<std::string> DeepBoostedJetTagInfoProducer::sv_features_{"sv_mask",
"sv_ptrel",
Expand Down Expand Up @@ -616,6 +640,10 @@ void DeepBoostedJetTagInfoProducer::fillParticleFeatures(DeepBoostedJetFeatures
auto candP3 = use_puppiP4_ ? puppi_wgt_cache.at(cand.key()) * cand->momentum() : cand->momentum();

if (use_scouting_features_) {
fts.fill("pfcand_px", candP4.px());
fts.fill("pfcand_py", candP4.py());
fts.fill("pfcand_pz", candP4.pz());
fts.fill("pfcand_energy", candP4.energy());
fts.fill("pfcand_charge", reco_cand->charge());
fts.fill("pfcand_isEl", std::abs(reco_cand->pdgId()) == 11);
fts.fill("pfcand_isMu", std::abs(reco_cand->pdgId()) == 13);
Expand Down