We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: A Simple and Interpretable Predictive Model for Healthcare

Abstract: Deep Learning based models are currently dominating most state-of-the-art solutions for disease prediction. Existing works employ RNNs along with multiple levels of attention mechanisms to provide interpretability. These deep learning models, with trainable parameters running into millions, require huge amounts of compute and data to train and deploy. These requirements are sometimes so huge that they render usage of such models as unfeasible. We address these challenges by developing a simpler yet interpretable non-deep learning based model for application to EHR data. We model and showcase our work's results on the task of predicting first occurrence of a diagnosis, often overlooked in existing works. We push the capabilities of a tree based model and come up with a strong baseline for more sophisticated models. Its performance shows an improvement over deep learning based solutions (both, with and without the first-occurrence constraint) all the while maintaining interpretability.
Comments: 7 pages, 10 figures
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG); Applications (stat.AP)
Journal reference: Proceedings of the Canadian Conference on Artificial Intelligence 2021
DOI: 10.21428/594757db.8a8324ec
Cite as: arXiv:2007.13351 [stat.ML]
  (or arXiv:2007.13351v1 [stat.ML] for this version)

Submission history

From: Subhadip Maji [view email]
[v1] Mon, 27 Jul 2020 08:13:37 GMT (2659kb,D)

Link back to: arXiv, form interface, contact.