Ultra-high-granularity detector simulation with intra-event aware generative adversarial network and self-supervised relational reasoning.

Baran Hashemi, Nikolai Hartmann, Sahand Sharifzadeh, James Kahn, Thomas Kuhr
Author Information
  1. Baran Hashemi: ORIGINS Data Science Lab, Technical University Munich, Munich, Germany. baran.hashemi@origins-cluster.de.
  2. Nikolai Hartmann: Faculty of Physics, Ludwig Maximilians University in Munich, Munich, Germany.
  3. Sahand Sharifzadeh: Faculty of Computer Science, Ludwig Maximilians University in Munich, Munich, Germany.
  4. James Kahn: Helmholtz AI, Karlsruhe, Germany.
  5. Thomas Kuhr: Faculty of Physics, Ludwig Maximilians University in Munich, Munich, Germany. ORCID

Abstract

Simulating high-resolution detector responses is a computationally intensive process that has long been challenging in Particle Physics. Despite the ability of generative models to streamline it, full ultra-high-granularity detector simulation still proves to be difficult as it contains correlated and fine-grained information. To overcome these limitations, we propose Intra-Event Aware Generative Adversarial Network (IEA-GAN). IEA-GAN presents a Transformer-based Relational Reasoning Module that approximates an event in detector simulation, generating contextualized high-resolution full detector responses with a proper relational inductive bias. IEA-GAN also introduces a Self-Supervised intra-event aware loss and Uniformity loss, significantly enhancing sample fidelity and diversity. We demonstrate IEA-GAN's application in generating sensor-dependent images for the ultra-high-granularity Pixel Vertex Detector (PXD), with more than 7.5 M information channels at the Belle II Experiment. Applications of this work span from Foundation Models for high-granularity detector simulation, such as at the HL-LHC (High Luminosity LHC), to simulation-based inference and fine-grained density estimation.

References

  1. Nature. 2016 Sep 14;537(7620):320-7 [PMID: 27629638]
  2. Neural Netw. 2017 Oct;94:255-259 [PMID: 28806718]
  3. IEEE Trans Pattern Anal Mach Intell. 2022 Dec;44(12):8927-8948 [PMID: 34752384]
  4. Curr Opin Struct Biol. 2022 Feb;72:226-236 [PMID: 34963082]
  5. Cancer Res. 1967 Feb;27(2):209-20 [PMID: 6018555]
  6. Arch Comput Methods Eng. 2023;30(4):2761-2775 [PMID: 36713767]
  7. Nat Commun. 2021 May 20;12(1):2985 [PMID: 34016982]
  8. Phys Rev Lett. 2018 Jan 26;120(4):042003 [PMID: 29437460]
  9. Proc Natl Acad Sci U S A. 2021 Apr 13;118(15): [PMID: 33833061]

Grants

  1. EXC 2094/Deutsche Forschungsgemeinschaft (German Research Foundation)

Word Cloud

Created with Highcharts 10.0.0detectorsimulationIEA-GANhigh-resolutionresponsesgenerativefullultra-high-granularityfine-grainedinformationgeneratingrelationalintra-eventawarelossSimulatingcomputationallyintensiveprocesslongchallengingParticlePhysicsDespiteabilitymodelsstreamlinestillprovesdifficultcontainscorrelatedovercomelimitationsproposeIntra-EventAwareGenerativeAdversarialNetworkpresentsTransformer-based RelationalReasoningModuleapproximateseventcontextualizedproperinductivebiasalsointroducesSelf-SupervisedUniformitysignificantlyenhancingsamplefidelitydiversitydemonstrateIEA-GAN'sapplicationsensor-dependentimagesPixelVertexDetectorPXD75MchannelsBelleIIExperimentApplicationsworkspanFoundationModelshigh-granularityHL-LHCHighLuminosityLHCsimulation-basedinferencedensityestimationUltra-high-granularityadversarialnetworkself-supervisedreasoning

Similar Articles

Cited By