Learning Disease Progression Models That Capture Health Disparities

Erica Chiang,Divya Shanmugam,Ashley N. Beecy,Gabriel Sayer,Nir Uriel,Deborah Estrin,Nikhil Garg,Emma Pierson

2024-12-21

Abstract:Disease progression models are widely used to inform the diagnosis and treatment of many progressive diseases. However, a significant limitation of existing models is that they do not account for health disparities that can bias the observed data. To address this, we develop an interpretable Bayesian disease progression model that captures three key health disparities: certain patient populations may (1) start receiving care only when their disease is more severe, (2) experience faster disease progression even while receiving care, or (3) receive follow-up care less frequently conditional on disease severity. We show theoretically and empirically that failing to account for disparities produces biased estimates of severity (underestimating severity for disadvantaged groups, for example). On a dataset of heart failure patients, we show that our model can identify groups that face each type of health disparity, and that accounting for these disparities meaningfully shifts which patients are considered high-risk.

Machine Learning,Artificial Intelligence,Computers and Society,Applications

What problem does this paper attempt to address?

The main problem that this paper attempts to solve is that existing disease progression models fail to fully consider health disparities, which leads to biased estimates of the disease severity in different populations. Specifically, the paper points out that the following three health disparities are not fully considered in existing models:

1. **Differences in initial severity**: Some patient groups may start receiving care when their diseases are more severe.
2. **Differences in disease progression rate**: Even when receiving care, some patient groups may have a faster disease progression rate.
3. **Differences in follow - up frequency**: Some patient groups have a lower frequency of follow - up care at the same disease severity.

These differences can lead to biases in the model's estimates of disease severity, such as underestimating the disease severity of vulnerable groups. To solve these problems, the author has developed an interpretable Bayesian disease progression model that can capture the above three health disparities. Through theoretical and empirical analysis, the author has proven that ignoring these differences will lead to biased estimates of disease severity and has shown that their model can identify groups facing these health disparities in a heart failure patient dataset, thereby more accurately assessing patients' high - risk states.

### Mathematical formula summary

- **Disease progression model**:
\[
Z_t = Z_0+R\cdot t
\]
where \(Z_t\) represents the disease severity at time \(t\), \(Z_0\) represents the initial severity, and \(R\) represents the disease progression rate.

- **Observed feature model**:
\[
X_t = f(Z_t)+\epsilon_t, \quad \epsilon_t\sim N(0,\Psi)
\]
where \(X_t\) is the observed symptom or feature, \(\epsilon_t\) is the noise term, and \(\Psi\) is the covariance matrix.

- **Visit frequency model**:
\[
\log(\lambda_t)=\beta_0+\beta_Z\cdot Z_t+\beta_A^{(a)}
\]
where \(\lambda_t\) is the visit rate, \(\beta_0\) is the intercept, \(\beta_Z\) is the severity coefficient, and \(\beta_A^{(a)}\) is the parameter for a specific group.

### Conclusion

By introducing these health disparities, the author's model can more accurately estimate the disease severity of different patient groups, thereby helping doctors make better diagnosis and treatment decisions. In addition, this model can also reveal more fine - grained descriptions of health disparities, which is helpful for improving the fairness issue in the healthcare system.