Saved in:
Bibliographic Details
Main Authors: Kurscheidt, Leander, Masina, Gabriele, Sebastiani, Roberto, Vergari, Antonio
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2602.08681
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912893389242368
author Kurscheidt, Leander
Masina, Gabriele
Sebastiani, Roberto
Vergari, Antonio
author_facet Kurscheidt, Leander
Masina, Gabriele
Sebastiani, Roberto
Vergari, Antonio
contents In many safety-critical settings, probabilistic ML systems have to make predictions subject to algebraic constraints, e.g., predicting the most likely trajectory that does not cross obstacles. These real-world constraints are rarely convex, nor the densities considered are (log-)concave. This makes computing this constrained maximum a posteriori (MAP) prediction efficiently and reliably extremely challenging. In this paper, we first investigate under which conditions we can perform constrained MAP inference over continuous variables exactly and efficiently and devise a scalable message-passing algorithm for this tractable fragment. Then, we devise a general constrained MAP strategy that interleaves partitioning the domain into convex feasible regions with numerical constrained optimization. We evaluate both methods on synthetic and real-world benchmarks, showing our approaches outperform constraint-agnostic baselines, and scale to complex densities intractable for SoTA exact solvers.
format Preprint
id arxiv_https___arxiv_org_abs_2602_08681
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle The Theory and Practice of MAP Inference over Non-Convex Constraints
Kurscheidt, Leander
Masina, Gabriele
Sebastiani, Roberto
Vergari, Antonio
Machine Learning
In many safety-critical settings, probabilistic ML systems have to make predictions subject to algebraic constraints, e.g., predicting the most likely trajectory that does not cross obstacles. These real-world constraints are rarely convex, nor the densities considered are (log-)concave. This makes computing this constrained maximum a posteriori (MAP) prediction efficiently and reliably extremely challenging. In this paper, we first investigate under which conditions we can perform constrained MAP inference over continuous variables exactly and efficiently and devise a scalable message-passing algorithm for this tractable fragment. Then, we devise a general constrained MAP strategy that interleaves partitioning the domain into convex feasible regions with numerical constrained optimization. We evaluate both methods on synthetic and real-world benchmarks, showing our approaches outperform constraint-agnostic baselines, and scale to complex densities intractable for SoTA exact solvers.
title The Theory and Practice of MAP Inference over Non-Convex Constraints
topic Machine Learning
url https://arxiv.org/abs/2602.08681