Paper Overview
Research Area: Machine Learning Authors: Marcos Galván-López, Nijesh Upreti, Hiram Calvo, Carlos Aguilar-Ibáñez, Vaishak Belle Published: 2026-09-15 arXiv: 2609.12247
Abstract
Neuro-symbolic models are usually trained with supervision only on final labels, leaving the intermediate concepts unobserved. Since many concept assignments are consistent with a given label, training can predict labels correctly while recovering the wrong concepts — a failure known as a reasoning shortcut.
Prototypical networks reduce shortcuts by anchoring each concept to a few labeled examples, but existing methods still couple perception and reasoning through a hand-crafted, task-specific differentiable loss that must be redesigned for every task.
Method: Soft-PNet
Soft-PNet removes this task-specific loss:
- Reframes concept grounding as a Metropolis walk over a precomputed cache of feasible symbolic solutions.
- The walk is guided by a prototype distribution built from a single labeled anchor per concept.
- Training uses a KL objective between the prototype-weighted solution cache and the network's concept predictions.
- The objective is fully consistent across all tasks and remains applicable when the solution space cannot be enumerated.
- Matches loss-engineered prototypical networks at both the concept and label levels.
- Recovers concepts missed by soft grounding baselines.
- Requires no loss engineering and achieves shorter training times.
- arXiv: https://arxiv.org/abs/2609.12247
Results
Under scarce supervision settings on MNIST-EvenOdd, visual Sudoku, and Kand-Logic: