Efficient Algorithms for Sum-of-Minimum Optimization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ding, Lisang, Chen, Ziang, Wang, Xinshang, Yin, Wotao
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909220057645056
author Ding, Lisang
Chen, Ziang
Wang, Xinshang
Yin, Wotao
author_facet Ding, Lisang
Chen, Ziang
Wang, Xinshang
Yin, Wotao
contents In this work, we propose a novel optimization model termed "sum-of-minimum" optimization. This model seeks to minimize the sum or average of $N$ objective functions over $k$ parameters, where each objective takes the minimum value of a predefined sub-function with respect to the $k$ parameters. This universal framework encompasses numerous clustering applications in machine learning and related fields. We develop efficient algorithms for solving sum-of-minimum optimization problems, inspired by a randomized initialization algorithm for the classic $k$-means (Arthur & Vassilvitskii, 2007) and Lloyd's algorithm (Lloyd, 1982). We establish a new tight bound for the generalized initialization algorithm and prove a gradient-descent-like convergence rate for generalized Lloyd's algorithm. The efficiency of our algorithms is numerically examined on multiple tasks, including generalized principal component analysis, mixed linear regression, and small-scale neural network training. Our approach compares favorably to previous ones based on simpler-but-less-precise optimization reformulations.
format Preprint
id arxiv_https___arxiv_org_abs_2402_07070
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient Algorithms for Sum-of-Minimum Optimization
Ding, Lisang
Chen, Ziang
Wang, Xinshang
Yin, Wotao
Optimization and Control
In this work, we propose a novel optimization model termed "sum-of-minimum" optimization. This model seeks to minimize the sum or average of $N$ objective functions over $k$ parameters, where each objective takes the minimum value of a predefined sub-function with respect to the $k$ parameters. This universal framework encompasses numerous clustering applications in machine learning and related fields. We develop efficient algorithms for solving sum-of-minimum optimization problems, inspired by a randomized initialization algorithm for the classic $k$-means (Arthur & Vassilvitskii, 2007) and Lloyd's algorithm (Lloyd, 1982). We establish a new tight bound for the generalized initialization algorithm and prove a gradient-descent-like convergence rate for generalized Lloyd's algorithm. The efficiency of our algorithms is numerically examined on multiple tasks, including generalized principal component analysis, mixed linear regression, and small-scale neural network training. Our approach compares favorably to previous ones based on simpler-but-less-precise optimization reformulations.
title Efficient Algorithms for Sum-of-Minimum Optimization
topic Optimization and Control
url https://arxiv.org/abs/2402.07070