arXiv 机器学习· Peng Xu, Nihar Koganti, Volodymyr Kindratenko, Xiaohui Chen·· 1 天前AI 评分35
GEM-KMeans：面向大规模聚类的高显存效率与高精度 GPU 优化方法
GEM-KMeans: Memory-Efficient and Accurate Clustering on Massive Scale with GPU Optimization
AI 导读
研究人员提出面向大规模聚类的 GPU 优化方法 GEM-KMeans，通过将梯度更新、非负投影和归一化等操作融合至矩阵乘法 epilogue 中，实现高显存效率与精准聚类。
来源：arXiv 机器学习 · arxiv.org