Heterogeneity-Preserving Discriminative Feature Selection for Subtype Discovery

Abstract

The identification of disease-specific subtypes can provide valuable insights into disease progression and potential individualized therapies, important aspects of precision medicine given the complex nature of disease heterogeneity. The advent of high-throughput technologies has enabled the generation and analysis of various molecular data types, such as single-cell RNA-seq, proteomic, and imaging datasets, on a large scale. While these datasets offer opportunities for subtype discovery, they also pose challenges in finding subtype signatures due to their high dimensionality. Feature selection, a key step in the machine learning pipeline, involves selecting signatures that reduce feature size for more efficient downstream computational analysis. Although many existing methods focus on selecting features that differentiate known diseases or cell states, they often struggle to identify features that both preserve heterogeneity and reveal subtypes. To address this, we utilized deep metric learning-based feature embedding to explore the statistical properties of features crucial for preserving heterogeneity. Our analysis indicated that features with a notable difference in interquartile range (IQR) between classes hold important subtype information. Guided by this insight, we developed a statistical method called PHet (Preserving Heterogeneity), which employs iterative subsampling and differential analysis of IQR combined with Fishers method to identify a small set of features that preserve heterogeneity and enhance subtype clustering quality. Validation on public single-cell RNA-seq and microarray datasets demonstrated PHets ability to maintain sample heterogeneity while distinguishing known disease/cell states, with a tendency to outperform previous differential expression and outlier-based methods. Furthermore, an analysis of a single-cell RNA-seq dataset from mouse tracheal epithelial cells identified two distinct basal cell subtypes differentiating towards a luminal secretory phenotype using PHet-based features, demonstrating promising results in a real-data application. These results highlight PHets potential to enhance our understanding of disease mechanisms and cell differentiation, contributing significantly to the field of personalized medicine.