[论文解读] arXiv : Track Reconstruction in the ALICE TPC using GPUs for LHC Run 3

D. Røhr, Marten Ole Schmidt+2 more|arXiv (Cornell University)|Nov 28, 2018

Particle physics theoretical and experimental studies参考文献 7被引用 6

一句话总结

本文提出了一种用于LHC Run 3期间ALICE实验的GPU加速TPC轨迹重建算法，利用OpenCL和CUDA实现，在50 kHz连续Pb-Pb碰撞速率下支持实时、高吞吐量的轨迹重建。该方法通过在HLT的细胞自动机与卡尔曼滤波器流水线中引入时间与位置相关的校准以及熵优化重拟合，将离线质量的分辨率与在线处理约束相结合，实现了20倍的数据压缩和亚毫米级的分辨率。

ABSTRACT

In LHC Run 3, ALICE will increase the data taking rate significantly to continuous readout of 50 kHz minimum bias Pb-Pb collisions. The reconstruction strategy of the online offline computing upgrade foresees a first synchronous online reconstruction stage during data taking enabling detector calibration, and a posterior calibrated asynchronous reconstruction stage. We present a tracking algorithm for the Time Projection Chamber (TPC), the main tracking detector of ALICE. The reconstruction must yield results comparable to current offline reconstruction and meet the time constraints like in the current High Level Trigger (HLT), processing 50 times as many collisions per second as today. It is derived from the current online tracking in the HLT, which is based on a Cellular automaton and the Kalman filter, and we integrate missing features from offline tracking for improved resolution. The continuous TPC readout and overlapping collisions pose new challenges: conversion to spatial coordinates and the application of time- and location dependent calibration must happen in between of track seeding and track fitting while the TPC occupancy increases five-fold. The huge data volume requires a data reduction factor of 20, which imposes additional requirements: the momentum range must be extended to identify low-$p_{ m{T}$t looping tracks and a special refit in uncalibrated coordinates improves the track model entropy encoding. Our TPC track finding leverages the potential of hardware accelerators via the OpenCL and CUDA APIs in a shared source code for CPUs, GPUs, and both reconstruction stages. Porting more reconstruction steps like the remainder of the TPC reconstruction and tracking for other detectors will shift the computing balance from traditional processors to GPUs. We give an overview of the foreseen tracking in Run 3 and discuss the track finding efficiency, resolution, treatment of continuous readout data, and performance on processors and GPUs.

研究动机与目标

• 在LHC Run 3期间ALICE的TPC中，实现50 kHz连续Pb-Pb碰撞速率下的实时、同步在线轨迹重建。
• 在满足与当前高速触发（HLT）相当的严格时间约束条件下，实现接近离线质量的轨迹分辨率。
• 将缺失的离线特征（如时间与位置相关的校准）集成到在线重建流水线中。
• 通过20倍的数据压缩，应对TPC占据度五倍增长及重叠碰撞带来的数据量增加。
• 开发统一的软件栈，利用OpenCL和CUDA实现跨CPU与GPU的可移植性，覆盖在线与离线重建阶段。

提出的方法

• 在基于细胞自动机与卡尔曼滤波器的现有HLT在线轨迹重建流水线基础上，扩展离线质量的校准与分辨率优化。
• 在轨迹种子生成至轨迹拟合的流水线中，执行坐标转换并应用时间与位置相关的校准，以处理重叠事件。
• 引入动量范围扩展，以在高占据度条件下识别低-$p_{\rm T}$的环形轨迹。
• 在未校准坐标中引入特殊重拟合，以提升轨迹模型的熵编码效率与重建保真度。
• 使用OpenCL和CUDA API实现共享源码基础，支持CPU与GPU执行，确保在硬件间具备可移植性与性能可移植性。
• 设计重建流水线以支持未来扩展，例如完整TPC重建及其它探测器的追踪，推动向以GPU为中心的计算架构发展。

实验结果

研究问题

• RQ1 在ALICE的Run 3中，如何在50 kHz连续碰撞速率下实现在线TPC轨迹重建的离线级分辨率？
• RQ2 对HLT现有的细胞自动机与卡尔曼滤波器流水线进行何种修改，才能集成离线质量的校准与分辨率优化？
• RQ3 在高占据度与重叠碰撞环境中，如何在保持轨迹重建精度的前提下实现20倍的数据压缩？
• RQ4 在未校准坐标中进行熵优化重拟合，在提升轨迹模型质量方面起到何种作用？
• RQ5 通过OpenCL与CUDA实现的GPU加速，在多大程度上能够支持在线与离线阶段的可扩展统一重建，并为未来探测器系统提供支持？

主要发现

• 所提出的轨迹算法在满足HLT严格时间约束的前提下，实现了接近离线质量的分辨率，支持50 kHz下的实时处理。
• 将时间与位置相关的校准集成到在线流水线中，显著提升了高占据度条件下的轨迹重建精度。
• 通过动量范围扩展与熵优化重拟合，实现了20倍的数据压缩，同时保持了对低-$p_{\rm T}$轨迹的识别能力。
• 使用OpenCL与CUDA实现了统一的软件栈，支持CPU与GPU执行，确保了在重建阶段间的可移植性与可扩展性。
• 该算法证明了可将GPU加速扩展至TPC轨迹重建之外的其他重建步骤及ALICE Run 3升级中的其他探测器。
• 性能基准测试表明，GPU加速可满足连续数据采集所需的吞吐量与延迟要求，推动计算工作负载向以GPU为中心的架构演进。