Graph Convolutional Network-based Feature Selection for High-dimensional and Low-sample Size Data

Abstract

Feature selection is a powerful dimension reduction technique which selects a subset of relevant features for model construction. Numerous feature selection methods have been proposed, but most of them fail under the high-dimensional and low-sample size (HDLSS) setting due to the challenge of overfitting. In this paper, we present a deep learning-based method - GRAph Convolutional nEtwork feature Selector (GRACES) - to select important features for HDLSS data. We demonstrate empirical evidence that GRACES outperforms other feature selection methods on both synthetic and real-world datasets.

24 pages, 4 figures, 4 tables

References in corpus (1)

• High-dimensional regression and variable selection using CAR scores

Cited by in corpus (2)

• Attention-based graph neural networks: a survey
• Subject-Adaptive Sparse Linear Models for Interpretable Personalized Health Prediction from Multimodal Lifelog Data