Title
A novel stability based feature selection framework for k-means clustering
Abstract
Stability of a learning algorithm with respect to small input perturbations is an important property, as it implies the derived models to be robust with respect to the presence of noisy features and/or data sample fluctuations. In this paper we explore the effect of stability optimization in the standard feature selection process for the continuous (PCA-based) k-means clustering problem. Interestingly, we derive that stability maximization naturally introduces a tradeoff between cluster separation and variance, leading to the selection of features that have a high cluster separation index that is not artificially inflated by the feature's variance. The proposed algorithmic setup is based on a Sparse PCA approach, that selects the features that maximize stability in a greedy fashion. In our study, we also analyze several properties of Sparse PCA relevant to stability that promote Sparse PCA as a viable feature selection mechanism for clustering. The practical relevance of the proposed method is demonstrated in the context of cancer research, where we consider the problem of detecting potential tumor biomarkers using microarray gene expression data. The application of our method to a leukemia dataset shows that the tradeoff between cluster separation and variance leads to the selection of features corresponding to important biomarker genes. Some of them have relative low variance and are not detected without the direct optimization of stability in Sparse PCA based k-means.
Year
DOI
Venue
2011
10.1007/978-3-642-23783-6_27
ECML/PKDD
Keywords
Field
DocType
k-means clustering,sparse pca approach,cluster separation,feature selection framework,high cluster separation index,viable feature selection mechanism,stability optimization,sparse pca,noisy feature,stability maximization,novel stability,standard feature selection process,relative low variance,k means,indexation,feature selection,k means clustering
k-means clustering,Data mining,Sparse PCA,Sample (statistics),Pattern recognition,Feature selection,Computer science,Tumor Biomarkers,Artificial intelligence,Microarray gene expression,Cluster analysis,Maximization
Conference
Volume
ISSN
Citations 
6912
0302-9743
1
PageRank 
References 
Authors
0.35
15
2
Name
Order
Citations
PageRank
Dimitrios Mavroeidis11309.50
Elena Marchiori21272164.66