Skip to main navigation Skip to search Skip to main content

CHEER: Rich Model Helps Poor Model via Knowledge Infusion

  • Cao Xiao
  • , Trong Nghia Hoang
  • , Shenda Hong
  • , Tengfei Ma
  • , Jimeng Sun
  • IQVIA Inc.
  • MIT-IBM Watson AI Lab
  • Georgia Institute of Technology
  • University of Illinois at Urbana-Champaign

Research output: Contribution to journalArticlepeer-review

3 Scopus citations

Abstract

There is a growing interest in applying deep learning (DL) to healthcare, driven by the availability of data with multiple feature channels in rich-data environments (e.g., intensive care units). However, in many other practical situations, we can only access data with much fewer feature channels in a poor-data environments (e.g., at home), which often results in predictive models with poor performance. How can we boost the performance of models learned from such poor-data environment by leveraging knowledge extracted from existing models trained using rich data in a related environment? To address this question, we develop a knowledge infusion framework named CHEER that can succinctly summarize such rich model into transferable representations, which can be incorporated into the poor model to improve its performance. The infused model is analyzed theoretically and evaluated empirically on several datasets. Our empirical results showed that CHEER outperformed baselines by 5.60 to 46.80 percent in terms of the macro-F1 score on multiple physiological datasets.

Original languageEnglish
Pages (from-to)531-543
Number of pages13
JournalIEEE Transactions on Knowledge and Data Engineering
Volume34
Issue number2
DOIs
StatePublished - Feb 1 2022

Keywords

  • Health analytics
  • embedding
  • representation learning

Fingerprint

Dive into the research topics of 'CHEER: Rich Model Helps Poor Model via Knowledge Infusion'. Together they form a unique fingerprint.

Cite this