PyClustrPath：基于 GPU 加速的聚类路径生成高效 Python 包
最优化与控制
2025-01-28 v1
摘要
凸聚类是一种无需先验知识确定聚类数的流行聚类模型。它可以通过连续求解一系列正则化参数值来生成聚类路径。本文介绍了 PyClustrPath，这是一个针对凸聚类模型进行 GPU 加速的高效 Python 包。PyClustrPath 实现了流行的一阶和二阶算法，并采用干净的模块化设计。这种设计使 PyClustrPath 更易于将来集成用于求解凸聚类模型的新算法。我们在流行的聚类数据集上广泛测试了 PyClustrPath 的数值性能，结果表明其在基于凸聚类模型生成聚类路径方面优于现有求解器。PyClustrPath 的实现可在 https://github.com/D3IntOpt/PyClustrPath 查阅。
引用
@article{arxiv.2501.15964,
title = {PyClustrPath: An efficient Python package for generating clustering paths with GPU acceleration},
author = {Hongfei Wu and Yancheng Yuan},
journal= {arXiv preprint arXiv:2501.15964},
year = {2025}
}
备注
13 pages, 10 figures, 3 tables