Sample-Efficient Geometry Reconstruction from Euclidean Distances using Non-Convex Optimization

The problem of finding suitable point embedding or geometric configurations given only Euclidean distance information of point pairs arises both as a core task and as a sub-problem in a variety of machine learning applications. In this paper, we aim to solve this problem given a minimal number of distance samples. To this end, we leverage continuous and non-convex rank minimization formulations of the problem and establish a local convergence guarantee for a variant of iteratively reweighted least squares (IRLS), which applies if a minimal random set of observed distances is provided. As a technical tool, we establish a restricted isometry property (RIP) restricted to a tangent space of the manifold of symmetric rank-$r$ matrices given random Euclidean distance measurements, which might be of independent interest for the analysis of other non-convex approaches. Furthermore, we assess data efficiency, scalability and generalizability of different reconstruction algorithms through numerical experiments with simulated data as well as real-world data, demonstrating the proposed algorithm's ability to identify the underlying geometry from fewer distance samples compared to the state-of-the-art.

Paper

References (80)

Scroll for more · 38 remaining

Similar papers

Peer review

Reviewer fWZw6/10 · confidence 3/52024-07-06

Summary

This paper proposed a method called MatrixIRLS to solve the EDG problem, an important problem in various fields, especially when pairwise distance is partially observed. The proposed method improves the number of random sampled distances to guarantee local convergence to the ground truth.

Strengths

The related work is discussed comprehensively. The method and theory seem solid. Experiments are convincing.

Weaknesses

1. There are quite a few typos or undefined notation that make some of the paragraphs hard to read. For example: a. Line 66, $\nu$ is not defined. Although it’s briefly mentioned in line 80, its rigorous definition is on Page 5. I suggest the following change to line 66: $O()$ (where $\nu$ is the coherence factor, see \cref{def:4.1} for details). b. Line 158, $\sigma_i(\cdot)$ is not defined, tho I can guess the meaning. c. Line 165: $X^{(k)}$ should be $X$. d. Eq (5) gives the update from k to k+1, while Eq (6) gives the rule from k-1 to k. e. Eq(9), at the end, there is a missing comma. 2. Figure 2 looks great, but it will be even better if we can observe the claimed rate $O(\nu rn\log n)$ empirically.

Questions

1. There are some vertical space issues, like Line 286 for Section 5. 2. Can the authors briefly discuss the main challenges for global convergence?

Rating

6

Confidence

3

Soundness

4

Presentation

3

Contribution

4

Limitations

Discussed in the supplementary material. I suspect the file in the supplementary zip file should be the main submitted file.

Reviewer Jqjh4/10 · confidence 4/52024-07-12

Summary

The paper studies the under-constrained problem to reconstruct n ordered points in Euclidean space R^r, which are given by fewer than all n(n-1)/2 pairwise distances.

Strengths

The authors should be highly praised for clearly stating main Problem 1 on page 1 and for formulating main Theorems 4.2 and 4.5 almost rigorously. Problem 1 is well-motivated by the AlphaFold approach to protein folding prediction when a protein backbone in 3D is reconstructed form partially known distances between alpha-carbon atoms.

Weaknesses

Different from the initial Problem 1, the actual considered problem is the rank minimization in (1) after line 54. Does the paper explain anywhere why it makes sense to minimize this integer rank of a Gram matrix even if we forget about initial points. Another weakness is the practical impossibility of solving this rank minimization, which is claimed in line 54 to be NP-hard. It would be useful to re-phrase a problem in another way that allows a polynomial solution. More importantly, the rank minimization in (1) and algorithm MatrixILRS appeared under the same name but with slightly different notations in the earlier paper [KV21]. So the problem and algorithm in the submitted paper look incremental updates of the past work, which even uses the same names, includes similar theorems, example structures and pseudo-codes.

Questions

For a simple example, if n=4 points in the plane are given by 4 distances that are edge lengths of a quadrilateral on these 4 points, then what's the meaning of this rank minimization in this case? Can Problem 1 be explicitly solved in some partial cases, e.g. for 5 distances on 4 points in the plane? In Def 3.2, is w_alpha a matrix n x n? In lines 177-178, how is measurement operator <w_alpha_l,X> computed? Are angle brackets missed for l>m in line 178? The concept of a nu-incoherent matrix appears in Theorems 4.2, 4.5 without earlier explanations. What are these matrices? The words "coherence factor" appear for the first time in line 80 without defining this factor nu. Also, nu appears earlier in line 66, still without explanations. Is it the same "coherence" appearing much later in Definition 4.1? The word "probability" appears for the first time in Theorem 4.2, later again in Theorem 4.5. How is the probability space defined to make the convergence with "high probability" meaningful? In Theorem 4.3, can mu be large enough so that the resulting inequality becomes equality and hence no convergence is guaranteed? What are the values of mu in the experiments form section 5? Lines 251-252 say that "an adaptation of the proof of [KV21] to the EDG setting is not possible due to non-orthogonality of the basis {w_alpha}_alpha of Definition 3.2". What specific step in the past proof requires this orthogonality? Lines 857-858 say that "The Procrustes analysis uses similarity transformations like scaling, rotation and maps into a common reference frame [ADSVF23]". However the minimization after line 859 includes only translations and orthogonal maps. Was uniform scaling missed? Why is the minimum achieved over infinitely many possible translations? Does this Procrustes distance satisfy the metric axioms and what is its computational complexity in the number of points? The biggest concern is line 82 saying that "from which the geometry of the points P is trivially recovered". Main Algorithm 1 (MatrixIRLS) outputs a matrix X^(k) through some iterative optimization. Why is this final n x n matrix realizable by n points in a Euclidean space R^r? For instance, we cannot arbitrarily choose 6 pairwise distances between 4 points in the plane. Even if these distances are positive and satisfy triangle inequalities, they should also satisfy a complicated polynomial equation saying that the tetrahedron on these 4 points has volume 0. For n>4 points, there are many more such equations, so a random symmetric matrix of positive numbers is realized by points with probability 0.

Rating

4

Confidence

4

Soundness

2

Presentation

2

Contribution

2

Limitations

This submission seems an incremental improvement of the past work [KV21] with little comparison between the two papers.

Reviewer wbRb6/10 · confidence 3/52024-07-13

Summary

This paper tackles the problem of reconstructing geometric configurations from minimal Euclidean distance samples. The authors propose an innovative algorithm based on iteratively reweighted least squares (IRLS) within a non-convex optimization framework. They provide a local convergence guarantee for the method, ensuring accurate geometry recovery from a minimal number of random distance samples. Additionally, the paper establishes a restricted isometry property (RIP) relevant to other non-convex approaches. Numerical experiments demonstrate the algorithm’s efficiency and robustness compared to state-of-the-art methods.

Strengths

1. The paper introduces a novel application of IRLS for non-convex optimization in Euclidean Distance Geometry (EDG) problems. Establishing the local convergence guarantee and RIP is a significant theoretical contribution. 2. The paper is well-researched, with rigorous theoretical analysis and proofs supporting the proposed method’s convergence and efficiency. The empirical validation through both synthetic and real-world datasets adds robustness to the claims. 3. The paper is well-structured and clearly written, making complex mathematical concepts accessible. The detailed algorithmic steps enhance understanding. 4. The proposed method addresses a fundamental problem in various applications, such as molecular conformation and sensor network localization. The ability to achieve accurate reconstruction with fewer samples has significant implications for data efficiency in these fields.

Weaknesses

1. The paper compares the proposed method with a few state-of-the-art algorithms. Including more diverse comparisons, particularly with newer methods covered in the existing literature, would provide a more comprehensive evaluation. 2. The experiments could be enhanced by adding more parameter analysis or robustness analysis. For example, testing the performance under different parameters (n, r, k) on the ill-conditioned data will further validate the effectiveness of the proposed method. 3. The paper provides a local convergence guarantee, but a more detailed discussion on the global convergence properties of the algorithm would be beneficial.

Questions

1.How sensitive is the proposed algorithm to the choice of parameters, such as rank estimate in the IRLS framework? Providing guidelines or heuristics for parameter selection would be helpful. 2.In the additional experiments presented in Appendix E.1, why does the MatrixIRLS method exhibit a large standard deviation when rho is in the range of 2.5 to 3.5? This suggests that the method is not very stable or robust across all settings. Considering the success probability, ALM performs better than MatrixIRLS. Could you explain the potential reasons for this discrepancy? 3.The time complexity of the MatrixIRLS algorithm appears to be missing a component in Step 7, which involves singular value decomposition with a complexity of O(n^3). Given this, what is the scalability of the MatrixIRLS method, such as the maximum problem size that can be solved within one hour? 4.Will the results of the MatrixIRLS method converge to the same solution as MDS when there is no incomplete information? *Small suggestions:* 1.The linewidth and point size is too large in Figs. 2, 6(b), and 7(b). 2.No captions in Figs. 6 and 7. 3.Mixed use of fig. and Figure. 4.Some typos: Line 52: should satisfy also satsify Line 813: figure Figure

Rating

6

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

It would be beneficial to include the scalability limitations, such as the maximum problem size that can be solved.

Reviewer rCFC7/10 · confidence 2/52024-07-13

Summary

The paper examines the problem of recovering point locations based on incomplete pairwise Euclidean distance information. The problem is cast as a low-rank matrix recovery problem, and an algorithm based on iteratively-reweighted least squares is proposed. The authors derive a lower bound for the minimum number of required samples for reconstruction, and show that their algorithm has quadratic convergence. Experimental results show that their method has comparable results to other state-of-the-art methods for most data, but is also uniquely robust to ill-conditioned cases where other methods fail to converge.

Strengths

* Provides the first theoretical convergence guarantee at the optimal amount of samples for the studied pairwise distance problem under non-convex optimization. * Performance comparable to state of the art methods are demonstrated on * The method converges quickly and effectively even for ill-conditioned matrices, where previous state of the art methods fail. * Detailed theoretical derivations and proofs. * An complete anonymized code repository is provided for reproducibility.

Weaknesses

* Example data is somewhat small-scale and synthetic, but appears sufficient for comparison with other recovery methods. * While this is not a weakness of the paper, my own theoretical background is insufficient to verify some of the proofs, such as the construction of the dual basis followed by the establishment of the restricted isometry property condition.

Questions

* While number of steps/time to convergence is shown for ill-conditioned matrix experiments, they are not shown for real data experiments. How much faster does the proposed MatrixIRLS derivative converge compared to ALM? Are they comparable? * Are there applicable downstream machine learning/deep learning applications that require this kind of point recovery, where improved results could be further demonstrated?

Rating

7

Confidence

2

Soundness

4

Presentation

4

Contribution

3

Limitations

The authors address the limitations of their work in detail in the paper, and there are no societal impacts to be addressed.

Reviewer rCFC2024-08-13

Thanks

I thank the authors for their work on the rebuttal. Based on their answers to my review, which are further positive points for the paper, as well as reading and considering the other reviews and their rebuttals, I see no reason to change my positive rating of the paper.

Reviewer Jqjh2024-08-10

Thank you for the detailed reply. >As the reviewer points out, not every random symmetric matrix corresponds to a set of points; positive definiteness is required to realize such points. Under the assumptions of our main theorem, the proposed algorithm converges to the true Gram matrix of rank $r$, which is both symmetric and positive definite. We then obtain the configuration of the points through an r-truncated eigendecomposition. Is it possible to specify the line numbers where the final matrix is proved to be symmetric and positive definite?

Authorsrebuttal2024-08-10

Regarding symmetry, we want to note that all iterates $\mathbf{X}^{(k)}$ of Algorithm 1 are symmetric by construction: By (6), the solution of the equality-constrained weighted least squares problem (5) (with $W_{k-1}$ instead of $W_{k}$) can be written as $$ \mathbf{X}^{(k)} = W_{k-1}^{-1}\mathcal{A}^*(z_{k-1}), $$ where $z_{k-1}:= (\mathcal{A}W_{k-1}^{-1} \mathcal{A}^*)^{-1}(\mathbf{y})$ is the solution of the linear system $\mathcal{A}W_{k-1}^{-1} \mathcal{A}^* \mathbf{z} = \mathbf{y}$. From the definition of the measurement operator $\mathcal{A}$ (see line 177-178), it becomes clear that the adjoint operator $\mathcal{A}^*$ of $\mathcal{A}$ maps a vector $z \in \mathbb{R}^{m + n}$ to $\mathcal{A}^*(z) = \sum_{\alpha_{\ell} \in \Omega} z_\ell w_{\alpha_\ell} + \sum_{i=1}^n z_{m+i} w_{(i-m,i-m)}$, where the $w_{\alpha}$ are as in (4). Since all $w_{\alpha}$ are symmetric, it holds therefore that also $\mathcal{A}^*(z_{k-1})$ is symmetric as a sum of symmetric matrices. Based on the definition $W_{k-1}: = W_{\mathbf{X}^{(k-1)},\epsilon_{k-1}}$ of $W_{k-1}$ and Definition 3.1, it can be seen that $W_{k-1}^{-1}$ maps a matrix $\mathbf{Z}$ to $\mathbf{U}\Sigma_{\epsilon_{k-1}}\mathbf{U}^*\mathbf{Z}\mathbf{U}\Sigma_{\epsilon_{k-1}}\mathbf{U}^*$, where $\mathbf{U}$ is a left singular vector matrix of $\mathbf{X}^{({k-1})}$, and $\Sigma_{\epsilon_{k-1}} = \operatorname{diag}(\max(\epsilon_{k-1},\sigma_i(\mathbf{X}^{(k-1)}))$, which is itself symmetric if $\mathbf{Z}$ is symmetric. Thus, each iterate $\mathbf{X}^{(k)}$ of Algorithm 1 is symmetric. Regarding positive semidefiniteness, it is true in general, iterates $\mathbf{X}^{(k)}$ can have negative eigenvalues. However, we prove in Theorem 4.3 that if Algorithm 1 is initialized close enough to a rank-$r$ ground truth matrix $\mathbf{X}^0$ and if the sample complexity satisfies the bound of line (229) under the mentioned random sampling model, with high probability, $\mathbf{X}^{(k)}$ converges to $\mathbf{X}^0$ (with quadratic rate). For the problems considered, the ground truth $\mathbf{X}^0 \in S_n$ is a realizable rank-$r$ matrix and thus p.s.d., so the iterates $\mathbf{X}^{(k)}$ converge to a psd matrix under these conditions. The proof of this theorem can be found in lines 786 to 809 (but uses also other intermediate lemmas, cf. the supplementary material).

Reviewer Jqjh2024-08-12

Thank you for your clarifications. To summarize: symmetry holds, but positive-definiteness of the result is true with high probability only if there exists a ground truth matrix which is reconstructed, but there is no proof that the optimization result is indeed positive-definite in all reconstruction scenarios. This coincides with my initial understanding.

Authorsrebuttal2024-08-13

> To summarize: symmetry holds, but positive-definiteness of the result is true with high probability only if there exists a ground truth matrix which is reconstructed, but there is no proof that the optimization result is indeed positive-definite in all reconstruction scenarios. That is correct. As previously mentioned, in such cases, one can obtain a realizable point configuration by applying a truncated eigendecomposition. This point configuration will not satisfy the distance constraints exactly anymore, but approximately. Comparing such an outcome with the results of reference algorithms such as ALM, RieEDG and ScaledSGD, we would like to point out that the iterates of these algorithms do not satisfy the distance constraints in the first place, unless their objectives converge to 0, which they do not for many challenging problem instances. On the other hand, the satisfaction of the distance constraints by all iterates is built into MatrixIRLS. Thus, applying a truncated eigendecomposition to the outcome of MatrixIRLS is not less justified than using any of the other algorithms.

Reviewer fWZw2024-08-11

Thanks for the response. Your explanation to the rate makes sense to me. It will be very helpful if the authors can discuss this in section 5, to call back the claimed rate in Section 1.1 Similarly, discussing the challenges for global convergence will make the story more complete. I raised the score to 6, mainly because the rate is explained clearly to me, which is part of the main contribution claimed in Section 1.1.

Reviewer wbRb2024-08-14

Thank you for your detailed clarification. As my concerns have been addressed, I will maintain my positive score. By the way, I would like to note that the time complexity you mentioned is not linear in $n$, but rather $O(n \log n)$.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC