Scalable and Globally Optimal Generalized L₁ K-center Clustering via Constraint Generation in Mixed Integer Linear Programming

Authors

  • Aravinth Chembu University of Toronto, Toronto
  • Scott Sanner University of Toronto, Toronto
  • Hassan Khurram University of Toronto, Toronto
  • Akshat Kumar Singapore Management University, Singapore

DOI:

https://doi.org/10.1609/aaai.v37i6.25857

Keywords:

ML: Clustering, CSO: Constraint Optimization, CSO: Mixed Discrete/Continuous Optimization

Abstract

The k-center clustering algorithm, introduced over 35 years ago, is known to be robust to class imbalance prevalent in many clustering problems and has various applications such as data summarization, document clustering, and facility location determination. Unfortunately, existing k-center algorithms provide highly suboptimal solutions that can limit their practical application, reproducibility, and clustering quality. In this paper, we provide a novel scalable and globally optimal solution to a popular variant of the k-center problem known as generalized L_1 k-center clustering that uses L_1 distance and allows the selection of arbitrary vectors as cluster centers. We show that this clustering objective can be reduced to a mixed-integer linear program (MILP) that facilitates globally optimal clustering solutions. However, solving such a MILP may be intractable for large datasets; to remedy this, we present a scalable algorithm that leverages constraint generation to efficiently and provably converge to its global optimum. We further enhance outlier handling through a simple but elegant extension to our MILP objective. We first evaluate our algorithm on a variety of synthetic datasets to better understand its properties and then validate on 20 real benchmark datasets where we compare its performance to both traditional L_1 distance k-center and k-medians baselines. Our results demonstrate significant suboptimality of existing algorithms in comparison to our approach and further demonstrate that we can find optimal generalized L_1 k-center clustering solutions up to an unprecedented 1,000,000 data points.

Downloads

Published

2023-06-26

How to Cite

Chembu, A., Sanner, S., Khurram, H., & Kumar, A. (2023). Scalable and Globally Optimal Generalized L₁ K-center Clustering via Constraint Generation in Mixed Integer Linear Programming. Proceedings of the AAAI Conference on Artificial Intelligence, 37(6), 7015-7023. https://doi.org/10.1609/aaai.v37i6.25857

Issue

Section

AAAI Technical Track on Machine Learning I