paperbot · PL 论文追踪

全部论文 待读 收藏

P2 Lobster: A GPU-Accelerated Framework for Neurosymbolic Programming

Lobster:面向神经符号编程的 GPU 加速框架
ASPLOS Proceedings of the 31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 12025匹配度 4/10 · 论文属于PL与AI交叉的神经符号编程/编译方向,涉及Datalog、GPU加速与程序推理,与用户对PL+AI的广义兴趣有交集;但未涉及静态分析、LLM/Agent、浮点程序分析或贝叶斯程序分析等核心追踪问题,匹配度有限。引用 5 inbox
Paul Biberstein, Ziyang Li, Joseph Devietti, Mayur Naik
crossref: 2026-09-12dblp: 2026-09-12arxiv: 2026-09-12

AI 速览

一句话:提出 Lobster,将基于 Datalog 的神经符号程序编译到 GPU,实现端到端加速与可扩展推理。

神经符号程序结合深度学习与符号推理,可提升数据效率、可解释性与泛化能力。但现有框架通常将高可扩展的 GPU 神经组件与较慢的 CPU 符号组件割裂结合,难以端到端利用 GPU。Lobster 是一个统一的神经符号学习框架,将基于 Datalog 的通用神经符号语言映射到 GPU 编程范式,并通过编译到新中间语言 APM 实现该映射。APM 的抽象使 Lobster 兼具灵活性,借助 provenance semiring 库支持 GPU 上的离散、概率与可微推理,同时通过新的优化 pass 保证性能。评估覆盖 NLP、图像处理、程序推理、生物信息与规划等 9 个应用;相较 Scallop,Lobster 平均加速 3.9 倍,并可将神经符号方案扩展到此前不可行的任务。

问题
现有神经符号学习框架将可扩展的 GPU 神经组件与较慢的 CPU 符号组件割裂结合,难以端到端利用 GPU。
方法
Lobster 将基于 Datalog 的通用神经符号语言映射到 GPU 编程范式,通过编译到中间语言 APM,并借助 provenance semiring 库和优化 pass 支持离散、概率与可微推理。
评估
在 9 个跨 NLP、图像处理、程序推理、生物信息与规划的应用上,Lobster 相较 Scallop 平均加速 3.9 倍,并能扩展到此前不可行的任务。
原文摘要(Abstract)

Neurosymbolic programs combine deep learning with symbolic reasoning to achieve better data efficiency, interpretability, and generalizability compared to standalone deep learning approaches. However, existing neurosymbolic learning frameworks implement an uneasy marriage between a highly scalable, GPU-accelerated neural component and a slower symbolic component that runs on CPUs. We propose Lobster, a unified framework for harnessing GPUs in an end-to-end manner for neurosymbolic learning. Lobster maps a general neurosymbolic language based on Datalog to the GPU programming paradigm. This mapping is implemented via compilation to a new intermediate language called APM. The extra abstraction provided by APM allows Lobster to be both flexible, supporting discrete, probabilistic, and differentiable modes of reasoning on GPU hardware with a library of provenance semirings, and performant, implementing new optimization passes. We demonstrate that Lobster programs can solve interesting problems spanning the domains of natural language processing, image processing, program reasoning, bioinformatics, and planning. On a suite of 9 applications, Lobster achieves an average speedup of 3.9x over Scallop, a state-of-the-art neurosymbolic framework, and enables scaling of neurosymbolic solutions to previously infeasible tasks.

链接与引用

DOI 原文 · arXiv · PDF(开放获取) · DBLP

BibTeX
@inproceedings{Biberstein0DN26,
  title = {Lobster: A GPU-Accelerated Framework for Neurosymbolic Programming},
  author = {Paul Biberstein and Ziyang Li and Joseph Devietti and Mayur Naik},
  booktitle = {ASPLOS},
  year = {2025},
  doi = {10.1145/3760250.3762232}
}