i-IF-Learn: Iterative Feature Selection and Unsupervised Learning for High-Dimensional Complex Data

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ma, Chen, Wang, Wanjie, Fan, Shuhao
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908912559587328
author Ma, Chen
Wang, Wanjie
Fan, Shuhao
author_facet Ma, Chen
Wang, Wanjie
Fan, Shuhao
contents Unsupervised learning of high-dimensional data is challenging due to irrelevant or noisy features obscuring underlying structures. It's common that only a few features, called the influential features, meaningfully define the clusters. Recovering these influential features is helpful in data interpretation and clustering. We propose i-IF-Learn, an iterative unsupervised framework that jointly performs feature selection and clustering. Our core innovation is an adaptive feature selection statistic that effectively combines pseudo-label supervision with unsupervised signals, dynamically adjusting based on intermediate label reliability to mitigate error propagation common in iterative frameworks. Leveraging low-dimensional embeddings (PCA or Laplacian eigenmaps) followed by $k$-means, i-IF-Learn simultaneously outputs influential feature subset and clustering labels. Numerical experiments on gene microarray and single-cell RNA-seq datasets show that i-IF-Learn significantly surpasses classical and deep clustering baselines. Furthermore, using our selected influential features as preprocessing substantially enhances downstream deep models such as DeepCluster, UMAP, and VAE, highlighting the importance and effectiveness of targeted feature selection.
format Preprint
id arxiv_https___arxiv_org_abs_2603_24025
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle i-IF-Learn: Iterative Feature Selection and Unsupervised Learning for High-Dimensional Complex Data
Ma, Chen
Wang, Wanjie
Fan, Shuhao
Machine Learning
Methodology
Unsupervised learning of high-dimensional data is challenging due to irrelevant or noisy features obscuring underlying structures. It's common that only a few features, called the influential features, meaningfully define the clusters. Recovering these influential features is helpful in data interpretation and clustering. We propose i-IF-Learn, an iterative unsupervised framework that jointly performs feature selection and clustering. Our core innovation is an adaptive feature selection statistic that effectively combines pseudo-label supervision with unsupervised signals, dynamically adjusting based on intermediate label reliability to mitigate error propagation common in iterative frameworks. Leveraging low-dimensional embeddings (PCA or Laplacian eigenmaps) followed by $k$-means, i-IF-Learn simultaneously outputs influential feature subset and clustering labels. Numerical experiments on gene microarray and single-cell RNA-seq datasets show that i-IF-Learn significantly surpasses classical and deep clustering baselines. Furthermore, using our selected influential features as preprocessing substantially enhances downstream deep models such as DeepCluster, UMAP, and VAE, highlighting the importance and effectiveness of targeted feature selection.
title i-IF-Learn: Iterative Feature Selection and Unsupervised Learning for High-Dimensional Complex Data
topic Machine Learning
Methodology
url https://arxiv.org/abs/2603.24025