Skip to content

Classifying, auto-encoding and reverse-engineering QUBO matrices

License

Notifications You must be signed in to change notification settings

instance01/qubo-nn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

QUBO - NN

qubo heat map

9 problems and their respective QUBO matrices.

QUBO matrices are used to describe an optimization problem as a matrix such that a Quantum Annealer (such as a D-Wave QA) can solve it.

Now, these matrices are quite an interesting construct.. Thus, a few questions arise:

  • Is it possible to classify the problem class based on the QUBO matrix?
  • Is it possible to reverse-engineer the problem parameters that led to a QUBO matrix?

Let's find out.

pip install qubo-nn

Project Structure

File Purpose
datasets/ Contains generated datasets.
models/ Contains trained models.
nn/ Contains neural network models.
plots/ Contains plotting scripts and generated plots.
problems/ Contains generators and evaluators for specific problems such as 3SAT or TSP.
runs/ Contains tensorboard logging files.
config.py Configuration (json) handling.
data.py LMDB data handling.
main.py Main entry point.
pipeline.py End to end training and testing of NNs on QUBO matrices.
simulations.json All experiments and configurations.

Problems (20) implemented so far:

  • Number Partitioning
  • Maximum Cut
  • Minimum Vertex Cover
  • Set Packing
  • Maximum 2-SAT
  • Set Partitioning
  • Graph Coloring
  • Quadratic Assignment
  • Quadratic Knapsack
  • Maximum 3-SAT
  • Travelling Salesman (TSP)
  • Graph Isomorphism
  • Sub-Graph Isomorphism
  • Maximum Clique
  • Exact Cover
  • Binary Integer Linear Programming
  • Maximum Independent Set
  • Minimum Maximum Matching
  • Set Cover
  • Knapsack with Integer Weights

Setup

pip install qubo-nn

OR

pip3 install -r requirements.txt
pip3 install -e .

Using

Classification / Reverse regression

usage: main.py [-h] [-t TYPE] [--eval] [--gendata] [--train] [-c CFG_ID] [-m [MODEL]] [-n [NRUNS]]

optional arguments:
  -h, --help            show this help message and exit
  -t TYPE, --type TYPE  Type (classify, reverse)
  --eval
  --gendata
  --train
  -c CFG_ID, --cfg_id CFG_ID
                        cfg_id
  -m [MODEL], --model [MODEL]
  -n [NRUNS], --nruns [NRUNS]

Examples for classification:

python3 -m qubo_nn.main -t classify -c 2 --train
python3 -m qubo_nn.main -t classify -c 2 --eval -m models/21-02-16_20\:28\:42-9893713-instances-MacBook-Pro.local-2 

Examples for reverse regression:

python3 -m qubo_nn.main -t reverse -c tsp1 --gendata
python3 -m qubo_nn.main -t reverse -c tsp1 --train -n 1

Generating QUBOs for arbitrary problems

This is an example on how to create a MaxCut instance and generate a QUBO matrix for it:

>>> graph = networkx.Graph([(1, 2), (1, 3), (2, 4), (3, 4), (4, 5), (3, 5)])
>>> problem = MaxCut(graph)
>>> matrix = problem.gen_qubo_matrix()
[
    [2, -1, -1, 0, 0],
    [-1, 2, 0, -1, 0],
    [-1, 0, 3, -1, -1],
    [0, -1, -1, 3, -1],
    [0, 0, -1, -1, 2]
]

The list of problems can be found in qubo_nn/problems/__init__.py. Also:

>>> from qubo_nn.problems import PROBLEM_REGISTRY
>>> PROBLEM_REGISTRY
{
    'NP': <class 'qubo_nn.problems.number_partitioning.NumberPartitioning'>,
    'MC': <class 'qubo_nn.problems.max_cut.MaxCut'>,
    'MVC': <class 'qubo_nn.problems.minimum_vertex_cover.MinimumVertexCover'>,
    'SP': <class 'qubo_nn.problems.set_packing.SetPacking'>,
    'M2SAT': <class 'qubo_nn.problems.max2sat.Max2SAT'>,
    'SPP': <class 'qubo_nn.problems.set_partitioning.SetPartitioning'>,
    'GC': <class 'qubo_nn.problems.graph_coloring.GraphColoring'>,
    'QA': <class 'qubo_nn.problems.quadratic_assignment.QuadraticAssignment'>,
    'QK': <class 'qubo_nn.problems.quadratic_knapsack.QuadraticKnapsack'>,
    'M3SAT': <class 'qubo_nn.problems.max3sat.Max3SAT'>,
    'TSP': <class 'qubo_nn.problems.tsp.TSP'>,
    'GI': <class 'qubo_nn.problems.graph_isomorphism.GraphIsomorphism'>,
    'SGI': <class 'qubo_nn.problems.subgraph_isomorphism.SubGraphIsomorphism'>,
    'MCQ': <class 'qubo_nn.problems.max_clique.MaxClique'>,
    'EC': <class 'qubo_nn.problems.exact_cover.ExactCover'>,
    'BIP': <class 'qubo_nn.problems.binary_integer_linear_programming.BinaryIntegerLinearProgramming'>,
    'MIS': <class 'qubo_nn.problems.max_independent_set.MaxIndependentSet'>,
    'MMM': <class 'qubo_nn.problems.minimum_maximum_matching.MinimumMaximumMatching'>,
    'SC': <class 'qubo_nn.problems.set_cover.SetCover'>,
    'KIW': <class 'qubo_nn.problems.knapsack_integer_weights.KnapsackIntegerWeights'>
    ...
}

Results

The pipeline of interest is as follows.

Reverse-engineering pipeline/architecture.

Given some QUBO matrix that was generated using a set of problem parameters, we first classify the problem in step a and then predict the parameters in step b.

Classification

Using parameter configuration 100_genX (see simulations.json), the average total misclassification rate over 20 models goes to near zero. The figure includes the 95% confidence interval. Scrambling QUBOs leads to a nearly similar effect. Note that this is using a generalized dataset, i.e. the dataset consists of not just 64x64 QUBO matrices for each problem, but also smaller sizes such as 32x32. The smaller sizes are zero-padded to the biggest supported size, which most of the time is 64x64 and in rare cases goes up to 144x144 (for Quadratic Assignment).

Avg total misclassification rate

The t-SNE plot for this experiment is shown below.

t-SNE

Reverse regression

This is preliminary. Some of the problems are easily learned by a neural network regressor. Each line represents 10 models and includes the 95% confidence interval.

Reversal regression losses over multiple problems

Reversal regression R**2 over multiple problems

Reversibility

This shows whether we can deduce the parameters that led to a QUBO matrix, given we predicted the problem beforehand. A lot of the graph based problems are easily reversable since the graph structure is kept intact in the QUBO matrix. Thus we can recreate the graph and other input parameters given a GraphColoring QUBO matrix.

This is still WIP - needs testing. These are hypotheses.

Reversing some problems like Quadratic Knapsack might be possible - an algorithm is an idea, but one could also make their life easy and try fitting a NN model to it.

Problem Reversibility Comment
Graph Coloring + Adjacency matrix found in QUBO.
Maximum 2-SAT ? Very complex to learn, but possible? C.f. m2sat_to_bip.py in contrib.
Maximum 3-SAT ? Very complex to learn, but possible?
Maximum Cut + Adjacency matrix found in QUBO.
Minimum Vertex Cover + Adjacency matrix found in QUBO.
Number Partitioning + Easy, create equation system from the upper triangular part of the matrix (triu).
Quadratic Assignment + Over-determined linear system of equations -> solvable. P does not act as salt. A bit complex to learn.
Quadratic Knapsack - Budgets can be deduced easily (Find argmin in first row. This column contains all the budgets.). P acts as a salt -> thus not reversible.
Set Packing - Multiple problem instances lead to the same QUBO.
Set Partitioning - Multiple problem instances lead to the same QUBO.
Travelling Salesman + Find a quadrant with non-zero entries (w/ an identical diagonal), transpose, the entries are the distance matrix. Norm result to between 0 and 1.
Graph Isomorphism + Adjacency matrix found in QUBO.
Sub-Graph Isomorphism + Adjacency matrix found in QUBO.
Maximum Clique + Adjacency matrix found in QUBO.
Exact Cover ? Not investigated.
Binary Integer Linear Programming ? Not investigated.
Maximum Independent Set ? Not investigated.
Minimum Maximum Matching ? Not investigated.
Set Cover ? Not investigated.
Knapsack with Integer Weights ? Not investigated.

Redundancy of QUBOs with AutoEncoders

The figure below shows that there are major differences between problem classes in terms of their overall redundancy.

Redundacy of QUBos with AutoEncoders, R**2

Contributing

Pull requests are very welcome. Before submitting one, run all tests with ./test.sh and make sure nothing is broken.

References

Glover, Fred, Gary Kochenberger, and Yu Du. "A tutorial on formulating and using qubo models." arXiv preprint arXiv:1811.11538 (2018).
Michael J. Dinneen, "Maximum 3-SAT as QUBO" https://canvas.auckland.ac.nz/courses/14782/files/574983/download?verifier=1xqRikUjTEBwm8PnObD8YVmKdeEhZ9Ui8axW8HwP&wrap=1
Lucas, Andrew. "Ising formulation of many NP-problems. Frontiers in Physics" (2014)
Cristian S. Calude, Michael J. Dinneen and Richard Hua. "QUBO Formulations for the Graph Isomorphism Problem and Related Problems" (2017)

Related Work / Blog Posts

Hadamard Gate Transformation for 3 or more QuBits

QUBOs for TSP and Maximum-3SAT

QUBO-NN - Reverse-Engineering QUBO matrices

A note on Adiabatic Evolution in Quantum Annealing

Quantum Annealing Hamiltonian Example Calculation

List of QUBO formulations (80)