arXiv · 2401.09651
Convex and Bilevel Optimization for Neuro-Symbolic Inference and Learning
Abstract
We leverage convex and bilevel optimization techniques to develop a general gradient-based parameter learning framework for neural-symbolic (NeSy) systems. We demonstrate our framework with NeuPSL, a state-of-the-art NeSy architecture. To achieve this, we propose a smooth primal and dual formulation of NeuPSL inference and show learning gradients are functions of the optimal dual variables. Additionally, we develop a dual block coordinate descent algorithm for the new formulation that naturally exploits warm-starts. This leads to over 100x learning runtime improvements over the current best NeuPSL inference method. Finally, we provide extensive empirical evaluations across 8 datasets covering a range of tasks and demonstrate our learning framework achieves up to a 16% point prediction performance improvement over alternative learning methods.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Charles Dickens, Changyu Gao, Connor Pryor, Stephen Wright, Lise Getoor. 2024-01-17. Convex and Bilevel Optimization for Neuro-Symbolic Inference and Learning. https://arxiv.org/abs/2401.09651
Cite the original work for its findings. Save a collection to share your selection of sources.