# Bridging data gaps of rare conditions in ICU: a multi-disease adaptation approach for clinical prediction

## TL;DR

This paper introduces KnowRare, an AI framework that improves clinical predictions for rare ICU conditions by leveraging data from similar common conditions.

## Contribution

KnowRare is a novel domain adaptation framework that addresses data scarcity and heterogeneity in rare ICU conditions using a condition knowledge graph.

## Key findings

- KnowRare outperformed existing models in five ICU prediction tasks.
- It showed better performance than established ICU scoring systems like APACHE IV.
- Case studies confirmed its adaptability and generalization capabilities.

## Abstract

Artificial Intelligence has revolutionised critical care for common conditions. Yet, rare conditions in the intensive care unit (ICU), including recognised rare diseases and low-prevalence conditions in the ICU, remain underserved due to data scarcity and intra-condition heterogeneity. To bridge such gaps, we developed KnowRare, a domain adaptation-based deep learning framework for predicting clinical outcomes for rare conditions in the ICU. KnowRare mitigates data scarcity by initially learning condition-agnostic representations from diverse electronic health records through self-supervised pre-training. It addresses intra-condition heterogeneity by selectively adapting knowledge from clinically similar conditions with a developed condition knowledge graph. Evaluated on two ICU datasets across five clinical prediction tasks (90-day mortality, 30-day readmission, ICU mortality, remaining length of stay, and phenotyping), KnowRare consistently outperformed existing state-of-the-art models. Additionally, KnowRare demonstrated superior predictive performance compared to established ICU scoring systems, including APACHE IV and IV-a. Case studies further demonstrated KnowRare’s flexibility in adapting its parameters to accommodate dataset-specific and task-specific characteristics, its generalisation to common conditions under limited data scenarios, and its rationality in selecting source conditions. These findings highlight KnowRare’s potential as a robust and practical solution for supporting clinical decision-making and improving care for rare conditions in the ICU.