A Product Manifold Method for Feature Selection
Mao Li, Zhilong Mi, Yingpeng Du, Yifan Cao, Qingcai He, Ziqiao Yin, Binghui Guo, Zhu Sun
Abstract
Feature selection is indispensable for mitigating overfitting and reducing feature redundancy in high-dimensional scenarios. However, most existing approaches rely on unstable overall separability and sample-wise geometry, thereby neglecting the stable class-specific discriminative structures and leading to poor performance especially in small sample regimes. Although product manifolds provide a natural geometric framework to model distinctive manifolds for different classes, their potential in feature selection is largely unexplored, e.g., theoretical guarantee on spectral convergence. In this paper, we propose a novel supervised feature selection method named PRISM, which leverages product manifold to theoretically disentangle class-specific features from the shared structure. Grounded in spectral analysis on product manifolds, we explicitly model the feature space to distinguish between shared and class-specific structures. To disentangle these class-specific structures, we design a spectral filtering mechanism that suppresses the shared components and iteratively extracts class-specific latent variables. Based on these extracted variables, we establish a scoring mechanism that identifies features with both high discriminative power and strong class specificity in high-dimensional small-sample scenarios. Crucially, we bridge the theoretical gap in spectral analysis and provide a theoretical guarantee for our method by deriving an asymptotic convergence proof under the product manifold setting, guaranteeing the reliable isolation of class-specific discriminative structures. Comprehensive experiments demonstrate that PRISM not only improves generalization performance and robustness to small sample size over leading baselines, but also achieves superior result reusability when new classes emerge.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get 1a2d32dd-b556-42a4-a42e-a8ac72b33a92Related papers
- Few-Sample Feature Selection via Feature Manifold LearningDavid Cohen, Tal Shnitzer, Yuval Kluger, Ronen TalmonICML 2023 · 14 citations
- Feature Selection for Latent Factor ModelsRittwika Kansabanik, Adrian BarbuCVPR 2025
- Unsupervised Feature Selection Through Group DiscoveryShira Lifshitz, Ofir Lindenbaum, Gal Mishne, Ron Meir et al.AAAI 2026
- DiSC: Differential Spectral Clustering of FeaturesRam Dyuthi Sristi, Gal Mishne, Ariel JaffeNeurIPS 2022 · 8 citations
- Partition First, Embed Later: Laplacian-Based Feature Partitioning for Refined Embedding and Visualization of High-Dimensional DataErez Peterfreund, Ofir Lindenbaum, Yuval Kluger, Boris LandaICML 2025
