Skip to content

Draft: FPGATrackSim: Add FPGATrackSimGNN as GNN Pipeline to FPGATrackSim

Jared Dynes Burleson requested to merge jaburles/athena:FPGATrackSimGNN into main

FPGATrackSim implementation of GNN-based pattern recognition.

This is an ongoing effort to implement each piece of the GNNPatternRecoTool as a configurable option for pattern recognition within FPGATrackSim. Each step of the GNN-based pattern recognition is isolated in it's own tool and it's progress is tracked through the following checklist.

  • FlagConfiguration: Not a tool, but add flag configuration capabilities for when running the python script in athena.
  • GraphHitSelectorTool: Converts FPGATrackSimHit to FPGATrackSimGNNHit (strip SPs with both cluster information).
  • GraphConstructionTool: Builds edges between hits using user-configured tool, default is doublet module map with minmax tolerance.
  • EdgeClassifierTool: Score edges using interaction network graph neural network inferencing.
  • RoadMakerTool: Use ConnectedComponents to build list of FPGATrackSimGNNHits chained together, then find the original FPGATrackSimHits and make a vector as each FPGATrackSimRoads.
  • Add proper header information and detailed comments to match athena-etiquette.
Edited by Jared Dynes Burleson

Merge request reports