arXiv:2503.21937v1 Announce Type: cross 摘要：神经符号程序结合了深度学习与符号推理，与单独使用深度学习的方法相比，在数据效率、可解释性和泛化能力方面表现出色。然而，现有的神经符号学习框架将一个高度可扩展、GPU加速的神经组件与一个较慢的在CPU上运行的符号组件以一种勉强的结合方式整合在一起。我们提出了Lobster，一种全栈框架，用于端到端地利用GPU进行神经符号学习。Lobster将基于Datalog的通用神经符号语言映射到GPU编程范式。这种映射通过编译到一种新的中间语言APM来实现。APM提供的额外抽象使得Lobster既能灵活地支持GPU硬件上的离散、概率性和可微推理，同时又高效地实现了新的优化阶段。我们展示了Lobster程序能够解决跨自然语言处理、图像处理、程序推理、生物信息学以及规划领域的问题。在一组8个应用上，Lobster相对于最先进的神经符号框架Scallop实现了平均5.3倍的加速，并使神经符号解决方案能够扩展到以前难以解决的任务。