Table of Contents
Fetching ...

Efficient compilation of expressive problem space specifications to neural network solvers

Matthew L. Daggitt, Wen Kokke, Robert Atkey

TL;DR

The paper tackles the embedding gap in neural network verification by introducing a high-level DSL that lets domain experts express properties in the problem space and a compilation algorithm that translate these specifications into embedding-space queries for neural network solvers. The approach supports arbitrary numbers of networks, nested applications, quantifiers, and conditionals, while enabling user variables to be linearly transformed into the problem space to preserve interpretability. It introduces a robust compilation pipeline with blocking/unblocking, vector-variable handling, and elimination techniques, and demonstrates scalable performance for common query classes even when embedding dimensions $m$ and $n$ exceed $1000$. The work advances practical verification tooling by enabling more expressive problem-space specifications and providing a principled, efficient path to generate solver-ready queries, with future attention to floating-point soundness and abstract-interpretation-based explanations.

Abstract

Recent work has described the presence of the embedding gap in neural network verification. On one side of the gap is a high-level specification about the network's behaviour, written by a domain expert in terms of the interpretable problem space. On the other side are a logically-equivalent set of satisfiability queries, expressed in the uninterpretable embedding space in a form suitable for neural network solvers. In this paper we describe an algorithm for compiling the former to the latter. We explore and overcome complications that arise from targeting neural network solvers as opposed to standard SMT solvers.

Efficient compilation of expressive problem space specifications to neural network solvers

TL;DR

The paper tackles the embedding gap in neural network verification by introducing a high-level DSL that lets domain experts express properties in the problem space and a compilation algorithm that translate these specifications into embedding-space queries for neural network solvers. The approach supports arbitrary numbers of networks, nested applications, quantifiers, and conditionals, while enabling user variables to be linearly transformed into the problem space to preserve interpretability. It introduces a robust compilation pipeline with blocking/unblocking, vector-variable handling, and elimination techniques, and demonstrates scalable performance for common query classes even when embedding dimensions and exceed . The work advances practical verification tooling by enabling more expressive problem-space specifications and providing a principled, efficient path to generate solver-ready queries, with future attention to floating-point soundness and abstract-interpretation-based explanations.

Abstract

Recent work has described the presence of the embedding gap in neural network verification. On one side of the gap is a high-level specification about the network's behaviour, written by a domain expert in terms of the interpretable problem space. On the other side are a logically-equivalent set of satisfiability queries, expressed in the uninterpretable embedding space in a form suitable for neural network solvers. In this paper we describe an algorithm for compiling the former to the latter. We explore and overcome complications that arise from targeting neural network solvers as opposed to standard SMT solvers.
Paper Structure (27 sections, 2 theorems, 2 equations, 2 figures)

This paper contains 27 sections, 2 theorems, 2 equations, 2 figures.

Key Result

lemma thmcounterlemma

If a blocked haskellValue $\:e$ is of type haskellVector or haskellIndex, there exists a set of blocking subexpressions, each one of which is either a i) a haskellNetworkApplication, ii) an haskellIf, iii) a haskellVar of type haskellVector or haskellIndex.

Figures (2)

  • Figure 1: Syntax for a generic neural network solver query language.
  • Figure 2: Syntax for the computational fragment of the Vehicle DSL

Theorems & Definitions (6)

  • definition thmcounterdefinition: Compilable
  • definition thmcounterdefinition: Blocked
  • lemma thmcounterlemma: Blocked vectors and indices
  • proof
  • lemma thmcounterlemma: Non-compilable booleans
  • proof