摘要
arXiv:2504.19323v1 Announce Type: cross
摘要:神经符号AI(Neuro-Symbolic AI, NSAI)是一种新兴范式,将神经网络与符号推理相结合,以增强AI系统的透明度、推理能力和数据效率。由于其在推理任务和人机协作场景中的出色性能,最近的NSAI系统开始受到关注。尽管在算法方面有所进步,但在现有硬件(例如CPU、GPU、TPU)上执行NSAI任务仍然充满挑战,因为这些硬件的异构计算内核、高内存_intensity_以及独特的内存访问模式。此外,当前的NSAI算法在操作类型和规模上表现出显著差异,这使得它们与现有的机器学习加速器不兼容。这些挑战表明,需要一个适应NSAI工作负载的多功能和灵活的加速框架。在本文中,我们提出NSFlow,这是一种基于FPGA的加速框架,旨在实现NSAI系统在高效率、可扩展性和多功能性上的高效率。NSFlow具有一个设计架构生成器,能够识别工作负载数据依赖关系并创建优化的数据流架构,以及一个可重构阵列,具有灵活的计算单元、可重新组织的内存和混合精度能力。在NSAI工作负载上进行评估,NSFlow在速度上比Jetson TX2提升了31倍,比GPU快2倍以上,比类似TPU的 systolic阵列快8倍以上,比Xilinx DPU快3倍以上。此外,NSFlow还展示了增强的可扩展性,在符号工作负载扩展150倍的情况下,运行时仅增加了4倍。据我们所知,NSFlow是第一个能够实现实时通用NSAI算法加速的框架,展示了下一代认知系统的一种有前途的解决方案。