Skip to content
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

Update DeepCore training for Run-3 #38169

Closed
wants to merge 1 commit into from
Closed

Update DeepCore training for Run-3 #38169

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
40 changes: 33 additions & 7 deletions RecoTracker/TkSeedGenerator/plugins/DeepCoreSeedGenerator.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,10 @@ class DeepCoreSeedGenerator : public edm::stream::EDProducer<edm::GlobalCache<te
static constexpr int Nover = 3; //Max number of tracks recorded per pixel
static constexpr int Npar = 5; //Number of track parameter

// DeepCore 2.2.1 thresholds delta
double dth[3] = {0.0, 0.15, 0.3};
double dthl[3] = {0.1, 0.05, 0};

private:
void produce(edm::Event&, const edm::EventSetup&) override;

Expand Down Expand Up @@ -210,10 +214,25 @@ void DeepCoreSeedGenerator::produce(edm::Event& iEvent, const edm::EventSetup& i
if (jet.pt() > ptMin_) {
std::set<unsigned long long int> ids;
const reco::Vertex& jetVertex = vertices[0];

// DeepCore 1.0:
/*
std::vector<GlobalVector> splitClustDirSet =
splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 1, inputPixelClusters_);
splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 1, inputPixelClusters_);
bool l2off = (splitClustDirSet.empty());

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// DeepCore 2.1.3 Threshold checks:
*/
std::vector<GlobalVector> splitClustDirSet =
splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 2, inputPixelClusters_);
bool l2off = (splitClustDirSet.empty()); // no adc BPIX2
splitClustDirSet = splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 3, inputPixelClusters_);
bool l134off = (splitClustDirSet.empty()); // no adc BPIX1 or BPIX3 or BPIX4
splitClustDirSet = splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 4, inputPixelClusters_);
l134off = (splitClustDirSet.empty() && l134off);
splitClustDirSet = splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 1, inputPixelClusters_);
l134off = (splitClustDirSet.empty() && l134off);

if (splitClustDirSet.empty()) { //if layer 1 is broken find direcitons on layer 2
splitClustDirSet = splittedClusterDirections(jet, tTopo, pixelCPE, jetVertex, 2, inputPixelClusters_);
}
Expand Down Expand Up @@ -296,7 +315,9 @@ void DeepCoreSeedGenerator::produce(edm::Event& iEvent, const edm::EventSetup& i
for (int i = 0; i < jetDimX; i++) {
for (int j = 0; j < jetDimY; j++) {
for (int o = 0; o < Nover; o++) {
if (seedParamNN.second[i][j][o] > (probThr_ - o * 0.1 - (l2off ? 0.35 : 0))) {
// if (seedParamNN.second[i][j][o] > (probThr_ - o * 0.1 - (l2off ? 0.35 : 0))) { // DeepCore 1.0 Threshold
if (seedParamNN.second[i][j][o] >
(probThr_ + dth[o] + (l2off ? 0.3 : 0) + (l134off ? dthl[o] : 0))) { // DeepCore 2.2.1 Threshold
std::pair<bool, Basic3DVector<float>> interPair =
findIntersection(bigClustDir, (reco::Candidate::Point)jetVertex.position(), globDet);
auto localInter = globDet->specificSurface().toLocal((GlobalPoint)interPair.second);
Expand All @@ -319,8 +340,9 @@ void DeepCoreSeedGenerator::produce(edm::Event& iEvent, const edm::EventSetup& i
double track_theta = 2 * std::atan(std::exp(-track_eta));
double track_phi =
seedParamNN.first[i][j][o][3] * 0.01 + bigClustDir.phi(); //pay attention to this 0.01

double pt = 1. / seedParamNN.first[i][j][o][4];
// DeepCore 1.0 predicts 1/pt instead of pt, while DeepCore 2.2.1 directly predicts pt
// double pt = 1. / seedParamNN.first[i][j][o][4];
double pt = seedParamNN.first[i][j][o][4];
double normdirR = pt / sin(track_theta);

const GlobalVector globSeedDir(
Expand Down Expand Up @@ -552,10 +574,14 @@ void DeepCoreSeedGenerator::fillDescriptions(edm::ConfigurationDescriptions& des
desc.add<std::string>("pixelCPE", "PixelCPEGeneric");
desc.add<edm::FileInPath>(
"weightFile",
edm::FileInPath("RecoTracker/TkSeedGenerator/data/DeepCore/DeepCoreSeedGenerator_TrainedModel_barrel_2017.pb"));
// DeepCore 1.0
// edm::FileInPath("RecoTracker/TkSeedGenerator/data/DeepCore/DeepCoreSeedGenerator_TrainedModel_barrel_2017.pb"));
// DeepCore 2.2.1
edm::FileInPath("RecoTracker/TkSeedGenerator/data/DeepCore/DeepCoreSeedGenerator_TrainedModel_barrel_2023.pb"));
desc.add<std::vector<std::string>>("inputTensorName", {"input_1", "input_2", "input_3"});
desc.add<std::vector<std::string>>("outputTensorName", {"output_node0", "output_node1"});
desc.add<double>("probThr", 0.85);
// desc.add<double>("probThr", 0.85); // DeepCore 1.0
desc.add<double>("probThr", 0.7); // DeepCore 2.2.1 Threshold
descriptions.add("deepCoreSeedGenerator", desc);
}

Expand Down