A Scalable Framework for Sparse Clustering Without Shrinkage

2020·arXiv

Abstract

Abstract

Clustering, a fundamental activity in unsupervised learning, is notoriously difficult when the feature space is high-dimensional. Fortunately, in many realistic scenarios, only a handful of features may be relevant in distinguishing clusters. This has motivated the development of sparse clustering techniques that typically rely on k-means within outer algorithms of high computational complexity. Current techniques also require careful tuning of shrinkage parameters, further limiting their scalability. In this paper, we propose a novel framework for sparse k-means clustering that is intuitive, simple to implement, and competitive with state-of-the-art algorithms. We show that our algorithm enjoys consistency and convergence guarantees. Our core method readily generalizes to several task-specific algorithms such as clustering on subsets of attributes and in partially observed data settings. We showcase these contributions thoroughly via simulated experiments and real data benchmarks, including a case study on protein expression in trisomic mice.

1 Introduction

Clustering is a ubiquitous task in unsupervised learning and exploratory data analysis. First proposed by Steinhaus [1], k-means clustering is one of the most widely used approaches owing to its conceptual simplicity, speed, and familiarity. The method aims to partition n data points into k clusters, with each data point assigned to the cluster with nearest center (mean). More recently, variants and extensions of k-means continue to be developed from a number of perspectives including Bayesian and nonparametric approaches [2–4], subspace clustering [5, 6], and continuous optimization [7–9]. After decades of refinements and generalizations, Lloyd’s algorithm [10] remains the most popular algorithm to perform k-means clustering.

Lloyd’s algorithm Consider a dataset comprised of n samples and p features. Without loss of generality, assume each feature is centered at the origin. To put features on the same footing, it is also recommended to standardize each of them to have unit variance [11]. Given a fixed number of k clusters, k-means assigns each row to a cluster label represented by optimal cluster centers that minimize the objective

Although the objective function is simple and intuitive, finding its optimal value is NP-hard, even when k = 2 or p = 2 [12–14]. A greedy algorithm, Lloyd’s k-means algorithm alternates between two steps until cluster assignments stabilize and no further changes occur in the objective function: 1) The algorithm assigns each row of X to the cluster whose center is closest. 2) It then redefines each center as the center of mass . While guaranteed to converge in finite iterations, Lloyd’s algorithm may stop short at a poor local minimum, rendering it extremely sensitive to the initial starting point. Many initialization methods have been proposed to ameliorate this defect [15–17]; the most popular scheme to date is called k-means++ [18, 19].

Feature selection and sparsity Despite its efficacy and widespread use, Lloyd’s k-means algorithm is known to deteriorate when the number of features p grows. In the presence of many uninformative features, the signal-to-noise ratio decreases, and the algorithm falls victim to poor local minima. In essence, the pairwise Euclidean distances appearing in the k-means objective (1) become less informative as p grows [20, 21]. One way of mitigating this curse of dimensionality is to cluster under a sparsity assumption. If only a subset of features distinguishes between clusters, then ignoring irrelevant features will not only alleviate computational and memory demands, but will also improve clustering accuracy. Feature selection may be desirable as an interpretable end in itself, shedding light on the underlying structure of the data.

Feature selection for clustering is an active area of research, with most of the well-known methods falling under the category of filter methods or wrapper models [22, 23]. Filter methods independently evaluate the clustering quality of the features without applying any clustering algorithm. Feature evaluation can be univariate [24] or multivariate [25]. Filter criteria include entropy-based distances [26] and Laplacian scores [27]. Wrapper model methods (also called hybrid methods) first select subsets of attributes with or without filtering, then evaluate clustering quality based on the candidate subsets, and finally select the subset giving to the highest clustering quality. This process can be iterated until a desired clustering quality is met. The wrapper method of Dy and Brodley [28] uses maximum likelihood criteria and mixture of Gaussians as the base clustering method. A variation of k-means called Clustering on Subsets of Attributes (COSA) [29] performs feature weighing to allow different clusters to be determined based by different sets of features. As formulated, COSA does not lead to sparse solutions.

Expanding on COSA, Witten and Tibshirani [30] proposed a framework for feature selection that can result in feature-sparse centers. Their sparse k-means method (SKM) aims to maximize the objective

with respect to the clusters and the feature weights subject to the constraints , , and for all m. Here denotes a dissimilarity measure between and , the most natural choice being on the weight vector w ensures a sparse selection of the features, while the gives more smoothness to the solution. Without smoothness, often only one component of w is estimated as positive. Constrained maximization of the objective (2) is accomplished by block descent: holding w fixed, the objective is optimized with respect to the clusters by the k-means tactic of assigning to the cluster with the closest center. In this context the weighted distance is operative. Holding the fixed, the objective is then optimized with respect to w. To handle outliers, Kondo et al. [31] propose trimming observations, and Brodinová et al. [32] assigns weights to observations. Such approaches with convex norm penalties or constraints on feature weights are popular since the work of Huang et al. [33], and remain active areas of research [34, 35].

Proposed contributions We propose a novel method that incorporates feature selection within Lloyd’s algorithm by adding a sorting step. This simple adjustment preserves the algorithm’s speed and scalability while addressing sparsity directly. Called Sparse k-Means with Feature Ranking (SKFR), the method forces all cluster centers to fall in the s-sparse ball with nonzero entries occurring along a shared set of feature indices. One may interpret the algorithm as switching between learning the most informative set of features in via ranking the components of according to their reduction of the loss, and then projecting onto this shared sparsity set at each iteration. Computing the projection is simple, and amounts to truncating all but the top s ranked components to 0. Importantly—and perhaps surprisingly—we will show that this ranking criterion can be computed without making a second pass through the data.

This feature ranking and truncation yields interpretable sparsity in , in contrast to penalty approaches that promote sparsity by proxy. The latter are known to introduce unwanted shrinkage, and may not always yield sparse solutions. The role of “interpretability" here is twofold: first at the input stage, the parameter s directly informs and controls the sparsity level, and is thus directly interpretable in contrast to tuning parameters in shrinkage approaches. Second, SKFR selects relevant features among the original dimensions, thus allowing them to retain their original interpretations in that feature space . For instance, our case study on mouse protein expression not only clusters the data but identifies the most relevant genes in expression space for predicting trisomy. In this sense the dimension reduction is interpretable, in contrast to generic dimension reduction such as PCA, where the axes (principal components) in the projected space no longer can be interpreted as, say, genes. The simplicity of the SKFR framework leads to straightforward modifications that extend to many salient data settings, allowing the method to accommodate for missing values, outliers, and cluster specific sparsity sets.

The rest of the paper is organized as follows. The proposed algorithms are detailed in Section 2, and its properties are discussed in Section 2.1. We demonstrate the flexibility of our core routine by deriving various extensions in Section 3. The methods are validated empirically via a suite of simulation studies and real data in Sections 4 and 5, followed by discussion.

2 Sparse k-means with feature ranking

We present two versions of the SKFR algorithm, which both begin by standardizing each feature to have zero mean and unit variance. The first version, which we will refer to as SKFR1, chooses an optimal set of s globally informative features based on a ranking criterion defined below. Each informative component of cluster j is then set equal to the within-cluster mean and the remaining uninformative features are left at zero.

A second local version SKFR2 will allow for the set of relevant features to vary across clusters: the ranking of top s features is performed for each cluster—akin to the COSA algorithm [29]—to permit richer distinctions between clusters. Within each cluster j, the s informative components are again updated by setting with remaining uninformative components left at zero. Both versions of SKFR alternate cluster reassignment with cluster re-centering. In cluster reassignment, each sample is assigned to the cluster with closest center, exactly as in Lloyd’s algorithm .

The choice of top features are determined based on the amount they affect the k-means objective. For SKFR1 (the global version), the difference

measures the reduction of the objective as feature l passes from uninformative to informative status. The final equality in (3) shows that can be written in a form that does not involve , which suggests that it can be computed only in terms of the current means and their relative sizes. In particular, it suggests we do not require taking a second pass through the data to perform the ranking step. The s largest differences identify the informative features. Alternatively, one can view the selection process as a projection to a sparsity set as mentioned previously. This simple observation provides the foundation to the transparent and efficient Algorithm 1.

In the local version SKFR2, the choice of informative features for cluster j depends on the differences

The s largest differences identify the informative features for cluster j. These informative features may or may not be shared with other clusters. The pseudocode summary appears as Algorithm 2.

As we detail in the next section, both versions of SKFR enjoy a descent property and are guaranteed to reduce the k-means objective at each iteration. Both versions dynamically choose informative features, while retaining the greedy nature and low O(npk) computational complexity of Lloyd’s algorithm. Like Lloyd’s algorithm, neither version is guaranteed to converge to a global minimum of the objective, though they are both guaranteed to converge in finite iterations.

We argue our framework is more straightforward and interpretable than existing alternatives. For instance, the SKM algorithm entails solving constrained subproblems with an inner bisection algorithm at each step to select dual parameters, and encourages sparsity through a less interpretable tuning constant [30]. The extensions for robust versions of such methods, for instance trimmed versions, entail even more computational overhead. In contrast, as we see in Section 3, robustification and other modifications accounting for specific data settings will only require natural and simple modifications of the core SKFR algorithm.

2.1 Properties

Here we discuss several theoretical properties of the algorithm. Complete proofs are detailed in the Supplement. We first observe that in the extreme case when s = p, our algorithm reduces to Lloyd’s original k-means algorithm. As the sorting step in ranking is only logarithmic in cost, it is straightforward to see that our algorithm enjoys the same O(npk) complexity overall.

Monotonicity of the objective To establish convergence of the algorithm, it suffices to prove that it possesses a descent property as the objective is bounded below. Monotonicity also lends itself to stability of the algorithm in practice.

Proposition 1. Each iteration of Algorithm 1 and Algorithm 2 monotonically decreases the k-means objective function

Strong consistency of centroids We additionally establish convergence in the statistical sense in that our method enjoys strong consistency in terms of the optimal centroids. In particular, Proposition 2 shows that our method inherits a property of standard k-means method established by Pollard [36]. Let be independently sampled from a distribution P, and let denote a set of k points. Define the population-level loss function

Let denote the minimizer of be the minimizer of empirical measure. The following proposition establishes strong consistency under an identifiability assumption and a selection consistency assumption.

Proposition 2. Assume that for any neighborhood , there exists for every . Then if eventually lie in the same dimension as (i.e., the nonzero feature indices of agree with

We remark that while the identifiability of is a mild condition, consistency of cluster centers relies on the top s features to be identified correctly. The latter is a considerably strong assumption, and it will be of interest to investigate conditions in the data that allow us to relax the assumption.

3 Extensions to SKFR

The simplicity of the SKFR framework allows for straightforward, modular extensions of our two basic algorithms. In the scenarios considered below, existing variations on k-means often require complex adjustments that must be derived on a case-by-case basis. Our simple core routine applies to a variety of common situations in data analysis.

Missing data Missing data are common in many real-world data applications. To recover a complete dataset amenable to traditional clustering, practitioners typically impute missing features or delete incomplete samples in a preprocessing step. Imputation can be computationally expensive and potentially erroneous. Deleting samples with missing features risks losing valuable information and is ill advised when the proportion of samples with missing data is high. Chi et al. [7] proposed a simple alternative called k-pod that does not rely on additional tuning parameters or assumptions on the pattern of missingness. Let denote the index set corresponding to the observed entries of the data matrix X. The k-means objective is rephrased here as

Their strategy invokes the majorization-minimization (MM) principle [37], which involves majorizing the objective by a surrogate function at each iteration and then minimizing the surrogate. This action provably drives the objective downhill [38]. At iteration n define by for and by for and assigned to cluster j. The surrogate function is then defined by In other words, each missing is filled in by the current best guess of its value. Because our sparse clustering framework shares the same skeleton as k-means, exactly the same imputation technique carries over. Every iteration under imputation then makes progress in reducing the objective, preserving convergence guarantees.

Outliers One popular way to deal with outlying data is via data trimming, but due to its additional computational overhead, this approach becomes quickly limited to only moderate dimensional settings [31, 32]. A viable alternative to handle outliers is to replace squared Euclidean norms by a robust alternative. The norm is such a choice, which entails replacing the within-cluster means of SKFR1 by within-cluster medians, and the global distances (3) by

An analogous expression holds for the local version of SKFR. Observe that the revised distances lose some of the algebraic simplicity of the original distances; now we must pass through the data to compute . Additionally, the user may choose not to standardize the data with respect to squared distance beforehand to further reduce outlier effects [39]. The overall algorithm remains identical to SKFR up to this substitution for

Choice of sparsity level In contrast to penalized methods, SKFR deals with sparsity directly through an interpretable sparsity level s. If one desires a particular level s or s is known in advance, then it does not require tuning. In sparse k-means (SKM) for instance [30], prior information does not directly inform the choice of the continuous shrinkage . When the sparsity level s must be learned, one can capitalize on a variant of the gap statistic [40] to find an optimal value of s. Witten and Tibshirani [30] suggest a permutation test and a score closely related to the original gap statistic to tune the bound parameter on the feature weight vector. Here we adopt a similar approach, employing the gap statistic which depends on the difference

between the total sum of squares and the sums of the within-cluster sum of squares at the optimal cluster. One now randomly permutes the observations within each column of X to obtain B independent datasets . Finally, the parameter s is chosen to maximize

To illustrate the performance of the permutation test, summarized in Algorithm 3, we simulated two datasets with n = 400 samples and k = 10 equally sized clusters. There are s = 15 informative featuures in each, while the first dataset has an ambient dimension of p = 50 while the second involves only p = 20 total features. These settings contrast sparse and dense feature-driven data; a detailed description of the simulation is described in Section 4. The true s can be recovered in both settings, with gap statistics plotted in Figure 1.

(a) Gap statistic when the number of noisy variables is much greater than the number of informative variables.

Figure 1: Gap statistics in sparse and dense settings. The ground truth s = 15, labeled by the red circle in both plots, is selected by the permutation test in both settings.

4 Performance and simulation study

In this section, we use simulated experiments to validate SKFR and its extensions and to compare them to competing peer algorithms. We consider both locally and globally informative features to assess both variants, and then showcase SKFR’s adaptability in handling missing data. In all simulations, the number of informative features s is chosen to be 10, and we explore a range of sparsity levels by varying the total number of features variant and all the competing algorithms are seeded by the k-means++ initialization scheme [18]. We use the adjusted Rand index (ARI) [41, 42] to quantify the performance of the algorithms. These values are plotted for ease of visual comparison, while detailed numerical ARI results as well as analogous results in terms of normalized variation of information (NVI) [43] are tabulated in the Supplement. Varying the dimension p from 20 to 1000, we run 30 trials with 20 restarts per trial. We tune SKM’s bound parameter over the range [2, 10] by the gap statistic.

Sparse clustering test In this experiment, we test SKFR1 against Lloyd’s algorithm and SKM. The latter algorithm is implemented in the R package sparcl. We follow the simulation setup of Brodinová et al. [32], with details in the Supplement.

Figure 2: Median ARI with (left) and without (right) outliers.

Figure 3: Subsets of discriminative features (left) and presence of missing data (right)

Figure 2 plots median ARI values over 30 trials under each simulation setting for the three algorithms. It is evident from the plot that Lloyd’s k-means algorithm eventually deteriorates as the dimension of the dataset and number of noisy features increase. Both SKFR and SKM perform well without outliers, illustrating the utility of sparse clustering in high-dimensional settings.

In terms of feature selection, SKFR maintains both false positive and false negative median rates oof 0 over 30 trials under all 6 experimental settings. In other words SKFR is able to select the 10 informative features exactly and consistently. For SKM, the entries of the weight vector w corresponding to the informative features have higher values than uninformative entries at convergence. However, SKM does not consistently deliver sparse solutions. For example, at convergence all entries of the weight vector w are non-zero for the p = 20 setting, while for the p = 1000 setting over 400 entries are non-zero. Therefore, we find SKFR preferable to SKM in terms of feature selection.

Table 1: Runtime in seconds of SKFR1 and Lloyd’s k-means, with (iteration count) below

Table 2: False Positive Rate of SKFR1 and SKM, with (false negative rate for SKM) below

This experiment also allows us to examine runtime and iteration counts for SKFR and standard k- means. Table 1 tabulates our results obtained in a Julia 1.1 implementation [44]. For each simulation setting, the table shows average results over 20 runs. In almost all cases, SKFR requires much less run time. For p = 1000, k-means terminates first, but is unable to handle the high number of noisy features. In such cases, iterations until convergence can be misleading as k-means clearly stops short at a poor local minimum, as evident from Figure 2. Since methods such as SKM are of much higher computational complexity, as they call k-means as a subroutine per iteration, they are omitted from the timing comparison. Our empirical study emphasizes that SKFR stands out in terms of runtime—not only is it on par with k-means in terms of complexity, but in practice may converge faster.

Robustness against outliers To test the robustness of SKFR to outliers, we adopt our earlier simulation settings and replace a modicum of the observations in each class by outliers, again following a design by Brodinová et al. [32] with details in the Supplement. For this experiment, we again generate k = 10 equally sized classes and a total of n = 400 observations. We compare SKFR1 to SKM, as well as the geometric k-median algorithm [45], which is designed to be robust to outliers compared to k-means. The latter algorithm is available in R in the Gmedian package. We generate 30 trials for each setting and use 20 restarts for each algorithm.

Median ARI values under outliers are plotted in the right panel of Figure 2. The geometric k-median algorithm deteriorates severely in the high dimensional cases, where there are many more noise features than informative features. To show SKFR’s feature selection capability in the presence of outliers, we tabulate median false positive and false negative rates. Since SKM did not typically result in a sparse solution, we set a threshold and choose weight components greater than 0.1 to be selected features. In all outlier settings, SKFR1 is noticeably superior to SKM in feature selection accuracy.

Clustering on different subsets of attributes We next consider simulated data with different sets of distinguishing attributes for different underlying classes. We generate n = 250 samples spread unevenly over k = 5 classes. The s informative features of samples from each class are multivariate standard normal samples whose centers are initialized in the hypercube with entries for , and the uninformative features are sampled from N(0, 3). The set of informative features of each cluster is chosen by independently sampling a subset of size s from all the p dimensions. This simulation setting is modeled after a design that has appeared through numerous previous studies of k-means [46–48]. We compare the local version SKFR2 to Lloyd’s k-means algorithm and SKM under the same levels of sparsity as the previous experiments. The median ARI results of the algorithms in this setting are plotted in the left panel of Figure 3.

Though SKM outperforms Lloyd’s k-means significantly, we observe that our SKFR2 method performs at least on par with SKM, and begins to outperform when p is large enough, despite its significantly lower computational cost.

Partially observed data This experiment follows a similar data generation procedure with 10 informative features shared across all classes. n = 250 samples are generated and spread unevenly over k = 5 classes at random. The informative features are again multivariate standard normal samples with centers , while the uninformative features are sampled from N(0, 1.5). To simulate missingness, 10% of the entries of are replaced by NA values uniformly at random. We compare SKFR1 with imputation under the k-pod algorithm as implemented in the kpodclustr R package. For each setting we run 30 trials, and report the best of 10 random restarts for each algorithm. The median ARI results of the 30 trials for both algorithms are plotted in Figure 3, right panel. As in the complete data scenarios, SKFR remains competitive and actually outperforms k-pod for high-dimensional cases in the presence of noisy data.

5 Case study and real data benchmarks

Mice protein data Having validated the algorithm on synthetic data, we begin our real data analysis by examining a mice protein expression dataset from a study of murine Down Syndrome [49]. The dataset consists of 1080 expression level measurements on 77 proteins measured in the cerebral cortex. There are 8 classes of mice (control versus trisomic, stimulated to learn versus not stimulated, injected with saline versus not injected). In the original paper, Higuera et al. [49] use self-organizing maps (SOMs) to discriminate subsets of the 77 proteins. The authors employ SOMs of size organize the control mice into 49 nodes. For classifying trisomic mice, the size of SOMs is chosen to be . In our treatment of the data, we use SKFR1 to cluster the control mice and the trisomic mice into 49 and 36 groups, respectively, and compare our clustering results with those obtained using Lloyd’s algorithm, the SOMs from the original analysis, and the power k-means algorithm which was recently used to reanalyze this dataset [48]. We evaluate clustering quality following measures proposed by Higuera et al. [49]: number of mixed class nodes, which result from assigning mice from more than one class, and the total number of mice in mixed-class nodes. All methods are seeded with k-means++ over 20 initializations, and the sparsity level for SKFR1 is chosen to be s = 24 as tuned via the gap statistic for both the control and trisomic mice data. The classification results are displayed in Table 3, showing that SKFR typically achieves a noticeably higher clustering quality than its competitors.

In contrast to the other methods, our algorithm not only assigns labels to the mice, but also produces interpretable output informing us of the most discriminating proteins. Out of the 24 selected proteins for the control group, 18 proteins are consistent with those identified by Higuera et al. [49]; for the trisomic mice group, 20 proteins match those from the original study. The complete lists of informative proteins selected by SKFR and further details of our analysis appear in the Supplement.

Table 3: Protein expression level clustering quality from a mouse trisomy learning study; a lower number of mixed nodes indicates better performance [49].

Benchmark datasets To further validate our proposed algorithm, we compare SKFR1 to widely used peer algorithms on 10 benchmark datasets collected from the Keel, ASU, and UCI machine learning repositories. A description of each dataset is given in the Supplement. Our algorithm competes with Lloyd’s k-means algorithm, sparse k-means algorithm (SKM) from the R package sparcl, and entropy-weighted k-means [5] from the R package wskm. While wskm is not a sparse clustering method per se, it is a popular approach that assigns weights to feature relevance, and we do not know whether exact sparsity holds in these real data scenarios. On each example, we run 20 independent trials, in which each competing algorithm is given matched k-means++ initial seedings for fair comparison. Both SKM and SKFR1 are tuned using the gap statistic, and we run wskm using its default settings. We evaluate the performance of the algorithms on each dataset with normalized mutual information (NMI) [43] and report the results in Table 4. The performance evaluated by ARI is given in the Supplement. SKFR1 gives the best result in terms of NMI for 9 of the 10 datasets.

6 Discussion

We present a novel framework (SKFR) for sparse k-means clustering with an emphasis on simplicity and scalability. In particular, SKFR does not rely on global shrinkage or penalization. Because its ranking phase is very quick, SKFR preserves the low time-complexity of Lloyd’s algorithm. Further, it inherits properties such as monotonicity, finite-time convergence, and consistency. As noted previously, feature selection consistency is an interesting avenue for further theoretical investigation. In the context of optimization, if we treat the centroids as the optimization variables, then both the optimization variables and the set of nonsparse features are being greedily optimized in each iteration. In contrast, in existing optimization techniques such as projected gradient descent and iterative hard thresholding, the greedy optimization operation is applied to the optimization variable itself, emphasizing the novelty of our proposed framework.

We find empirically that SKFR is competitive with alternatives in sparse clustering despite incurring (often significantly) lower computational overhead. In contrast to its competitors, SKFR is highly scalable and modular. As demonstrated in Section 4, its extensions gracefully handle data settings including missing values and outliers. Finally, SKFR requires just one hyperparameter, the sparsity level s, which can be tuned using the gap statistic or cross-validation. In some cases, a desired or known level s may be specified in advance. As parameter tuning in unsupervised settings is often costly and may suffer from instability, our approach can be quite advantageous. In contrast to applying generic dimension reduction before clustering, for instance pre-processing using principal component analysis (PCA), our approach also selects features that are interpretable in the original space—for instance, in identifying relevant genes—while PC space fails to preserve this interpretability.

The core idea of ranking and truncating relevant features can be viewed as a special case of clustering under a general set constraint on the cluster centers. Here the set should be closed, but alternative constraints rather than the sparse ball may be relevant. Further exploration of this principle is warranted since it may be helpful in designing useful extensions and has proven successful for related estimation tasks [50–52]. In the cluster center recalculation step, for each cluster the loss can be improved by minimizing the criterion

where d is a constant irrelevant to the minimization process. The solution

can be phrased in terms of the center of mass of the data points and the projection operator taking a point y to the closest point in T. Many projection operators can be computed by explicit formulas or highly efficient algorithms [53, 54], while the same principle may be employed for clustering under more general divergences such as Bregman divergences [55]. The notion of projection carries over and remains computationally tractable for geometrically simple yet relevant sets such as box constraints or halfspaces [9]. Thus, our novel framework and algorithmic primitive for sparse clustering apply to a broad class of constrained clustering problems. In our view, this is a fruitful avenue for future research.

Broader Impact

Our work contributes theoretically and computationally to unsupervised learning and feature selection. Because we simply provide a more scalable and transparent algorithm for addressing a classical task, our work brings no immediate ethical or societal concerns. Since our work focuses on improving clustering accuracy and preserving the interpretability and identifiability of features during and after the clustering process, it can help to reduce spurious conclusions drawn from off-the-shelf methods that either lack interpretability or are more likely to produce poor solutions. Indeed, as many scientific problems with societal impacts such as discovering gene associations for rare diseases or risk factors in precision medicine can be partially addressed through the lens of unsupervised learning, our work can positively impact these areas with wide implications. While our method does not leverage any bias in data in any way, misinterpreting or having too much confidence in solutions produced by our method on highly noisy, biased, or imbalanced data can potentially yield to inaccurate interpretations. We advise all users to understand the assumptions behind our method and the limitations of unsupervised learning, especially in challenging high-dimensional settings, when basing decisions off of the results of any machine learning approach such as that we propose here.

Acknowledgments and Disclosure of Funding

This work was partially supported by NSF DMS 2030355, NHGRI HG006139 and NIH GM053275. We thank the anonymous referees and Saptarshi Chakraborty for their constructive comments and discussion.

References

[1] Hugo Steinhaus. Sur la division des corp materiels en parties. Bull. Acad. Polon. Sci, 1(804): 801, 1956.

[2] Katherine A Heller and Zoubin Ghahramani. Bayesian hierarchical clustering. In Proceedings of the 22nd international conference on Machine learning, pages 297–304, 2005.

[3] Brian Kulis and Michael I Jordan. Revisiting k-means: New algorithms via bayesian nonparametrics. arXiv preprint arXiv:1111.0352, 2011.

[4] Briana JK Stephenson, Amy H Herring, and Andrew Olshan. Robust clustering with subpopulation-specific deviations. Journal of the American Statistical Association, 115(530): 521–537, 2020.

[5] Liping Jing, Michael K Ng, and Joshua Zhexue Huang. An entropy weighting k-means algorithm for subspace clustering of high-dimensional sparse data. IEEE Transactions on knowledge and data engineering, 19(8):1026–1041, 2007.

[6] René Vidal. Subspace clustering. IEEE Signal Processing Magazine, 28(2):52–68, 2011.

[7] Jocelyn T Chi, Eric C Chi, and Richard G Baraniuk. k-pod: A method for k-means clustering of missing data. The American Statistician, 70(1):91–99, 2016.

[8] Sohil Atul Shah and Vladlen Koltun. Robust continuous clustering. Proceedings of the National Academy of Sciences, 114(37):9814–9819, 2017.

[9] Jason Xu, Eric C Chi, Meng Yang, and Kenneth Lange. A majorization–minimization algorithm for split feasibility problems. Computational Optimization and Applications, 71(3):795–828, 2018.

[10] Stuart Lloyd. Least squares quantization in PCM. IEEE Transactions on Information Theory, 28(2):129–137, 1982.

[11] Ismail Bin Mohamad and Dauda Usman. Standardization and its effects on k-means clustering algorithm. Research Journal of Applied Sciences, Engineering and Technology, 6(17):3299– 3303, 2013.

[12] Sanjoy Dasgupta. The hardness of k-means clustering. Department of Computer Science and Engineering, University of California, 2008.

[13] Andrea Vattani. The hardness of k-means clustering in the plane. Manuscript, accessible at http://cseweb. ucsd. edu/avattani/papers/kmeans_hardness. pdf, 617, 2009.

[14] Meena Mahajan, Prajakta Nimbhorkar, and Kasturi Varadarajan. The planar k-means problem is NP-hard. Theoretical Computer Science, 442:13–21, 2012.

[15] Paul S Bradley and Usama M Fayyad. Refining initial points for k-means clustering. In ICML, volume 98, pages 91–99. Citeseer, 1998.

[16] Mohammad Al Hasan, Vineet Chaoji, Saeed Salem, and Mohammed J Zaki. Robust partitional clustering by outlier and density insensitive seeding. Pattern Recognition Letters, 30(11): 994–1002, 2009.

[17] M Emre Celebi, Hassan A Kingravi, and Patricio A Vela. A comparative study of efficient initialization methods for the k-means clustering algorithm. Expert Systems with Applications, 40(1):200–210, 2013.

[18] David Arthur and Sergei Vassilvitskii. k-means++: The advantages of careful seeding. In Proceedings of the eighteenth annual ACM-SIAM symposium on Discrete algorithms, pages 1027–1035. Society for Industrial and Applied Mathematics, 2007.

[19] Rafail Ostrovsky, Yuval Rabani, Leonard J Schulman, and Chaitanya Swamy. The effectiveness of Lloyd-type methods for the k-means problem. Journal of the ACM (JACM), 59(6):28, 2012.

[20] Kevin Beyer, Jonathan Goldstein, Raghu Ramakrishnan, and Uri Shaft. When is “nearest neighbor” meaningful? In International Conference on Database Theory, pages 217–235. Springer, 1999.

[21] Charu C Aggarwal, Alexander Hinneburg, and Daniel A Keim. On the surprising behavior of distance metrics in high dimensional space. In International Conference on Database Theory, pages 420–434. Springer, 2001.

[22] Ron Kohavi and George H John. Wrappers for feature subset selection. Artificial Intelligence, 97(1-2):273–324, 1997.

[23] Salem Alelyani, Jiliang Tang, and Huan Liu. Feature selection for clustering: A review. In Data Clustering, pages 29–60. Chapman and Hall/CRC, 2018.

[24] Zheng Zhao and Huan Liu. Spectral feature selection for supervised and unsupervised learning. In Proceedings of the 24th International Conference on Machine Learning, pages 1151–1157. ACM, 2007.

[25] Zheng Alan Zhao and Huan Liu. Spectral Feature Selection for Data Mining (Open Access). Chapman and Hall/CRC, 2011.

[26] Luis Talavera. Feature selection as a preprocessing step for hierarchical clustering. In ICML, volume 99, pages 389–397. Citeseer, 1999.

[27] Xiaofei He, Deng Cai, and Partha Niyogi. Laplacian score for feature selection. In Advances in Neural Information Processing Systems, pages 507–514, 2006.

[28] Jennifer G Dy and Carla E Brodley. Feature selection for unsupervised learning. Journal of Machine Learning Research, 5(Aug):845–889, 2004.

[29] Jerome H Friedman and Jacqueline J Meulman. Clustering objects on subsets of attributes (with discussion). Journal of the Royal Statistical Society: Series B (Statistical Methodology), 66(4): 815–849, 2004.

[30] Daniela M Witten and Robert Tibshirani. A framework for feature selection in clustering. Journal of the American Statistical Association, 105(490):713–726, 2010.

[31] Yumi Kondo, Matias Salibian-Barrera, and Ruben Zamar. A robust and sparse k-means clustering algorithm. arXiv preprint arXiv:1201.6082, 2012.

[32] Šárka Brodinová, Peter Filzmoser, Thomas Ortner, Christian Breiteneder, and Maia Rohm. Robust and sparse k-means clustering for high-dimensional data. Advances in Data Analysis and Classification, pages 1–28, 2017.

[33] Joshua Zhexue Huang, Michael K Ng, Hongqiang Rong, and Zichen Li. Automated variable weighting in k-means type clustering. IEEE Transactions on Pattern Analysis and Machine Intelligence, 27(5):657–668, 2005.

[34] Saptarshi Chakraborty, Debolina Paul, Swagatam Das, and Jason Xu. Entropy weighted power k-means clustering. In International Conference on Artificial Intelligence and Statistics, pages 691–701. PMLR, 2020.

[35] Saptarshi Chakraborty and Jason Xu. Biconvex clustering. arXiv preprint arXiv:2008.01760, 2020.

[36] David Pollard. Strong consistency of k-means clustering. The Annals of Statistics, pages 135–140, 1981.

[37] Mark P Becker, Ilsoon Yang, and Kenneth Lange. EM algorithms without missing data. Statistical Methods in Medical Research, 6(1):38–54, 1997.

[38] Kenneth Lange. MM Optimization Algorithms, volume 147. SIAM, 2016.

[39] Minjie Wang and Genevera I Allen. Integrative generalized convex clustering optimization and feature selection for mixed multi-view data. arXiv preprint arXiv:1912.05449, 2019.

[40] Robert Tibshirani, Guenther Walther, and Trevor Hastie. Estimating the number of clusters in a data set via the gap statistic. Journal of the Royal Statistical Society: Series B (Statistical Methodology), 63(2):411–423, 2001.

[41] William M Rand. Objective criteria for the evaluation of clustering methods. Journal of the American Statistical association, 66(336):846–850, 1971.

[42] Nguyen Xuan Vinh, Julien Epps, and James Bailey. Information theoretic measures for clusterings comparison: is a correction for chance necessary? In Proceedings of the 26th Annual International Conference on Machine Learning, pages 1073–1080. ACM, 2009.

[43] Nguyen Xuan Vinh, Julien Epps, and James Bailey. Information theoretic measures for clusterings comparison: Variants, properties, normalization and correction for chance. Journal of Machine Learning Research, 11(Oct):2837–2854, 2010.

[44] Jeff Bezanson, Alan Edelman, Stefan Karpinski, and Viral B Shah. Julia: A fresh approach to numerical computing. SIAM Review, 59(1):65–98, 2017.

[45] Hervé Cardot, Peggy Cénac, and Jean-Marie Monnez. A fast and recursive algorithm for clustering large datasets with k-medians. Computational Statistics & Data Analysis, 56(6): 1434–1449, 2012.

[46] Bin Zhang. Generalized k-harmonic means–dynamic weighting of data in unsupervised learning. In Proceedings of the 2001 SIAM International Conference on Data Mining, pages 1–13. SIAM, 2001.

[47] Greg Hamerly and Charles Elkan. Alternatives to the k-means algorithm that find better clusterings. In Proceedings of the Eleventh International Conference on Information and Knowledge Management, pages 600–607. ACM, 2002.

[48] Jason Xu and Kenneth Lange. Power k-means Clustering. In Proceedings of the 36th International Conference on Machine Learning, volume 97 of Proceedings of Machine Learning Research, pages 6921–6931, Long Beach, California, USA, 09–15 Jun 2019. PMLR.

[49] Clara Higuera, Katheleen J Gardiner, and Krzysztof J Cios. Self-organizing feature maps identify proteins critical to learning in a mouse model of down syndrome. PloS One, 10(6): e0129126, 2015.

[50] Eric C Chi, Hua Zhou, and Kenneth Lange. Distance majorization and its applications. Mathematical programming, 146(1-2):409–436, 2014.

[51] Jason Xu, Eric Chi, and Kenneth Lange. Generalized linear model regression under distance-to-set penalties. In Advances in Neural Information Processing Systems, pages 1385–1395, 2017.

[52] Kevin L Keys, Hua Zhou, and Kenneth Lange. Proximal distance algorithms: Theory and practice. J. Mach. Learn. Res., 20(66):1–38, 2019.

[53] Heinz H Bauschke and Patrick L Combettes. Convex Analysis and Monotone Operator Theory in Hilbert Spaces, volume 408. Springer, 2011.

[54] Amir Beck. First-order Methods in Optimization. SIAM, 2017.

[55] Arindam Banerjee, Srujana Merugu, Inderjit S Dhillon, and Joydeep Ghosh. Clustering with Bregman divergences. Journal of machine learning research, 6(Oct):1705–1749, 2005.

Designed for Accessibility and to further Open Science