GIST: Greedy Independent Set Thresholding for Max-Min Diversification with Submodular Utility
Matthew Fahrbach, Srikumar Ramalingam, Morteza Zadimoghaddam, Sara Ahmadian, Gui Citovsky, Giulia DeSalvo
TL;DR
The paper introduces MDMS, a subset selection problem that combines a monotone submodular utility with a max-min diversification term, formalized as maximizing $f(S)=g(S)+\lambda\cdot\text{div}(S)$ under a cardinality constraint. It proposes the GIST algorithm, which achieves a $\tfrac{1}{2}-\varepsilon$ approximation by sweeping distance thresholds and solving bicriteria maximum-weight independent-set problems on intersection graphs, with a stronger $\tfrac{2}{3}-\varepsilon$ guarantee for linear utilities; it also establishes hardness results, including a $0.5584$-approximation barrier for general metrics and APX-completeness for the Euclidean case with linear utility. The paper further strengthens the theory with a warm-up $0.387$-approximation and analyzes the linear-utility setting in depth, providing matching hardness results. Empirically, GIST outperforms state-of-the-art baselines on synthetic tasks and improves single-shot data sampling for ImageNet, demonstrating practical benefits for data summarization and training set curation.
Abstract
This work studies a novel subset selection problem called max-min diversification with monotone submodular utility ($\textsf{MDMS}$), which has a wide range of applications in machine learning, e.g., data sampling and feature selection. Given a set of points in a metric space, the goal of $\textsf{MDMS}$ is to maximize $f(S) = g(S) + λ\cdot \texttt{div}(S)$ subject to a cardinality constraint $|S| \le k$, where $g(S)$ is a monotone submodular function and $\texttt{div}(S) = \min_{u,v \in S : u \ne v} \text{dist}(u,v)$ is the max-min diversity objective. We propose the $\texttt{GIST}$ algorithm, which gives a $\frac{1}{2}$-approximation guarantee for $\textsf{MDMS}$ by approximating a series of maximum independent set problems with a bicriteria greedy algorithm. We also prove that it is NP-hard to approximate within a factor of $0.5584$. Finally, we show in our empirical study that $\texttt{GIST}$ outperforms state-of-the-art benchmarks for a single-shot data sampling task on ImageNet.
