← Search

Yubo Zhuang

4 accepted papers

2024

Statistically Optimal $K$-means Clustering via Nonnegative Low-rank Semidefinite Programming

ICLR 2024oral

$K$-means clustering is a widely used machine learning method for identifying patterns in large datasets. Recently, semidefinite programming (SDP) relaxations have been proposed for solving the $K$-means optimization problem, which enjoy strong statistical optimality guarantees. However, the prohibi…

Cited by 2SourcePDFScholar
2022

Sketch-and-lift: scalable subsampled semidefinite program for K-means clustering

AISTATS 2022poster

Semidefinite programming (SDP) is a powerful tool for tackling a wide range of computationally hard problems such as clustering. Despite the high accuracy, semidefinite programs are often too slow in practice with poor scalability on large (or even moderate) datasets. In this paper, we introduce a l…