用二进制分词让大模型读懂粒子碰撞数据,实现高能物理分类任务
Scaling Particle Collision Data Analysis
- 采用二进制分词方法处理数值型实验数据,支持文本与大规模数值数据联合预训练
- 在喷注起源识别任务中性能媲美专用模型,且随数据量增长表现持续提升
- 可作为粒子物理基础模型,适用于大科学、工业制造等多领域数值分析
几十年来,研究者针对不同学科的科学挑战开发了专用模型。近年来,大语言模型(LLMs)展现出处理通用任务的强大能力,但在涉及大规模数值数据分析的实际科学问题上仍面临困难,尤其是在高能物理实验领域。这主要源于传统BPE分词对数值数据的处理效率低下。本文提出一种无任务依赖架构BBT-Neutron,采用二进制分词方法,在混合文本与大规模数值实验数据上进行预训练。我们展示了其在喷注起源识别(JoI)这一高能物理关键分类任务中的应用,该任务旨在区分来自不同夸克或胶子的喷注。结果表明,BBT-Neutron性能可与当前最先进的专用JoI模型相媲美。此外,我们分析了模型性能随数据规模增长的缩放规律,表明其具备作为粒子物理数据基础模型的潜力,并可能扩展至大科学实验、工业制造及空间计算等广泛科学计算场景。项目代码已开源:https://github.com/supersymmetry-technologies/bbt-neutron。
原文摘要 · Abstract (English)
For decades, researchers have developed task-specific models to address scientific challenges across diverse disciplines. Recently, large language models (LLMs) have shown enormous capabilities in handling general tasks; however, these models encounter difficulties in addressing real-world scientific problems, particularly in domains involving large-scale numerical data analysis, such as experimental high energy physics. This limitation is primarily due to BPE tokenization's inefficacy with numerical data. In this paper, we propose a task-agnostic architecture, BBT-Neutron, which employs a binary tokenization method to facilitate pretraining on a mixture of textual and large-scale numerical experimental data. We demonstrate the application of BBT-Neutron to Jet Origin Identification (JoI), a critical categorization challenge in high-energy physics that distinguishes jets originating from various quarks or gluons. Our results indicate that BBT-Neutron achieves comparable performance to state-of-the-art task-specific JoI models. Furthermore, we examine the scaling behavior of BBT-Neutron's performance with increasing data volume, suggesting the potential for BBT-Neutron to serve as a foundational model for particle physics data analysis, with possible extensions to a broad spectrum of scientific computing applications for Big Science experiments, industrial manufacturing and spacial computing. The project code is available at https://github.com/supersymmetry-technologies/bbt-neutron.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。