PCA工作原理 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

PCA工作原理

主成分分析 (PCA) 是一种学习算法,它可以降低数据集中的维度(特征数量),同时仍保留尽可能多的信息。

PCA通过寻找一组称为组件的新特征来降低维度,这些特征是原始特征的组合,但彼此不相关。第一个成分在数据中可能存在最大的可变性,第二个成分是第二个最易变的,以此类推。

它是一种自主型维数减少算法。在自主学习中,不使用可能与训练数据集中的对象相关联的标签。

假定输入中的矩阵包含行 x_1,…,x_n ,每个行具有维度 1 * d,数据将被分成小批量行,并分布到训练节点(工作线程)中。然后,每个工作线程计算其数据的摘要。然后在计算结束时,不同工作线程的摘要统一为一个解决方案。

模式

Amazon SageMaker PCA 算法使用两种模式中的任何一种来计算这些摘要,具体视情况而定:

  • 常规:针对具有稀疏数据以及适度数量的观察和特征的数据集。

  • 随机:针对具有大量观察和特征的数据集。此模式使用近似算法。

作为算法的最后一步,它在统一解决方案上执行单值分解,然后将会从中导出主成分。

模式 1:常规

工作线程联合计算 Equation in text-form: \sum x_i^T x_i Equation in text-form: \sum x_i

注意

由于 Equation in text-form: x_i 1 * d 个行向量, Equation in text-form: x_i^T x_i 是一个矩阵(非标量)。在代码中使用行向量可以使我们获得高效的缓存。

协方差矩阵的计算方式为 Equation in text-form: \sum x_i^T x_i - (1/n) (\sum x_i)^T \sum x_i ,其最前面的 num_components 个奇异向量构成模型。

注意

如果 subtract_meanFalse,我们会避免计算和减去 Equation in text-form: \sum x_i

当满足以下条件时使用此算法:向量的维度 d 足够小,以致 Equation in text-form: d^2 可以放入内存中。

模式 2:随机

当输入数据集中的特征数量较大时,我们使用一个方法来近似计算协方差指标。对于维度 b * d 的每个小批量 Equation in text-form: X_t ,我们随机初始化一个我们与每个小批量相乘的 (num_components + extra_components) * b 矩阵,从而创建一个 (num_components + extra_components) * d 矩阵。这些矩阵的总和由工作人员计算,服务器在最终(num_components + extra_components) * d的矩阵SVD上执行。其右上方的 num_components 单向量是输入矩阵的顶部单向量的近似值。

指定 Equation in text-form: \ell = num_components + extra_components。假定一个维度 b * d 的小批量 Equation in text-form: X_t ,工作线程会提取维度 Equation in text-form: \ell * b 的随机矩阵 Equation in text-form: H_t 。根据环境使用GPU还是CPU和维度大小,矩阵要么是每个条目所在的随机符号矩阵,要+-1么是 FJLT(Johnson Lindenstrauss 的快速变换;有关信息,请参阅FJLT变换和后续论文)。随后工作线程会计算 Equation in text-form: H_t X_t 并维护 Equation in text-form: B = \sum H_t X_t 。工作线程还维护 Equation in text-form: h^T Equation in text-form: H_1,..,H_T T 为小批量的总数)的列总和以及 s(所有输入行的总和)。在处理完整个数据碎片后,工作线程会向服务器发送 Bhsn (输入行的数量)。

将不同输入对服务器表示为 Equation in text-form: B^1, h^1, s^1, n^1,… 。服务器会计算 Bhsn 的各自输入的总和。然后,它计算 Equation in text-form: C = B – (1/n) h^T s ,并查找其奇异值分解。C 的右上单向量和单值被用作解决问题的近似方法。