NVIDIA Technical Blog 发布于 08/19 00:48

Run Massive-Scale UMAP in Minutes Using Multiple GPUs—Without Losing Accuracy

(翻译)使用多个 GPU 在几分钟内运行大规模 UMAP——且不损失精度

查看原文

资讯摘要

Uniform Manifold Approximation and Projection (UMAP) is a dimensionality reduction technique widely used for visualization and feature extraction.

AI 摘要

NVIDIA cuML 与 cuVS 25.06 新增多 GPU UMAP 功能,将大规模数据的全邻居 kNN 图构建分布式到多块 GPU 上,显著缩短训练时间。文章介绍算法原理、内存估算、配置建议及 Python 示例,可在数分钟内处理数亿向量的数据集。

打开原文