Skip to content

三维点云处理(十二):高斯混合模型 (GMM) 聚类 ​

在点云聚类中,传统的 K-Means 算法采用“硬分配(Hard Assignment)”,强行将每个点划归为单一簇,且假设簇为球形各向同性分布。然而,真实点云数据常常包含各向异性(椭球状)分布、重叠区域以及噪声。高斯混合模型(Gaussian Mixture Model, GMM) 通过引入多维高斯分布与概率“软分配(Soft Assignment)”,为点云聚类提供了坚实的概率建模基础。


一、 从硬分配到概率软分配 ​

1.1 K-Means 的局限与 GMM 的动机 ​

  • 不确定性评估(Uncertainty Estimation):位于两簇交界边缘或噪声区域的点,K-Means 依然强制赋予硬标签 0 或 1,缺乏置信度表达。
  • 各向异性形状拟合:K-Means 假设各个方向方差一致(球形);而 GMM 通过协方差矩阵 可以拟合任意方向拉伸、倾斜的椭球状点云簇。

二、 多维高斯分布与 GMM 概率表达 ​

2.1 多维高斯分布 (Multivariate Gaussian Distribution) ​

对于 维空间中的点 ,均值向量为 ,协方差矩阵为 ,其概率密度函数定义为:

  • 几何直觉:协方差矩阵 决定了椭球分布的主轴方向与拉伸程度(相当于三维点云的 PCA 协方差矩阵特征向量与特征值)。

2.2 GMM 的线性组合表达 ​

高斯混合模型将数据分布假设为 个单高斯分布的线性组合:

其中 为第 个高斯成分的混合系数/先验权重(Mixing Coefficients),满足约束:

参数集合表示为 。我们可以利用拟合好的 生成新的数据点,但聚类的核心目的是求解反向的后验归属问题。

2.3 多目标拟合的工程物理意义 ​

在三维点云处理中,GMM 的数学公式具有极其直观的物理意义。将一个复杂的自动驾驶场景拟合为 个高斯混合分布,本质上是将场景解构为 个基础几何体:

  • 均值 :代表第 个物体的空间质心位置。
  • 协方差矩阵 :代表第 个物体的尺寸、形状与朝向。例如,如果 的特征值在一个方向上极大,在另两个方向极小,说明拟合出了一个细长的物体(如电线杆或行人);如果两个特征值大,一个极小,说明拟合出了一个平面(如一段路面或墙壁)。
  • 混合权重 :代表第 个物体的相对点数规模。一辆大卡车的 显然会远大于一个孤立的交通锥。

与强制划分“势力范围”的 K-Means 不同,GMM 允许这些椭球体在空间中发生交叠(Overlap)。对于处在交叠区域的点,GMM 能够自然地输出“它有 60% 的概率属于电线杆,40% 的概率属于背后的墙面”这样的软概率,为后续的跟踪算法或分类器提供宝贵的置信度(Confidence)信息。


三、 隐变量表达与后验概率 (Posterior Responsibility) ​

3.1 隐变量 的引入 ​

为了用图模型表达 GMM,引入 维二进制隐变量 ,采用 (独热编码)表示:

联合概率分布写为:

对隐变量 求和边缘化后,即可还原出上一节的概率密度 。

3.2 后验概率(责任度 )推导 ​

给定已知数据点 ,其属于第 个高斯分布的后验概率定义为责任度(Responsibility )。根据贝叶斯法则(Bayes' Rule)与全概率公式:

  • 含义: 即为点 分配给第 个簇的软概率,满足 。

四、 极大似然估计 (MLE) 与奇异性危机 ​

4.1 极大似然估计 (MLE) 目标 ​

给定 个独立同分布的数据点集 ,GMM 参数的对数似然函数为:

由于对数符号内部包含了求和符号 ,无法直接求出封闭解析解。

4.2 奇异性问题 (Singularity Problem) ​

在直接对 GMM 进行 MLE 求解时,会出现严重的数值退化/奇异性危机:

解决方案与工程技巧: ​

  1. 启发式重置:在迭代过程中若发现某高斯成分的协方差行列式 过小,将其质心 重新初始化为随机数据点,并重置 为较大初始值;
  2. MAP(最大后验估计):对协方差矩阵加入先验约束(如 Inverse-Wishart 先验);
  3. 贝叶斯 GMM (Bayesian GMM):使用变分推导引入不确定性估计。

五、 GMM 参数迭代更新公式汇总 ​

通过计算对数似然函数对各个参数的偏导数(或利用 EM 算法),可导出 GMM 参数的闭式迭代更新公式:

定义第 个簇的有效样本数(Effective Number of Points):

1. 均值向量 更新: ​

  • 解读:新质心是以责任度 为权重的加权平均值。

2. 协方差矩阵 更新: ​

  • 解读:新协方差矩阵是以责任度为权重的加权协方差。

3. 混合权重 更新: ​

  • 解读:新权重等于该簇有效点数占总点数的比例。

六、 总结与算法优缺点 ​

维度特性总结
计算复杂度,涉及 协方差矩阵与逆矩阵计算
算法优点1. 提供完整的概率不确定性评估(软分配);
2. 能拟合各向异性椭球状分布(比 K-Means 适应性强);
3. 对边缘噪声具有更强的鲁棒性。
算法缺点1. 需预先指定簇数量 ;
2. 强假设每个簇服从凸高斯分布(无法处理非凸流形);
3. 存在似然函数奇异性坍塌风险(Singularity)。

基于 VitePress 强力驱动 | 记录技术与生活