Timezone: »

 
Explanation-guided dynamic feature selection for medical risk prediction
Nicasia Beebe-Wang · Wei Qiu · Su-In Lee
Event URL: https://openreview.net/forum?id=1itfhff53V »

In medical risk prediction scenarios, machine learning methods have demonstrated an ability to learn complex and predictive relationships among rich feature sets. However, in practice, when faced with new patients, we may not have access all information expected by a trained risk model. We propose a framework to simultaneously provide flexible risk estimates for samples with missing features, as well as context-dependent feature recommendations to identify what piece of information may be most valuable to collect next. Our approach uses a fixed prediction model, a local feature explainer, and ensembles of imputed samples to generate risk prediction intervals and feature recommendations. Applied to a myocardial infarction risk prediction task in the UK Biobank dataset, we find that our approach can more efficiently predict risk of a heart attack with fewer observed features than traditional fixed imputation and global feature selection methods.

Author Information

Nicasia Beebe-Wang (Paul G Allen School of Computer Science and Engineering, University of Washington)
Wei Qiu (University of washington)
Su-In Lee (University of Washington)

More from the Same Authors

  • 2023 Poster: Learning to Maximize Mutual Information for Dynamic Feature Selection »
    Ian Covert · Wei Qiu · MingYu Lu · Na Yoon Kim · Nathan White · Su-In Lee
  • 2021 : Explainable AI for healthcare »
    Su-In Lee
  • 2019 : Poster Session & Lunch break »
    Kay Wiese · Brandon Carter · Dan DeBlasio · Mohammad Hashir · Rachel Chan · Matteo Manica · Ali Oskooei · Zhenqin Wu · Karren Yang · François FAGES · Ruishan Liu · Nicasia Beebe-Wang · Bryan He · Jacopo Cirrone · Pekka Marttinen · Elior Rahmani · Harri Lähdesmäki · Nikhil Yadala · Andreea-Ioana Deac · Ava Soleimany · Mansi Ranjit Mane · Jason Ernst · Joseph Paul Cohen · Joel Mathew · Vishal Agarwal · AN ZHENG