DeepMETProducer Class Reference

Inheritance diagram for DeepMETProducer:

Public Member Functions
	DeepMETProducer (const edm::ParameterSet &, const tensorflow::SessionCache *)
void	produce (edm::Event &event, const edm::EventSetup &setup) override
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< tensorflow::SessionCache > >
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
static void	globalEndJob (tensorflow::SessionCache *)
static std::unique_ptr< tensorflow::SessionCache >	initializeGlobalCache (const edm::ParameterSet &)

Private Attributes
const bool	ignore_leptons_
tensorflow::Tensor	input_
tensorflow::Tensor	input_cat0_
tensorflow::Tensor	input_cat1_
tensorflow::Tensor	input_cat2_
const unsigned int	max_n_pf_
const float	norm_
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	pf_token_
const tensorflow::Session *	session_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< tensorflow::SessionCache > >
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 14 of file DeepMETProducer.cc.

Constructor & Destructor Documentation

DeepMETProducer()

DeepMETProducer::DeepMETProducer ( const edm::ParameterSet &  cfg, const tensorflow::SessionCache *  cache  )

explicit

Definition at line 38 of file DeepMETProducer.cc.

References input_, input_cat0_, input_cat1_, input_cat2_, max_n_pf_, and l1trig_cff::shape.

    : pf_token_(consumes<std::vector<pat::PackedCandidate> >(cfg.getParameter<edm::InputTag>("pf_src"))),
      norm_(cfg.getParameter<double>("norm_factor")),
      ignore_leptons_(cfg.getParameter<bool>("ignore_leptons")),
      max_n_pf_(cfg.getParameter<unsigned int>("max_n_pf")),
      session_(cache->getSession()) {
  produces<pat::METCollection>();

  // Workaround for missing constructor TensorShape::TensorShape(absl::Slice<long>),
  // the constructor expects Slice<int64> or initializer_list<int64> and is marked explicit
  tensorflow::TensorShape shape;
  shape.AddDim(1);
  shape.AddDim(max_n_pf_);
  shape.AddDim(8);

  tensorflow::TensorShape cat_shape;
  cat_shape.AddDim(1);
  cat_shape.AddDim(max_n_pf_);
  cat_shape.AddDim(1);

  input_ = tensorflow::Tensor(tensorflow::DT_FLOAT, shape);
  input_cat0_ = tensorflow::Tensor(tensorflow::DT_FLOAT, cat_shape);
  input_cat1_ = tensorflow::Tensor(tensorflow::DT_FLOAT, cat_shape);
  input_cat2_ = tensorflow::Tensor(tensorflow::DT_FLOAT, cat_shape);
}

Member Function Documentation

fillDescriptions()

void DeepMETProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 139 of file DeepMETProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, ProducerED_cfi::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                 {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("pf_src", edm::InputTag("packedPFCandidates"));
  desc.add<bool>("ignore_leptons", false);
  desc.add<double>("norm_factor", 50.);
  desc.add<unsigned int>("max_n_pf", 4500);
  desc.add<std::string>("graph_path", "RecoMET/METPUSubtraction/data/models/deepmet/deepmet_v1_2018/model.graphdef");
  descriptions.add("deepMETProducer", desc);
}

globalEndJob()

static void DeepMETProducer::globalEndJob ( tensorflow::SessionCache *  )

inlinestatic

Definition at line 22 of file DeepMETProducer.cc.

{};

initializeGlobalCache()

std::unique_ptr< tensorflow::SessionCache > DeepMETProducer::initializeGlobalCache ( const edm::ParameterSet &  params )

static

Definition at line 133 of file DeepMETProducer.cc.

References edm::FileInPath::fullPath(), HLT_2023v12_cff::graphPath, submitPVValidationJobs::params, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                                          {
  // this method is supposed to create, initialize and return a SessionCache instance
  std::string graphPath = edm::FileInPath(params.getParameter<std::string>("graph_path")).fullPath();
  return std::make_unique<tensorflow::SessionCache>(graphPath);
}

produce()

void DeepMETProducer::produce ( edm::Event &  event, const edm::EventSetup &  setup  )

override

Definition at line 64 of file DeepMETProducer.cc.

References funct::abs(), deepmet_helper::charge_embedding, nano_mu_digi_cff::float, ignore_leptons_, input_, input_cat0_, input_cat1_, input_cat2_, LogDebug, max_n_pf_, eostools::move(), norm_, HLT_2023v12_cff::output_names, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, HiggsValidation_cfi::pdg_id, deepmet_helper::pdg_id_embedding, packedPFCandidateRefMixer_cfi::pf, pf_token_, multPhiCorr_741_25nsDY_cfi::px, multPhiCorr_741_25nsDY_cfi::py, tensorflow::run(), pfClustersFromCombinedCaloHF_cfi::scale, deepmet_helper::scale_and_rm_outlier(), and session_.

                                                                         {
  auto const& pfs = event.get(pf_token_);

  const tensorflow::NamedTensorList input_list = {
      {"input", input_}, {"input_cat0", input_cat0_}, {"input_cat1", input_cat1_}, {"input_cat2", input_cat2_}};

  // Set all inputs to zero
  input_.flat<float>().setZero();
  input_cat0_.flat<float>().setZero();
  input_cat1_.flat<float>().setZero();
  input_cat2_.flat<float>().setZero();

  size_t i_pf = 0;
  float px_leptons = 0.;
  float py_leptons = 0.;
  const float scale = 1. / norm_;
  for (const auto& pf : pfs) {
    if (ignore_leptons_) {
      int pdg_id = std::abs(pf.pdgId());
      if (pdg_id == 11 || pdg_id == 13) {
        px_leptons += pf.px();
        py_leptons += pf.py();
        continue;
      }
    }

    // fill the tensor
    // PF keys [b'PF_dxy', b'PF_dz', b'PF_eta', b'PF_mass', b'PF_pt', b'PF_puppiWeight', b'PF_px', b'PF_py']
    float* ptr = &input_.tensor<float, 3>()(0, i_pf, 0);
    *ptr = pf.dxy();
    *(++ptr) = pf.dz();
    *(++ptr) = pf.eta();
    *(++ptr) = pf.mass();
    *(++ptr) = scale_and_rm_outlier(pf.pt(), scale);
    *(++ptr) = pf.puppiWeight();
    *(++ptr) = scale_and_rm_outlier(pf.px(), scale);
    *(++ptr) = scale_and_rm_outlier(pf.py(), scale);
    input_cat0_.tensor<float, 3>()(0, i_pf, 0) = charge_embedding.at(pf.charge());
    input_cat1_.tensor<float, 3>()(0, i_pf, 0) = pdg_id_embedding.at(pf.pdgId());
    input_cat2_.tensor<float, 3>()(0, i_pf, 0) = pf.fromPV();

    ++i_pf;
    if (i_pf == max_n_pf_) {
      break;  // output a warning?
    }
  }

  std::vector<tensorflow::Tensor> outputs;
  const std::vector<std::string> output_names = {"output/BiasAdd"};

  // run the inference and return met
  tensorflow::run(session_, input_list, output_names, &outputs);

  // The DNN directly estimates the missing px and py, not the recoil
  float px = outputs[0].tensor<float, 2>()(0, 0) * norm_;
  float py = outputs[0].tensor<float, 2>()(0, 1) * norm_;

  px -= px_leptons;
  py -= py_leptons;

  LogDebug("produce") << "<DeepMETProducer::produce>:" << std::endl
                      << " MET from DeepMET Producer is MET_x " << px << " and MET_y " << py << std::endl;

  auto pf_mets = std::make_unique<pat::METCollection>();
  const reco::Candidate::LorentzVector p4(px, py, 0., std::hypot(px, py));
  pf_mets->emplace_back(reco::MET(p4, {}));
  event.put(std::move(pf_mets));
}

Member Data Documentation

ignore_leptons_

const bool DeepMETProducer::ignore_leptons_

private

Definition at line 27 of file DeepMETProducer.cc.

Referenced by produce().

input_

tensorflow::Tensor DeepMETProducer::input_

private

Definition at line 32 of file DeepMETProducer.cc.

Referenced by DeepMETProducer(), and produce().

input_cat0_

tensorflow::Tensor DeepMETProducer::input_cat0_

private

Definition at line 33 of file DeepMETProducer.cc.

Referenced by DeepMETProducer(), and produce().

input_cat1_

tensorflow::Tensor DeepMETProducer::input_cat1_

private

Definition at line 34 of file DeepMETProducer.cc.

Referenced by DeepMETProducer(), and produce().

input_cat2_

tensorflow::Tensor DeepMETProducer::input_cat2_

private

Definition at line 35 of file DeepMETProducer.cc.

Referenced by DeepMETProducer(), and produce().

max_n_pf_

const unsigned int DeepMETProducer::max_n_pf_

private

Definition at line 28 of file DeepMETProducer.cc.

Referenced by DeepMETProducer(), and produce().

norm_

const float DeepMETProducer::norm_

private

Definition at line 26 of file DeepMETProducer.cc.

Referenced by produce().

pf_token_

const edm::EDGetTokenT<std::vector<pat::PackedCandidate> > DeepMETProducer::pf_token_

private

Definition at line 22 of file DeepMETProducer.cc.

Referenced by produce().

session_

const tensorflow::Session* DeepMETProducer::session_

private

Definition at line 30 of file DeepMETProducer.cc.

Referenced by produce().