b-Jet charge ID using DNN

Overview

On this page:

Usefull Links

• Framework used to produce the ntuples: CMSDeepFlavour/DeepNTuples
• Framework used to train the DNN: DL4Jets/DeepJet

Production of inputs

DeepNTuples

The following code used to produce frat ntuples used for traning of the neural net

install the code  Hide 

cmsrel CMSSW_10_0_1
cd CMSSW_10_0_1/src/
cmsenv
git cms-init
git clone https://github.com/CMSDeepFlavour/DeepNTuples
cd DeepNTuples
git checkout 94X
# Add JetToolBox
git submodule init
git submodule update
#compile
scram b -j 4

• Initial setup:

cd CMSSW_10_0_1/src
cmsenv

• to run local tests execute:

file=/store/mc/PhaseIFall16MiniAOD/TT_TuneCUETP8M2T4_13TeV-powheg-pythia8/MINIAODSIM/PhaseIFall16PUFlat20to50_PhaseIFall16_81X_upgrade2017_realistic_v26-v1/50000/12893E5C-DFE3-E611-88C2-D4AE526A0DAE.root
cmsRun DeepNtuplizer/production/DeepNtuplizer.py inputFiles=$file maxEvents=2000 skipEvents=1000 outputFile=output1

Full list of files can be found here. If file doesn't load copy it locally.

copy file  Hide 

xrdcp root://cms-xrd-global.cern.ch/$file .
#OR
site=T1_IT_CNAF_MSS
lfn=$file
pfl=`curl -ks "https://cmsweb.cern.ch/phedex/datasvc/perl/prod/lfn2pfn?node=${site}&lfn=${lfn}&protocol=srmv2" | grep PFN | cut -d "'" -f4`
env -i X509_USER_PROXY=/tmp/x509up_u58751 gfal-copy -n 1 $pfl "file:///`pwd`/miniAOD.root"
file="file:///`pwd`/miniAOD.root"

Before running the code, add extra variables (such as jet_charge) to the ntuple_JetInfo.cc file:

modified DeepNtuplizer.cc file  Hide 

   // jet charge implementation:
   gen_parton_pdgid_ = 0; gen_hadron_pdgid_ = 0; gen_hadron_pt_ = 0;
   if(jet.genParton()) gen_parton_pdgid_ = int(jet.genParton()->pdgId());
   if(jet.jetFlavourInfo().getbHadrons().size()) {
      gen_hadron_pdgid_ = jet.jetFlavourInfo().getbHadrons().at(0)->pdgId();
      gen_hadron_pt_ = jet.jetFlavourInfo().getbHadrons().at(0)->pt();
   }

Charge of the jet can be obtained from the charge of the b-quark or b-hadron. The implementation is the following (introduced in the ntuple_JetInfo.cc file):

if(jet.genParton()) gen_parton_pdgid_ = int(jet.genParton()->pdgId());
if(jet.jetFlavourInfo().getbHadrons().size()) gen_hadron_pdgid_ = jet.jetFlavourInfo().getbHadrons().at(0)->pdgId();

slim ntuples:

ntuple_JetInfo::fillBranches()

• Skim only leptonic b-jets: if(isPhysLeptonicB_ && isPhysLeptonicB_C_) returnval=false
• Skim only hadronic b-jets: if(isPhysB_==0 && isPhysBB_==0) returnval=false;

Training

DeepJetCore

Singularity

env -i PATH=/usr/bin/ SINGULARITY_CACHEDIR="/tmp/$(whoami)/singularity" singularity run -B /home -B /eos -B /afs --bind /etc/krb5.conf:/etc/krb5.conf --bind /proc/fs/openafs/afs_ioctl:/proc/fs/openafs/afs_ioctl --bind /usr/vice/etc:/usr/vice/etc /eos/home-j/jkiesele/singularity/images/deepjetcore3_latest.sif

Build a new subpackage:

createSubpackage.py BJetChargeID --data
cd BJetChargeID; source env.sh

Once you have the data in the example_data folder, convert it to format that will be used by the ML algorithm (modify modules/datastructure/TrainData_example.py to setup correct inputs and labels):

modified TrainDataexample.py file  Hide 

   // labels
   truth = np.expand_dims((urfile.array("gen_parton_pdgid")/5+1)/2, axis=1)
 
        //inputs:
        lep_pt=[]; lep_charge=[]
   pt=charge=0
        muons_pt = urfile.array("muons_pt")
        electrons_pt = urfile.array("electrons_pt")
        muons_charge = urfile.array("muons_charge")
        electrons_charge = urfile.array("electrons_charge")   
   for i in range(urfile.numentries):
          nmu=len(muons_pt[i]); nel=len(electrons_pt[i])
          if nmu:
              pt=muons_pt[i][0]
              charge=muons_charge[i][0]
              if nel and (pt < electrons_pt[i][0]):
                  pt=electrons_pt[i][0]
                  charge=electrons_charge[i][0]
          elif nel:
              pt=electrons_pt[i][0]
              charge=electrons_charge[i][0] 
          lep_pt.append(pt)
          lep_charge.append(charge)
        
        feature_array = np.concatenate([
                                np.expand_dims(lep_pt, axis=1), 
                                np.expand_dims(lep_charge, axis=1)
                                ],axis=1)

convertFromSource.py -i example_data/train_files.txt -o train_data -c TrainData_example

The training is done using the following script (modify the Train/training_example.py script to set your own model):

python3 Train/training_example.py train_data/dataCollection.djcdc result_train --gpu none

The results will appear in result_train, to predict output for a new sample run:

predict.py result_train/KERAS_check_best_model.h5 result_train/trainsamples.djcdc example_data/test_files.txt result_test --gpu none

The output file in result_test/ will contain predicted values

To compare predicted to truth you can do the following:

root -l example_data/out_bjet_lep_10.root
tree->AddFriend("tree","result_test/pred_out_bjet_lep_10.root")
tree->Draw("gen_parton_pdgid : tree.prob_isA","","box")

-- MichaelPitt - 2020-01-23