Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper enhances the NeurASP framework by implementing vectorization, batch processing, and caching techniques to significantly improve its computational efficiency. The authors demonstrate that these optimizations lead to speedups of multiple orders of magnitude, enabling NeurASP to tackle more complex tasks that were previously infeasible due to computational constraints. A new dataset featuring challenging tasks involving playing cards is introduced to validate the effectiveness of the improved learning function in NeurASP.
Speeding up NeurASP by orders of magnitude unlocks its potential for tackling complex tasks that were previously too computationally intensive.
Neurosymbolic AI combines neural networks with symbolic programs to create robust and explainable predictions. One such framework is NeurASP, which trains a neural network to predict concepts and reasons over them using rules written in answer set programming (ASP) to solve downstream tasks. Crucially, labels are only provided for the downstream prediction produced by the symbolic rules, not for the latent concepts themselves.Backpropagation through the non-differentiable ASP component requires expensive probability and gradient calculations, which has hindered scalability to more sophisticated tasks.In this paper, we address the current limitations of NeurASP by improving its computational performance through vectorization, batch processing and caching of intermediate computations during training. We compare computation speeds between the original and our new implementation of NeurASP and report speedups of multiple orders of magnitude for larger tasks. To this end, we propose a new dataset of difficult tasks involving playing cards, which we use to test the capabilities of NeurASP's enhanced learning function.