Improving variable selection properties with data integration and transfer learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Rognon-Vael, Paul, Rossell, David, Zwiernik, Piotr
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911445492432896
author Rognon-Vael, Paul
Rossell, David
Zwiernik, Piotr
author_facet Rognon-Vael, Paul
Rossell, David
Zwiernik, Piotr
contents We study variable selection (also called support recovery) in high-dimensional sparse linear regression when one has external information on which variables are likely to be associated with the response. Consistent recovery is only possible under somewhat restrictive conditions on sample size, dimension, signal strength, and sparsity. We investigate how these conditions can be relaxed by incorporating said external information. A key application that we consider is structural transfer learning, where variables selected in one or more source datasets are used to guide variable selection in a target dataset. We introduce a family of likelihood penalties that depend on the external information, motivated by connections to Bayesian variable selection. We show that these methods achieve variable selection consistency in regimes where any method ignoring external information fails, and that they achieve consistency at faster rates. We first quantify the potential gains under ideal, oracle-chosen, penalties. We then propose computationally efficient empirical Bayes procedures that learn suitable penalties from the data. We prove that these procedures have improved variable selection properties compared to methods that do not use external information. We illustrate our approach using simulations and a genomics application, where results from mouse experiments are used to inform variable selection for gene expression data in humans.
format Preprint
id arxiv_https___arxiv_org_abs_2502_15584
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Improving variable selection properties with data integration and transfer learning
Rognon-Vael, Paul
Rossell, David
Zwiernik, Piotr
Statistics Theory
Methodology
62F07 (Primary) 62C12, 62R07 (Secondary)
We study variable selection (also called support recovery) in high-dimensional sparse linear regression when one has external information on which variables are likely to be associated with the response. Consistent recovery is only possible under somewhat restrictive conditions on sample size, dimension, signal strength, and sparsity. We investigate how these conditions can be relaxed by incorporating said external information. A key application that we consider is structural transfer learning, where variables selected in one or more source datasets are used to guide variable selection in a target dataset. We introduce a family of likelihood penalties that depend on the external information, motivated by connections to Bayesian variable selection. We show that these methods achieve variable selection consistency in regimes where any method ignoring external information fails, and that they achieve consistency at faster rates. We first quantify the potential gains under ideal, oracle-chosen, penalties. We then propose computationally efficient empirical Bayes procedures that learn suitable penalties from the data. We prove that these procedures have improved variable selection properties compared to methods that do not use external information. We illustrate our approach using simulations and a genomics application, where results from mouse experiments are used to inform variable selection for gene expression data in humans.
title Improving variable selection properties with data integration and transfer learning
topic Statistics Theory
Methodology
62F07 (Primary) 62C12, 62R07 (Secondary)
url https://arxiv.org/abs/2502.15584