TrackCore-F: FPGA Accelerated Tracking

Description

TrackCore-F is a methodology and collection of techniques to deploy Machine Learning (ML) models involved in ML-assisted particle track reconstruction algorithms on FPGAs. With TrackCore-F, the focus is the deployment of the ML model element, specifically aiming at the Transformer architecture. As such, FPGA-accelerated inference can boost tracing performance by reducing the computation latency. The small form factor and low power requirements of FPGAs enables on-site installation and paves the way for potential online tracking, as opposed to current post-mortem approaches. Currently, TrackCore-F operates to convert pre-trained models taken from the project TrackFormers as input. However, the techniques to convert and evaluate common Transformer blocks, e.g., encoder layers, are generally applicable.

Development workflow

Achieving deployable syntheses for pre-trained ML models will require going through multiple steps, performing analysis, and involves a variety of tooling. We consider models expressed in PyTorch model format as the starting point. FPGA deployment will also demand considerations and deciding on a full or partial acceleration. Either strategy can be motivated based on different factors, some being use-case dependent. The complete development workflow for a partial acceleration is depicted in Figure 1.

Development flow
Selective ML model slice synthesis using ONNX model graphs

PyTorch model format does not provide granular visibility and individual access to a model's internal computational steps. By converting the model to Open Neural Network Exchange (ONNX) format, a detailed model graph can be extracted. Partitioning the model into separate slices is done in Python. This is followed by coding a computation kernel in HLS, synthesising this kernel, and deployment using AMD Vitis HLS, AMD Vivado HLS, and PYNQ, respectively. Further details can be found in our EuCAIFCon 2025 proceedings paper [1].


Feature set

TBA ...

Category Feature Status
TBA TBA
Features supported by TrackCore-F

Code repository

TBA ...


Authors and acknowledgement

The TrackCore-F project, aiming to achieve FPGA accelerated particle track reconstruction, is a joint effort authored by:


Publications

[1] Arjan Blankestijn, Uraz Odyurt, Amirreza Yousefzadeh. 2025. European AI for Fundamental Physics Conference (EuCAIFCon 2025). "TrackCore-F: Tracking Transformer Synthesis for Low-Latency FPGA Deployment".
DOI: TBA