A Novel Convex Relaxation for Non-Binary Discrete Tomography
Jan Kuske, Paul Swoboda, Stefania Petra

TL;DR
This paper introduces a new convex relaxation method for non-binary discrete tomography that jointly solves reconstruction and labeling, leading to more accurate and tighter solutions than existing approaches.
Contribution
A novel joint convex relaxation formulation for non-binary discrete tomography that improves solution tightness and integrates reconstruction and labeling tasks.
Findings
Tighter convex relaxation achieved compared to previous methods.
Experimental results show superior reconstruction quality.
The approach outperforms existing relaxations both mathematically and empirically.
Abstract
We present a novel convex relaxation and a corresponding inference algorithm for the non-binary discrete tomography problem, that is, reconstructing discrete-valued images from few linear measurements. In contrast to state of the art approaches that split the problem into a continuous reconstruction problem for the linear measurement constraints and a discrete labeling problem to enforce discrete-valued reconstructions, we propose a joint formulation that addresses both problems simultaneously, resulting in a tighter convex relaxation. For this purpose a constrained graphical model is set up and evaluated using a novel relaxation optimized by dual decomposition. We evaluate our approach experimentally and show superior solutions both mathematically (tighter relaxation) and experimentally in comparison to previously proposed relaxations.
Click any figure to enlarge with its caption.
Figure 1
Figure 2
Figure 3
Figure 4
Figure 5| STD relax | STD BB | CTG CB | CTG relax | CTG BB | |
|---|---|---|---|---|---|
| duality gap ”” | |||||
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
\standaloneconfig
mode=buildnew, build=latexoptions=-interaction=batchmode -shell-escape -jobname
\buildjobname
11institutetext: †MIG, Inst. Appl. Mathematics, Heidelberg University
‡ Institute of Science and Technology (IST), Austria
A Novel Convex Relaxation for Non-Binary Discrete Tomography
Jan Kuske†
Paul Swoboda‡ and Stefania Petra†
Abstract
We present a novel convex relaxation and a corresponding inference algorithm for the non-binary discrete tomography problem, that is, reconstructing discrete-valued images from few linear measurements. In contrast to state of the art approaches that split the problem into a continuous reconstruction problem for the linear measurement constraints and a discrete labeling problem to enforce discrete-valued reconstructions, we propose a joint formulation that addresses both problems simultaneously, resulting in a tighter convex relaxation. For this purpose a constrained graphical model is set up and evaluated using a novel relaxation optimized by dual decomposition. We evaluate our approach experimentally and show superior solutions both mathematically (tighter relaxation) and experimentally in comparison to previously proposed relaxations.
†† Acknowledgments: We gratefully acknowledge support by the DFG (German Science Foundation), Grant GRK 1653. This work is partially funded by the European Research Council under the European Unions 7th Framework Programme (FP7/2007-2013)/ERC grant agreement no 616160. The authors would like to thank Vladimir Kolmogorov for helpful discussions.
1 Introduction
We study the discrete tomography problem, that is reconstructing a discrete-valued image from a small number of linear measurements (tomographic projections), see Figure 1 for an illustration. The main difficulty in reconstructing the original image is that there are usually far too few measurements, making the problem ill-posed. Hence, it is common to search for a discrete-valued image that (i) satisfies the measurements and (ii) minimizes an appropriate energy function.
More generally, the discrete tomography problem can be regarded as reconstructing a discrete-valued synthesis/analysis-sparse signal from few measurements which is observed by deterministic sensors . This is, in turn, a special instance of the compressed sensing problem [7], for which it has been shown that discreteness constraints on the possible values of the reconstructed function can significantly reduce the number of required measurements [10].
However, the discreteness constraint leads to great computational challenges. Simple outer convex relaxations coming from continuous scenarios are doomed to fail, as they will not output discrete solutions, unless the signal sparsity satisfies favourable relations and is well-conditioned on the class of sparse signals, e.g. a random matrix. In fact, in most practical scenarios the projection matrices fall short of assumptions that underlie rigorous compressed sensing theory (like e.g. the restricted isometry property [7]), and standard algorithms from the continuous -setting cannot be applied any more. Rounding continuous solutions will on the other hand render the solutions infeasible for the measurement constraints.
Therefore, algorithms exploiting the combinatorial structure of the discrete tomography problem are necessary to successfully exploit discreteness as prior knowledge and to reduce the number of required measurements.
Related work.
Several algorithms have been proposed to solve the discrete tomography problem. Among them are (i) linear programming-based algorithms [9, 17], (ii) belief propagation [8], (iii) network flow techniques [3], (iv) convex-convave programming [19, 14] (v) evolutionary algorithms [2] and other heuristic algorithms [4, 6, 11, 12]. Not all approaches are applicable to the general discrete tomography problem we treat here: algorithms [9, 17, 8, 6, 11, 12] only support binary labels, while [4, 3] solves only the feasiblity problem and does not permit any energy. Algorithms [19, 14] are applicable to the setting we propose but are purely primal algorithms and do not output dual lower bounds (which we do). Hence, one cannot judge proximity of solutions computed by [19, 14] to global optimal ones, prohibiting its use in branch and bound. In case convex relaxations were considered [9, 19, 17, 3], they were less tight than the one we propose, leading to inferior dual bounds.
Contribution.
We propose the (to our knowledge) first LP-based algorithm for the non-binary discrete tomography problem. In particular, we
- •
recast the discrete tomography problem as a Maximum-A-Posteriori inference problem in a graphical model with additional linear constraints coming from the tomographic projections in Section 2,
- •
construct higher order factors in the graphical model such that a feasible solution to the higher order factors coincides with solutions feasible for the tomographic projections and
- •
present an efficient exact algorithm for solving the special case when exactly one ray constraint is present and the energy factorizes as a chain (such a problem will be called a one-dimensional discrete tomography problem) in Section 3,
- •
decompose the whole problem into such subproblems and solve this decomposition with bundle methods in Section 4.
Our approach leads to significantly tighter bounds as compared to generalising previously proposed relaxations to the non-binary case, see Proposition 1 in Section 4 and experiments in Section 5. Code and datasets are available on GitHub111https://github.com/pawelswoboda/LP_MP.
Notation.
Let be the set of natural numbers between and . For we denote by and the floor and ceiling function.
2 Problem Statement
The discrete tomography problem we study consists in finding a discrete labeling such that (i) tomographic projection constraints given by with are fulfilled and (ii) minimizes some energy . We assume that factorizes according to a pairwise graphical model: given a graph , together with a label space , , the energy is a sum of unary potentials and pairwise ones . The full problem hence reads
[TABLE]
For the discrete tomography problem we usually choose to be a grid graph corresponding to the pixels of the image to be reconstructed, zero unary potentials , as no local information about the image values is known, and pairwise potentials penalize intensity transitions, e.g. (TV) or (Potts). Such choice of pairwise potentials assigns small energy to labelings with a regular spatial structure.
3 One-Dimensional Non-Binary Discrete Tomography
A natural decomposition of the discrete tomography problem (1) consists of (i) considering a subproblem for each ray constraint separately and (ii) joining them together via Lagrangian variables. We will study the first aspect below and the second one in the next section. In particular, let be the variables from a single ray constraint corresponding to a row of the projection matrix in (1). Assume that pairwise potentials form a chain, i.e. they are , . The one-dimensional discrete tomography problem is
[TABLE]
We present an exact linear programming relaxation and an efficient message-passing routine to solve (2) below.
3.1 Linear Programming Model
The one-dimensional discrete tomography subproblem (2) could naively be solved by dynamic programming by going over all variables sequentially. This however would entail quadratic space complexity in the number of nodes in , as the state space for variable would need to include costs for all possible labels and all values of the intermediate sum . The latter sum can have possible values. To achieve a better space complexity, we will recursively (i) equipartition variables , (ii) define LP-subproblems in terms of so-called counting factors which are exact on each subpartition and (iii) join them together to eventually obtain an exact LP-relaxation for (2). Our approach is inspired by [16].
Partition of variables.
Given the nodes , we choose an equipartition and . We recursively equipartition into and and do likewise for . For we obtain a recursive partitioning as in Figure 2.
Counting factors.
Given an interval, a counting factor holds the states of its left and right end and the value of the intermediate sum.
Definition 1 (Counting label space)
The counting label space for interval is with holding all possible intermediate sums. A counting label consists of the three components : its left endpoint label , intermediate sum and right endpoint label .
See again Figure 2 for the exemplary case .
For interval there are distinct counting labels. We associate to each counting factor counting marginals satisfying .
Assuming an uniform label space , the total space complexity of all counting factors is , hence subquadratic in the number of nodes in .
Joining counting factors.
Assume the partitioning of variables has produced two adjacent subsets and , which were constructed from their common subset . The associated three counting factors with marginals and introduced above shall be consistent with respect to each other.
Definition 2 (Label consistency)
Label , and are consistent with each other, denoted by iff (i) left endpoint labels of and match, (ii) right endpoint labels of and match and (ii) intermediate sums match .
We enforce this by introducing a higher order marginal to bind together and .
[TABLE]
The recursive arrangement of counting factors is illustrated in Figure 3.
Remark 1
The constraints between and and are analoguous to the marginalization constraints between pairwise and unary marginals in the local polytope relaxation for pairwise graphical models [18]. The constraints between and however are different. Hence, specialized efficient solvers for inference in graphical models cannot be applied.
Costs.
Above we have described the polytope for the one-dimensional discrete tomography problem (2). The LP-objective consists of vectors for each counting marginal and for each higher order marginal. Accounting for the pairwise costs in (2) we set for the counting factors and for the higher order factors we set . For the projection constraint in (2) we set costs of the top counting marginal as .
3.2 Message Passing Algorithm
Above we have introduced a linear program formulation for the one-dimensional discrete tomography problem (2). While it is possible to solve it with a standard LP-solver, doing so would be slow. As the counting factors and the higher order marginals connecting them form a tree, it is possible to devise a message passing algorithm optimizing (2) exactly. First, this implies that the linear programming relaxation for (2) is exact, as message passing amounts to optimizing the Lagrangian dual of this same relaxation. Second, marginals do not need to be held explicitly, holding messages is enough. The size of all messages equals the size of all counting factors, hence giving again subquadratic space complexity.
Message passing for (2) is detailed in Algorithm 1. It proceeds by first computing up messages from adjacent fine subsets to coarser subsets (i.e. going up the tree in Figure 3) and afterwards computing down messages from coarse subsets to their equipartition (i.e. going down the tree in Figure 3). Messages reparametrize costs of counting and higher order factors.
Reparametrization
Let indices be given, where , and are subsets generated by the recursive partitioning. Let messages correspond to constraints (3), (4) and (5) respectively. Messages act on (reparametrize) costs as
[TABLE]
Fast message computation
Naively computing one up messages would result in time complexity , which would make the algorithm prohibitively slow. We will describe a fast message computation technique for (1), which uses the structure of the corresponding linear constraints (5) and relies on the latent factorization of . Specifically, when we fix the endpoints of interval and of , (1) becomes
[TABLE]
Problem (9) is an instance of the min-sum convolution problem: Given , compute , where . This can be seen by replacing by , by and noting that is a constant, as and were fixed. For the min-sum convolution problem efficient algorithms [5] were proposed with expected running time under the assumption that sorting and results in permutations occurring with uniform probability. Problem (9) can be efficiently computed by performing min-sum convolutions (one convolution for every choice of endpoints).
Remark 2 (Comparison to [16])
While our approach for solving (2) is inspired by [16], it is notably different: (i) our model includes pairwise potentials forming a chain, while [16] assumes that pairwise potentials do not occur between neighboring subsets. This necessitates to store left and right endpoints in counting factors. (ii) [16] optimizes a different objective: they solve the sum-product version of (2) (i.e. they exchange min by and by in (2)). This allows [16] to use fast Fourier transforms for message computations, instead of the harder min-sum convolution problems.
4 Discrete Tomography Graphical Model
The discrete tomography problem (1) consists of m = #rows distinct one-dimensional subproblems (2). We connect all subproblems (2) via Lagrangian variables into one large problem. This procedure is called dual decomposition, see [15] for an introduction. Specifically, in our discrete tomograpy problems subproblems only share variables , but not edges (shared edges can be handled analoguously). Then for each node which participates in the -th subproblem, we introduce the Lagrangian variable . The -th subproblem then consists of solving (2) with the subset of variables , where the unary potentials are the Lagrangian variables . We denote its energy by . The overall problem is
[TABLE]
An exemplary model with eight subproblems coming from two projection directions can be seen in Figure 4.
Optimization of relaxation (CTG).
To maximize (CTG) we use the bundle solver ConicBundle222https://www-user.tu-chemnitz.de/~helmberg/ConicBundle/ to find optimal Lagrangian variables and Algorithm 1 to find solutions to the one-dimensional subproblems. The bundle method will only give us a dual lower bound to the value of the optimal reconstruction.
Primal solution.
To obtain a feasible reconstruction, we solve a reduced problem by excluding labels with high cost: Given dual variables , let be the optimal solution to the -th subproblem on variables . For each label , we compute the energy of the minimal reconstruction for subproblem when the label at is fixed to (this value can be read off from the reparametrization output by Algorithm 1). Only if the gap is smaller than some given threshold, we consider the label . We construct the discrete tomograpy problem on this reduced set of possible labelings and solve the problem with CPLEX [1].
Comparison to previously used relaxation.
It can be shown that the algorithms in [9, 19, 17, 3] use the following relaxation.
[TABLE]
This relaxation (STD) is the straigtforward generalization of the local polytope relaxation [18] to the discrete tomography problem. The only difference are the linear constraints in the last line of (STD). When specialized to the one-dimensional discrete tomography problem (2), the difference between (STD) and our approach is: for (STD) the tomographic projections are directly enforced through the unary marginals instead of enforcing them through the counting factors and higher order ones as we did in Section 3. This more simplistic relaxation (STD) is however less tight.
Proposition 1
Relaxation (STD) is less tight than (CTG).
Proof
Relaxation (STD) is equivalent to applying it to each tomographic projection separately and then joining every subproblem by Lagrangian variables as we did with our approach above (CTG), see [15, Section 1.6]. Hence, it is enough to show that (STD) is not tight in the one-dimensional case (2). We give a counter-example. Assume and we are given Potts pairwise potentials and zero unary potentials . Set unary marginals and and pairwise marginals as . Such marginals are feasible to (STD), yet give cost [math]. On the other hand for e.g. and there must be at least one label transition, which the Potts potential penalizes with cost .∎
5 Experiments
Test images.
We used randomly generated images with three distinct intensity values , examples of which can be seen in Figure 5. Matrices for the tomographic projections were constructed as in [13]. For each test image we consider two tomographic problems: (i) measuring along horizontal and vertical directions or (ii) measuring along horizontal, vertical and two diagonal directions (left upper to right lower and left lower to right upper corner). This gives 400 test problems in total. Potentials for energy in (1) are: unary potentials are zero, while pairwise ones are (that corresponds to TV). Due to integrality of all costs, optimality is ascertained through a duality gap .
Algorithms.
We identify our solvers by a prefix {CTG|STD} depending on whether (CTG) or (STD) is solved and by a suffix {CB|relax|BB} depending on whether ConicBundle, CPLEX [1] or CPLEX with branch and bound enabled was utilized. This gives in total 5 solvers: CTG_CB, CTG_relax, CTG_BB, STD_relax and STD_BB. We set a timelimit of 1 hour for all algorithms.
Unfortunately, CPLEX cannot solve problems larger than . When solving the relaxation (CTG), it already consumes multiple GB of memory for images. Solving (STD) on the other hand leads to low memory consumption, but CPLEX takes too much time for larger problems ( hour). Hence, to have a baseline, we stick to images.
Results.
We have proved in Proposition 1 that relaxation (STD) is less tight than our relaxation (CTG). In fact, the first line in Table 2 shows that this occurs times. Furthermore, our tighter relaxation also actually helps in giving optimality certificates. In Table 1 we confirm this numerically: STD_relax can provide optimality certificates times, while CBC_CB and CTG_relax can do so in total times. Interestingly, when using the branch and bound capabilities of CPLEX, the picture changes and STD_BB outperforms CTG_BB. This is probably due to the fact that CPLEX can solve the underlying relaxation (STD) much faster than (CTG). We conjecture that the picture will change if the more efficient implementation CBC_CB is used as a bounds provider inside a branch and bound solver. This is however outside the scope of our work.
In Figure 6 we give a detailed plot on how much our relaxation (CTG) improved upon (STD).
Also, our relaxation helps in reconstructing the signal. Out of 238 instances, where our heuristic could find an optimal integral solution (third line in Table 2) there were 12 cases, where only our heuristic could do so (second line in Table 2).
6 Conclusion
We have proposed a novel convex relaxation and an accompanying algorithm for the non-binary discrete tomography problem. We have showed theoretically and empirically that our novel relaxation is tighter than the traditionally used relaxation. Solving our new relaxation helps in decoding tomographic reconstructions.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] IBM ILOG CPLEX Optimizer. http://www-01.ibm.com/software/integration/optimization/cplex-optimizer/ .
- 2[2] K. J. Batenburg. An evolutionary algorithm for discrete tomography. Discr. Appl. Math. , 151(1):36–54, 2005.
- 3[3] K. J. Batenburg. A network flow algorithm for reconstructing binary images from continuous x-rays. JMIV , 30(3):231–248, 2008.
- 4[4] K. J. Batenburg and J. Sijbers. DART: A practical reconstruction algorithm for discrete tomography. IEEE TIP , 20(9):2542–2553, 2011.
- 5[5] M. Bussieck, H. Hassler, G. J. Woeginger, and U. T. Zimmermann. Fast algorithms for the maximum convolution problem. Oper. Res. Let. , 15:1–5, 1994.
- 6[6] B. M. Carvalho, G. T. Herman, S. Matej, C. Salzberg, and E. Vardi. Binary tomography for triplane cardiography. IPMI, 1999.
- 7[7] S. Foucart and H. Rauhut. A Mathematical Introduction to Compressive Sensing . Birkhäuser Basel, 2013.
- 8[8] E. Gouillart, F. Krzakala, M. Mézard, and L. Zdeborová. Belief-propagation reconstruction for discrete tomography. Inverse Problems , 29(3):035003, 2013.
