arXiv:2605.28501cs.LG2026-05

用流形优化解决未知数量超平面拟合难题,精度更高更鲁棒。

Fitting Unknown Number of Hyperplanes with Manifold Optimization

论文配图:Fitting Unknown Number of Hyperplanes with Manifold Optimization
图 1 · 摘自论文原文
  • 在单位球流形上重构问题,用流形梯度下降求解非凸优化。
  • 两阶段算法:先软估计后硬匹配,实现几何严格最优。
  • 基于投影密度初始化,提升全局收敛性,适合高维数据场景。

将未知数量超平面的拟合问题重新表述为单位球流形 $\ ext{S}^{\ ext{dim}-1}$ 上的无监督学习任务。该方法有效处理非凸约束并线性化距离计算,使梯度下降可操作。提出两阶段流形优化算法:第一阶段采用带重尾核的黎曼期望-最大化过程,稳健估计点属于各超平面的后验概率,缓解相交超平面间点分布模糊问题;第二阶段在软估计收敛后,概率权重退化为硬匹配,生成严格满足几何定义的精确局部最优解。此外,引入投影密度估计策略进行初始化,显著压缩特征描述空间与搜索复杂度,促进全局收敛。大量实验表明,该方法在几何精度与鲁棒性方面均优于现有最先进基线。

原文摘要 · Abstract (English)

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.

流形优化超平面拟合无监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。