cs.DSMar 18, 2026

Learning-Augmented Algorithms for kk-median via Online Learning

Authors: Anish Hebbar, Rong Ge, Amit Kumar, Debmalya Panigrahi

Organizations: Duke University · IIT Delhi

Abstract

The field of learning-augmented algorithms seeks to use ML techniques on past instances of a problem to inform an algorithm designed for a future instance. In this paper, we introduce a novel model for learning-augmented algorithms inspired by online learning. In this model, we are given a sequence of instances of a problem and the goal of the learning-augmented algorithm is to use prior instances to propose a solution to a future instance of the problem. The performance of the algorithm is measured by its average performance across all the instances, where the performance on a single instance is the ratio between the cost of the algorithm's solution and that of an optimal solution for that instance. We apply this framework to the classic kk-median clustering problem, and give an efficient learning algorithm that can approximately match the average performance of the best fixed kk-median solution in hindsight across all the instances. We also experimentally evaluate our algorithm and show that its empirical performance is close to optimal, and also that it automatically adapts the solution to a dynamically changing sequence.

Figures & tables

Appendix figures & tables8 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Sample-and-Search: An Effective Algorithm for Learning-Augmented k-Median Clustering in High dimensions

    Mar 11, 2026Kangke Cheng, Shihong Song, Guanlin Mo +1ClusteringLearning-Augmented Algorithms

  2. Learning-Augmented Online Minimization with Dual Predictions

    Jun 3, 2026Christian Coester, Alexa Tudose, Alexander TuroczyCombinatorial OptimizationLearning-Augmented Algorithms

  3. Parsimonious Learning-Augmented Online Metric Matching

    May 26, 2026Yongho Shin, Phanu VajanopathLearning-Augmented Algorithms