2.8. 密度估计#

密度估计介于无监督学习、特征工程和数据建模之间。一些最流行和有用的密度估计技术包括混合模型,如高斯混合模型(GaussianMixture),以及基于邻居的方法,如核密度估计(KernelDensity)。高斯混合模型在聚类的上下文中进行了更充分的讨论,因为该技术作为一种无监督聚类方案也很有用。

密度估计是一个非常简单的概念,大多数人已经熟悉一种常见的密度估计技术:直方图。

2.8.1. 密度估计:直方图#

直方图是一种简单的数据可视化方法,其中定义了数据“桶”,并统计每个桶内的数据点数量。以下图的左上角面板展示了一个直方图示例。

hist_to_kde

然而,直方图的一个主要问题是,分桶的选择可能对最终的可视化产生不成比例的影响。考虑上图的右上角面板。它显示了相同数据的直方图,但数据桶向右平移。两个可视化的结果看起来完全不同,可能导致对数据的不同解释。

直观地,我们也可以将直方图想象成一堆方块,每个数据点一个方块。通过在适当的网格空间中堆叠方块,我们得到了直方图。但是,如果不是在规则网格上堆叠方块,而是将每个方块中心放置在其所代表的数据点上,并求和每个位置的总高度呢?这个想法就产生了左下角的可是化结果。它可能不如直方图那么整洁,但数据驱动方块位置的事实意味着它能更好地表示底层数据。

这种可视化是核密度估计的一个例子,本例中使用了平顶核(即在每个点上放置一个方形方块)。通过使用更平滑的核,我们可以得到一个更平滑的分布。右下角的图显示了高斯核密度估计,其中每个点都为总和贡献了一条高斯曲线。结果是一个平滑的密度估计,它从数据中得出,并作为点分布的强大非参数模型。

2.8.2. 核密度估计#

scikit-learn 中的核密度估计在 KernelDensity 估计器中实现,该估计器使用 Ball Tree 或 KD Tree 进行高效查询(有关这些的讨论,请参阅最近邻)。尽管上述示例为简单起见使用了1D数据集,但核密度估计可以在任意维度上进行,尽管在实践中,维度灾难会导致其在高维度下的性能下降。

在下图中,从双峰分布中抽取了100个点,并显示了三种不同核选择下的核密度估计

kde_1d_distribution

很明显,核的形状如何影响所得分布的平滑度。scikit-learn 核密度估计器可以按如下方式使用

>>> from sklearn.neighbors import KernelDensity
>>> import numpy as np
>>> X = np.array([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])
>>> kde = KernelDensity(kernel='gaussian', bandwidth=0.2).fit(X)
>>> kde.score_samples(X)
array([-0.41075698, -0.41075698, -0.41076071, -0.41075698, -0.41075698,
       -0.41076071])

这里我们使用了 kernel='gaussian',如上所示。在数学上,核是一个正函数 \(K(x;h)\),由带宽参数 \(h\) 控制。给定这种核形式,点集 \(x_i; i=1\cdots N\) 中点 \(y\) 处的密度估计由下式给出

\[\rho_K(y) = \sum_{i=1}^{N} K(y - x_i; h)\]

这里的带宽充当平滑参数,控制结果中偏差和方差之间的权衡。大带宽导致非常平滑(即高偏差)的密度分布。小带宽导致不平滑(即高方差)的密度分布。

参数 bandwidth 控制这种平滑。可以手动设置此参数,也可以使用 Scott 和 Silverman 的估计方法。

KernelDensity 实现了几种常见的核形式,如下图所示

kde_kernels

核的数学表达式#

这些核的形式如下:

  • 高斯核(kernel = 'gaussian'

    \(K(x; h) \propto \exp(- \frac{x^2}{2h^2} )\)

  • 平顶核(kernel = 'tophat'

    \(K(x; h) \propto 1\)\(x < h\)

  • Epanechnikov 核(kernel = 'epanechnikov'

    \(K(x; h) \propto 1 - \frac{x^2}{h^2}\)

  • 指数核(kernel = 'exponential'

    \(K(x; h) \propto \exp(-x/h)\)

  • 线性核(kernel = 'linear'

    \(K(x; h) \propto 1 - x/h\)\(x < h\)

  • 余弦核(kernel = 'cosine'

    \(K(x; h) \propto \cos(\frac{\pi x}{2h})\)\(x < h\)

核密度估计器可以与任何有效的距离度量一起使用(有关可用度量列表,请参阅DistanceMetric),尽管结果仅在欧几里得度量下才进行正确归一化。一个特别有用的度量是半正矢距离,它测量球体上点之间的角距离。以下是使用核密度估计可视化地理空间数据的示例,本例中为南美洲大陆上两种不同物种的观测分布

species_kde

核密度估计的另一个有用应用是学习数据集的非参数生成模型,以便有效地从该生成模型中抽取新样本。以下是使用此过程创建一组新手写数字的示例,其中使用了基于数据PCA投影学习的高斯核

digits_kde

“新”数据由输入数据的线性组合组成,其权重根据 KDE 模型进行概率性抽取。

示例