Nonparametric variable importance using an augmented neural network with multi-task learning

Jean Feng, Brian Williamson, Noah Simon, Marco Carone
Proceedings of the 35th International Conference on Machine Learning, PMLR 80:1496-1505, 2018.

Abstract

In predictive modeling applications, it is often of interest to determine the relative contribution of subsets of features in explaining the variability of an outcome. It is useful to consider this variable importance as a function of the unknown, underlying data-generating mechanism rather than the specific predictive algorithm used to fit the data. In this paper, we connect these ideas in nonparametric variable importance to machine learning, and provide a method for efficient estimation of variable importance when building a predictive model using a neural network. We show how a single augmented neural network with multi-task learning simultaneously estimates the importance of many feature subsets, improving on previous procedures for estimating importance. We demonstrate on simulated data that our method is both accurate and computationally efficient, and apply our method to both a study of heart disease and for predicting mortality in ICU patients.

Cite this Paper


BibTeX
@InProceedings{pmlr-v80-feng18a, title = {Nonparametric variable importance using an augmented neural network with multi-task learning}, author = {Feng, Jean and Williamson, Brian and Simon, Noah and Carone, Marco}, booktitle = {Proceedings of the 35th International Conference on Machine Learning}, pages = {1496--1505}, year = {2018}, editor = {Dy, Jennifer and Krause, Andreas}, volume = {80}, series = {Proceedings of Machine Learning Research}, month = {10--15 Jul}, publisher = {PMLR}, pdf = {http://proceedings.mlr.press/v80/feng18a/feng18a.pdf}, url = {https://proceedings.mlr.press/v80/feng18a.html}, abstract = {In predictive modeling applications, it is often of interest to determine the relative contribution of subsets of features in explaining the variability of an outcome. It is useful to consider this variable importance as a function of the unknown, underlying data-generating mechanism rather than the specific predictive algorithm used to fit the data. In this paper, we connect these ideas in nonparametric variable importance to machine learning, and provide a method for efficient estimation of variable importance when building a predictive model using a neural network. We show how a single augmented neural network with multi-task learning simultaneously estimates the importance of many feature subsets, improving on previous procedures for estimating importance. We demonstrate on simulated data that our method is both accurate and computationally efficient, and apply our method to both a study of heart disease and for predicting mortality in ICU patients.} }
Endnote
%0 Conference Paper %T Nonparametric variable importance using an augmented neural network with multi-task learning %A Jean Feng %A Brian Williamson %A Noah Simon %A Marco Carone %B Proceedings of the 35th International Conference on Machine Learning %C Proceedings of Machine Learning Research %D 2018 %E Jennifer Dy %E Andreas Krause %F pmlr-v80-feng18a %I PMLR %P 1496--1505 %U https://proceedings.mlr.press/v80/feng18a.html %V 80 %X In predictive modeling applications, it is often of interest to determine the relative contribution of subsets of features in explaining the variability of an outcome. It is useful to consider this variable importance as a function of the unknown, underlying data-generating mechanism rather than the specific predictive algorithm used to fit the data. In this paper, we connect these ideas in nonparametric variable importance to machine learning, and provide a method for efficient estimation of variable importance when building a predictive model using a neural network. We show how a single augmented neural network with multi-task learning simultaneously estimates the importance of many feature subsets, improving on previous procedures for estimating importance. We demonstrate on simulated data that our method is both accurate and computationally efficient, and apply our method to both a study of heart disease and for predicting mortality in ICU patients.
APA
Feng, J., Williamson, B., Simon, N. & Carone, M.. (2018). Nonparametric variable importance using an augmented neural network with multi-task learning. Proceedings of the 35th International Conference on Machine Learning, in Proceedings of Machine Learning Research 80:1496-1505 Available from https://proceedings.mlr.press/v80/feng18a.html.

Related Material