Search papers, labs, and topics across Lattice.
This paper tackles the problem of fitting an unknown number of hyperplanes to data by reformulating it as an unsupervised learning task on the unit sphere manifold. They introduce a Two-Stage Manifold Optimization algorithm, using a Riemannian Expectation-Maximization process with a heavy-tailed kernel in Phase I for robust estimation, followed by hard matching in Phase II for precise local optima. The method also incorporates a projected density estimation strategy for initialization, leading to improved geometric accuracy and robustness compared to existing methods.
Manifold optimization provides a surprisingly effective and geometrically consistent approach to fitting an unknown number of hyperplanes, outperforming traditional methods.
Fitting an unknown number of hyperplanes to data is a fundamental yet challenging problem in machine learning, characterized by its non-convexity, non-differentiability, and unknown model order. Existing approaches often struggle with local optima or lack geometric consistency. To address these limitations, we propose a novel framework based on Manifold Optimization. We reformulate the problem as an unsupervised learning task on the unit sphere manifold $\mathcal{S}^{\textbf{dim}-1}$. This formulation effectively handles the non-convex constraints and linearizes the distance measurement, rendering the gradient descent tractable. We propose a Two-Stage Manifold Optimization algorithm. In Phase I, we employ a Riemannian Expectation-Maximization process with a heavy-tailed kernel to robustly estimate posterior probabilities, effectively resolving the ambiguities of point distribution between intersecting hyperplanes. In Phase II, upon convergence of the soft estimates, the probabilistic weights degenerate into hard matching, generating a precise local optimum that strictly satisfies the geometric definition. Furthermore, we introduce a projected density estimation strategy for initialization to facilitate global convergence by significantly reducing the feature description space and search complexity. Extensive experiments demonstrate that our method outperforms state-of-the-art baselines in both geometric accuracy and robustness.
