8 December 2023
DESY
Europe/Berlin timezone

Ahead-of-time (AOT) compilation of Tensorflow models

8 Dec 2023, 12:12
12m
Flash Seminar Room (DESY)

Flash Seminar Room

DESY

Flash Talk Session II

Speaker

Bogdan Wiederspan (UNI/EXP (Uni Hamburg, Institut fur Experimentalphysik))

Description

In a wide range of high-energy particle physics analyses, machine learning methods have proven as powerful tools to enhance analysis sensitivity.
In the past years, various machine learning applications were also integrated in central CMS workflows, leading to great improvements in reconstruction and object identification efficiencies.

However, the continuation of successful deployments might be limited in the future due to memory and processing time constraints of more advanced models evaluated on central infrastructure.

A novel inference approach for models trained with TensorFlow, based on Ahead-of-time (AOT) compilation is presented. This approach offers a substantial reduction in memory footprints while preserving or even improving computational performance.

This talk outlines strategies and limitations of this novel approach, and presents integration workflow for deploying AOT models in production.

Primary authors

Bogdan Wiederspan (UNI/EXP (Uni Hamburg, Institut fur Experimentalphysik)) Marcel Rieger (UNI/EXP (Uni Hamburg, Institut fur Experimentalphysik))

Presentation materials