ArXiv Preprint
Drug recommendation assists doctors in prescribing personalized medications
to patients based on their health conditions. Existing drug recommendation
solutions adopt the supervised multi-label classification setup and only work
with existing drugs with sufficient prescription data from many patients.
However, newly approved drugs do not have much historical prescription data and
cannot leverage existing drug recommendation methods. To address this, we
formulate the new drug recommendation as a few-shot learning problem. Yet,
directly applying existing few-shot learning algorithms faces two challenges:
(1) complex relations among diseases and drugs and (2) numerous false-negative
patients who were eligible but did not yet use the new drugs. To tackle these
challenges, we propose EDGE, which can quickly adapt to the recommendation for
a new drug with limited prescription data from a few support patients. EDGE
maintains a drug-dependent multi-phenotype few-shot learner to bridge the gap
between existing and new drugs. Specifically, EDGE leverages the drug ontology
to link new drugs to existing drugs with similar treatment effects and learns
ontology-based drug representations. Such drug representations are used to
customize the metric space of the phenotype-driven patient representations,
which are composed of a set of phenotypes capturing complex patient health
status. Lastly, EDGE eliminates the false-negative supervision signal using an
external drug-disease knowledge base. We evaluate EDGE on two real-world
datasets: the public EHR data (MIMIC-IV) and private industrial claims data.
Results show that EDGE achieves 7.3% improvement on the ROC-AUC score over the
best baseline.
Zhenbang Wu, Huaxiu Yao, Zhe Su, David M Liebovitz, Lucas M Glass, James Zou, Chelsea Finn, Jimeng Sun
2022-10-11
