Dynamic Reduction Network for photon energy regression in miniAOD using the SonicTriton service

Configuration/ProcessModifiers/python/photonDRN_cff.py

@@ -0,0 +1,5 @@
+import FWCore.ParameterSet.Config as cms
+
+#modifier to enable DRN energy regression for photons
+
+photonDRN = cms.Modifier()

Configuration/PyReleaseValidation/README.md

@@ -53,6 +53,7 @@ The offsets currently in use are:
 * 0.17: Run-3 deep core seeding for JetCore iteration
 * 0.21: Production-like sequence
 * 0.24: 0 Tesla (Run-2, Run-3)
+* 0.31: Photon energy corrections with DRN architecture
 * 0.61: `phase2_ecal_devel` era
 * 0.91: Track DNN modifier
 * 0.97: Premixing stage1

Configuration/PyReleaseValidation/python/relval_2017.py

@@ -18,6 +18,7 @@
 #        (TTbar trackingOnly, trackingRun2, trackingOnlyRun2, trackingLowPU, pixelTrackingOnly)
 #        (TTbar PU with JME NanoAOD, disable for now due to Run-3 Nano-Prompt preparation)
 #   2018 (ele guns 10, 35, 1000; pho guns 10, 35; mu guns 1, 10, 100, 1000, QCD 3TeV, QCD Flat)
+#        (pho guns 10, 35 with photonDRN enabled)
 #   2018 (ZMM, TTbar, ZEE, MinBias, TTbar PU, ZEE PU, TTbar design)
 #        (TTbar trackingOnly, pixelTrackingOnly)
 #        (HE collapse: TTbar, TTbar PU, TTbar design)
@@ -47,6 +48,7 @@
            10024.1,10024.2,10024.3,10024.4,10024.5,
            #10224.15,
            10801.0,10802.0,10803.0,10804.0,10805.0,10806.0,10807.0,10808.0,10809.0,10859.0,10871.0,
+           10804.31, 10805.31,
            10842.0,10824.0,10825.0,10826.0,10823.0,11024.0,11025.0,11224.0,
            10824.1,10824.5,
            10824.6,11024.6,11224.6,

Configuration/PyReleaseValidation/python/upgradeWorkflowComponents.py

@@ -490,6 +490,32 @@ def condition(self, fragment, stepList, key, hasHarvest):
     '--procModifiers': 'mlpf'
 }
 
+# photonDRN workflows
+class UpgradeWorkflow_photonDRN(UpgradeWorkflow):
+    def setup_(self, step, stepName, stepDict, k, properties):
+        if 'Reco' in step:
+            stepDict[stepName][k] = merge([self.step3, stepDict[step][k]])
+    def condition(self, fragment, stepList, key, hasHarvest):
+        return '2018' in key and "SingleGamma" in fragment
+
+upgradeWFs['photonDRN'] = UpgradeWorkflow_photonDRN(
+    steps = [
+        'Reco',
+        'RecoNano',
+        'RecoFakeHLT'
+    ],
+    PU = [
+        'Reco',
+        'RecoNano',
+        'RecoFakeHLT'
+    ],
+    suffix = '_photonDRN',
+    offset = 0.31,
+)
+upgradeWFs['photonDRN'].step3 = {
+    '--procModifiers': 'enableSonicTriton,photonDRN'
+}
+
 # Patatrack workflows:
 #   - 2018 conditions, TTbar
 #   - 2018 conditions, Z->mumu,

PhysicsTools/PatAlgos/python/slimming/patElectronDRNCorrector_cfi.py

@@ -0,0 +1,18 @@
+import FWCore.ParameterSet.Config as cms
+
+patElectronsDRN = cms.EDProducer("PatElectronDRNCorrectionProducer",
+    particleSource = cms.InputTag("selectedPatElectrons"),
+    rhoName = cms.InputTag("fixedGridRhoFastjetAll"),
+    reducedEcalRecHitsEB = cms.InputTag("reducedEcalRecHitsEB"),
+    reducedEcalRecHitsEE = cms.InputTag("reducedEcalRecHitsEE"),
+    reducedEcalRecHitsES = cms.InputTag("reducedEcalRecHitsES"),
+
+    Client = cms.PSet(
+      mode = cms.string("Async"),
+      modelName = cms.string("electronObjectEnsemble"),
+      modelConfigPath = cms.FileInPath("RecoEgamma/EgammaElectronProducers/data/models/electronObjectEnsemble/config.pbtxt"),
+      allowedTries = cms.untracked.uint32(1),
+      timeout = cms.untracked.uint32(10),
+      useSharedMemory = cms.untracked.bool(False),
+    )
+)

PhysicsTools/PatAlgos/python/slimming/patPhotonDRNCorrector_cfi.py

@@ -0,0 +1,18 @@
+import FWCore.ParameterSet.Config as cms
+
+patPhotonsDRN = cms.EDProducer("PatPhotonDRNCorrectionProducer",
+    particleSource = cms.InputTag("selectedPatPhotons"),
+    rhoName = cms.InputTag("fixedGridRhoFastjetAll"),
+    reducedEcalRecHitsEB = cms.InputTag("reducedEcalRecHitsEB"),
+    reducedEcalRecHitsEE = cms.InputTag("reducedEcalRecHitsEE"),
+    reducedEcalRecHitsES = cms.InputTag("reducedEcalRecHitsES"),
+
+    Client = cms.PSet(
+      mode = cms.string("Async"),
+      modelName = cms.string("photonObjectEnsemble"),
+      modelConfigPath = cms.FileInPath("RecoEgamma/EgammaPhotonProducers/data/models/photonObjectEnsemble/config.pbtxt"),
+      allowedTries = cms.untracked.uint32(1),
+      timeout = cms.untracked.uint32(10),
+      useSharedMemory = cms.untracked.bool(False),
+    )
+)

PhysicsTools/PatAlgos/python/slimming/slimming_cff.py

@@ -102,3 +102,7 @@
 _phase2_timing_slimmingTask = cms.Task(slimmingTask.copy(),
                                        offlineSlimmedPrimaryVertices4D)
 phase2_timing.toReplaceWith(slimmingTask,_phase2_timing_slimmingTask)
+
+from PhysicsTools.PatAlgos.slimming.patPhotonDRNCorrector_cfi import patPhotonsDRN
+from Configuration.ProcessModifiers.photonDRN_cff import photonDRN
+photonDRN.toReplaceWith(slimmingTask, cms.Task(slimmingTask.copy(), patPhotonsDRN))

RecoEgamma/EgammaTools/BuildFile.xml

@@ -17,6 +17,7 @@
 <use name="RecoEcal/EgammaCoreTools"/>
 <use name="DataFormats/ParticleFlowReco"/>
 <use name="PhysicsTools/TensorFlow" />
+<use name="HeterogeneousCore/SonicTriton"/>
 <export>
   <lib name="1"/>
 </export>