`

Timezone: »

 
Poster
MiME: Multilevel Medical Embedding of Electronic Health Records for Predictive Healthcare
Edward Choi · Cao Xiao · Walter Stewart · Jimeng Sun

Tue Dec 04 02:00 PM -- 04:00 PM (PST) @ Room 517 AB #102

Deep learning models exhibit state-of-the-art performance for many predictive healthcare tasks using electronic health records (EHR) data, but these models typically require training data volume that exceeds the capacity of most healthcare systems. External resources such as medical ontologies are used to bridge the data volume constraint, but this approach is often not directly applicable or useful because of inconsistencies with terminology. To solve the data insufficiency challenge, we leverage the inherent multilevel structure of EHR data and, in particular, the encoded relationships among medical codes. We propose Multilevel Medical Embedding (MiME) which learns the multilevel embedding of EHR data while jointly performing auxiliary prediction tasks that rely on this inherent EHR structure without the need for external labels. We conducted two prediction tasks, heart failure prediction and sequential disease prediction, where MiME outperformed baseline methods in diverse evaluation settings. In particular, MiME consistently outperformed all baselines when predicting heart failure on datasets of different volumes, especially demonstrating the greatest performance improvement (15% relative gain in PR-AUC over the best baseline) on the smallest dataset, demonstrating its ability to effectively model the multilevel structure of EHR data.

Author Information

Edward Choi (Google)
Cao Xiao (IBM Research)
Walter Stewart (No Affiliation)
Jimeng Sun (Georgia Tech)

deep learning for healthcare tensor factorization

More from the Same Authors

  • 2021 : Conditional Generation of Periodic Signals with Fourier-Based Decoder »
    Jiyoung Lee · Wonjae Kim · Daehoon Gwak · Edward Choi
  • 2021 : Conditional Generation of Periodic Signals with Fourier-Based Decoder »
    Jiyoung Lee · Wonjae Kim · Daehoon Gwak · Edward Choi
  • 2021 Poster: Locally Valid and Discriminative Prediction Intervals for Deep Learning Models »
    Zhen Lin · Shubhendu Trivedi · Jimeng Sun
  • 2019 : Poster Session I »
    Shuangjia Zheng · Arnav Kapur · Umar Asif · Eyal Rozenberg · Cyprien Gilet · Oleksii Sidorov · Yogesh Kumar · Tom Van Steenkiste · William Boag · David Ouyang · Paul Jaeger · Sheng Liu · Aparna Balagopalan · Deepta Rajan · Marta Skreta · Nikhil Pattisapu · Jann Goschenhofer · Viraj Prabhu · Di Jin · Laura-Jayne Gardiner · Irene Li · sriram kumar · Qiyuan Hu · Mehul Motani · Justin Lovelace · Usman Roshan · Lucy Lu Wang · Ilya Valmianski · Hyeonwoo Lee · Sunil Mallya · Elias Chaibub Neto · Jonas Kemp · Marie Charpignon · Amber Nigam · Wei-Hung Weng · Sabri Boughorbel · Alexis Bellot · Lovedeep Gondara · Haoran Zhang · Mohammad Taha Bahadori · John Zech · Rulin Shao · Edward Choi · Laleh Seyyed-Kalantari · Emily Aiken · Ioana Bica · Yiqiu Shen · Kieran Chin-Cheong · Subhrajit Roy · Ioana Baldini · So Yeon Min · Dirk Deschrijver · Pekka Marttinen · Damian Pascual Ortiz · Supriya Nagesh · Niklas Rindtorff · Andriy Mulyar · Katharina Hoebel · Martha Shaka · Pierre Machart · Leon Gatys · Nathan Ng · Matthias Hüser · Devin Taylor · Dennis Barbour · Natalia Martinez · Clara McCreery · Benjamin Eyre · Vivek Natarajan · Ren Yi · Ruibin Ma · Chirag Nagpal · Nan Du · Chufan Gao · Anup Tuladhar · Sam Shleifer · Jason Ren · Pouria Mashouri · Ming Yang Lu · Farideh Bagherzadeh-Khiabani · Olivia Choudhury · Maithra Raghu · Scott Fleming · Mika Jain · GUO YANG · Alena Harley · Stephen Pfohl · Elisabeth Rumetshofer · Alex Fedorov · Saloni Dash · Jacob Pfau · Sabina Tomkins · Colin Targonski · Michael Brudno · Xinyu Li · Yiyang Yu · Nisarg Patel
  • 2018 Poster: Constrained Generation of Semantically Valid Graphs via Regularizing Variational Autoencoders »
    Tengfei Ma · Jie Chen · Cao Xiao
  • 2017 : Contributed Talks 1 »
    Cinjon Resnick · Ying Wen · Stephan Zheng · Mukul Bhutani · Edward Choi
  • 2016 Poster: RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanism »
    Edward Choi · Mohammad Taha Bahadori · Jimeng Sun · Joshua Kulas · Andy Schuetz · Walter Stewart
  • 2015 Poster: Time-Sensitive Recommendation From Recurrent User Activities »
    Nan Du · Yichen Wang · Niao He · Jimeng Sun · Le Song