Back

Predicting 30 Days Hospital Readmission for Heart Failure patients using word embeddings

Shakya, P. R.; Khaneja, A.; Wagholikar, K. B.

2025-02-10 health informatics
10.1101/2025.02.07.25321871 medRxiv
Show abstract

Heart Failure (HF) is a public health concern with a wider impact on quality of life and cost of care. One of the major challenges in HF is the higher rate of unplanned readmissions and sub-optimal performance of models to predict the readmissions. Hence, in this study, we implemented embeddings-based approaches to generate features for improving model performance. Specifically, we compared three embedding approaches including word2vec on terminology codes and CUIs, and BERT on concept descriptions with baseline (one hot-encoding). We found that the embedding approaches significantly improved the performance of the prediction models, and word2vec on the study dataset outperformed pre-trained BERT model.

Matching journals

The top 4 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.