中文

qrpca:一个具有 GPU 加速的快速主成分分析包

天体物理仪器与方法 2022-09-07 v2 数学软件

摘要

我们推出 qrpca,一个快速且可扩展的 QR 分解主成分分析包。该软件以 R 和 python 两种语言编写,内部矩阵计算利用 torch 并支持可用的 GPU 加速。qrpca 分别提供与 prcomp(R)和 sklearn(python)包类似的功能。基准测试表明,对于大维度矩阵,qrpca 的计算速度可比默认实现快 10-20 ×\times,而对于光谱数据立方体的标准分解至少快两倍。qrpca 源代码已免费向社区开放。

关键词

引用

@article{arxiv.2206.06797,
  title  = {qrpca: A Package for Fast Principal Component Analysis with GPU Acceleration},
  author = {Rafael S. de Souza and Xu Quanfeng and Shiyin Shen and Chen Peng and Zihao Mu},
  journal= {arXiv preprint arXiv:2206.06797},
  year   = {2022}
}