Diagnosis Uncertain Models For Medical Risk Prediction

Alexander Peysakhovich,Rich Caruana,Yin Aphinyanaphongs
DOI: https://doi.org/10.48550/arXiv.2306.17337
2023-06-30
Abstract:We consider a patient risk models which has access to patient features such as vital signs, lab values, and prior history but does not have access to a patient's diagnosis. For example, this occurs in a model deployed at intake time for triage purposes. We show that such `all-cause' risk models have good generalization across diagnoses but have a predictable failure mode. When the same lab/vital/history profiles can result from diagnoses with different risk profiles (e.g. <a class="link-external link-http" href="http://E.coli" rel="external noopener nofollow">this http URL</a> vs. MRSA) the risk estimate is a probability weighted average of these two profiles. This leads to an under-estimation of risk for rare but highly risky diagnoses. We propose a fix for this problem by explicitly modeling the uncertainty in risk prediction coming from uncertainty in patient diagnoses. This gives practitioners an interpretable way to understand patient risk beyond a single risk number.
Artificial Intelligence
What problem does this paper attempt to address?