In some supervised learning settings, the practitioner might have additional information on the features used for prediction. We propose a new method which leverages this additional information for better prediction. The method, which we call the feature-weighted elastic net ("fwelnet"), uses these "features of features" to adapt the relative penalties on the feature coefficients in the elastic net penalty. In our simulations, fwelnet outperforms the lasso in terms of test mean squared error and usually gives an improvement in true positive rate or false positive rate for feature selection. We also apply this method to early prediction of preeclampsia, where fwelnet outperforms the lasso in terms of 10-fold cross-validated area under the curve (0.86 vs. 0.80). We also provide a connection between fwelnet and the group lasso and suggest how fwelnet might be used for multi-task learning.
Description
Feature-weighted elastic net: using “features of features” for better prediction - PMC
%0 Journal Article
%1 tay2023featureweighted
%A Tay, J Kenneth
%A Aghaeepour, Nima
%A Hastie, Trevor
%A Tibshirani, Robert
%C China (Republic : 1949- )
%D 2023
%J Statistica Sinica
%K background bki ien informed integration knowledge learning linear machine ml regression
%N 1
%P 259--279
%R 10.5705/ss.202020.0226
%T Feature-weighted elastic net: using "features of features" for better prediction
%U https://pubmed.ncbi.nlm.nih.gov/37102071
%V 33
%X In some supervised learning settings, the practitioner might have additional information on the features used for prediction. We propose a new method which leverages this additional information for better prediction. The method, which we call the feature-weighted elastic net ("fwelnet"), uses these "features of features" to adapt the relative penalties on the feature coefficients in the elastic net penalty. In our simulations, fwelnet outperforms the lasso in terms of test mean squared error and usually gives an improvement in true positive rate or false positive rate for feature selection. We also apply this method to early prediction of preeclampsia, where fwelnet outperforms the lasso in terms of 10-fold cross-validated area under the curve (0.86 vs. 0.80). We also provide a connection between fwelnet and the group lasso and suggest how fwelnet might be used for multi-task learning.
@article{tay2023featureweighted,
abstract = {In some supervised learning settings, the practitioner might have additional information on the features used for prediction. We propose a new method which leverages this additional information for better prediction. The method, which we call the feature-weighted elastic net ("fwelnet"), uses these "features of features" to adapt the relative penalties on the feature coefficients in the elastic net penalty. In our simulations, fwelnet outperforms the lasso in terms of test mean squared error and usually gives an improvement in true positive rate or false positive rate for feature selection. We also apply this method to early prediction of preeclampsia, where fwelnet outperforms the lasso in terms of 10-fold cross-validated area under the curve (0.86 vs. 0.80). We also provide a connection between fwelnet and the group lasso and suggest how fwelnet might be used for multi-task learning.},
added-at = {2024-02-09T08:23:56.000+0100},
address = {China (Republic : 1949- )},
author = {Tay, J Kenneth and Aghaeepour, Nima and Hastie, Trevor and Tibshirani, Robert},
biburl = {https://www.bibsonomy.org/bibtex/25ffab3b3bc0829b1c389fe81d2eda406/becker},
comment = {37102071[pmid]
PMC10129060[pmcid]},
description = {Feature-weighted elastic net: using “features of features” for better prediction - PMC},
doi = {10.5705/ss.202020.0226},
interhash = {cda8f4ff69edc9c185e452ad639b2dfc},
intrahash = {5ffab3b3bc0829b1c389fe81d2eda406},
issn = {10170405},
journal = {Statistica Sinica},
keywords = {background bki ien informed integration knowledge learning linear machine ml regression},
month = jan,
number = 1,
pages = {259--279},
timestamp = {2024-02-09T08:23:56.000+0100},
title = {Feature-weighted elastic net: using "features of features" for better prediction},
url = {https://pubmed.ncbi.nlm.nih.gov/37102071},
volume = 33,
year = 2023
}