arXiv：cs.LG（机器学习，全量分类）· Peng Xu, Nihar Koganti, Volodymyr Kindratenko, Xiaohui Chen·· 1 天前AI 评分36
GEM-KMeans：面向大规模聚类的 GPU 显存优化方法
GEM-KMeans: Memory-Efficient and Accurate Clustering on Massive Scale with GPU Optimization
AI 导读
GEM-KMeans 提出一种谱归一化但数学等价的非负低秩矩阵分解形式，将梯度更新、非负投影与归一化充分统计量融合进矩阵乘法 epilogue，在 HBM 中只需保留单个因子及小型 tile-reduction 数组，而非三个大规模因子数组。
来源：arXiv：cs.LG（机器学习，全量分类） · arxiv.org