Publications

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

Yubo Zhuang, Xiaohui Chen, Yun Yang, Richard Y. Zhang. 

International Conference on Learning Representations (ICLR), 2024. [Code] 

Selected for Oral presentation. (One of 85/7262 submissions. Top 1.2%)

Likelihood Adjusted Semidefinite Programs for Clustering Heterogeneous Data. 

Yubo Zhuang, Xiaohui Chen, Yun Yang. 

International Conference on Machine Learning (ICML), 2023.

Wasserstein K-means for Clustering Probability Distributions. 

Yubo Zhuang, Xiaohui Chen, Yun Yang. 

Conference on Neural Information Processing Systems (NeurIPS), 2022. [Code] 

Sketch-and-Lift: Scalable Subsampled Semidefinite Program for K-means Clustering. 

Yubo Zhuang, Xiaohui Chen, Yun Yang. 

Artificial Intelligence and Statistics Conference (AISTATS), 2022. [Code]