Improving Prediction of Need for Mechanical Ventilation using Cross-Attention
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866910538840145920 |
|---|---|
| author | Mohanty, Anwesh Shashikumar, Supreeth P. Lam, Jonathan Y. Nemati, Shamim |
| author_facet | Mohanty, Anwesh Shashikumar, Supreeth P. Lam, Jonathan Y. Nemati, Shamim |
| contents | In the intensive care unit, the capability to predict the need for mechanical ventilation (MV) facilitates more timely interventions to improve patient outcomes. Recent works have demonstrated good performance in this task utilizing machine learning models. This paper explores the novel application of a deep learning model with multi-head attention (FFNN-MHA) to make more accurate MV predictions and reduce false positives by learning personalized contextual information of individual patients. Utilizing the publicly available MIMIC-IV dataset, FFNN-MHA demonstrates an improvement of 0.0379 in AUC and a 17.8\% decrease in false positives compared to baseline models such as feed-forward neural networks. Our results highlight the potential of the FFNN-MHA model as an effective tool for accurate prediction of the need for mechanical ventilation in critical care settings. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2407_15885 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Improving Prediction of Need for Mechanical Ventilation using Cross-Attention Mohanty, Anwesh Shashikumar, Supreeth P. Lam, Jonathan Y. Nemati, Shamim Machine Learning Quantitative Methods In the intensive care unit, the capability to predict the need for mechanical ventilation (MV) facilitates more timely interventions to improve patient outcomes. Recent works have demonstrated good performance in this task utilizing machine learning models. This paper explores the novel application of a deep learning model with multi-head attention (FFNN-MHA) to make more accurate MV predictions and reduce false positives by learning personalized contextual information of individual patients. Utilizing the publicly available MIMIC-IV dataset, FFNN-MHA demonstrates an improvement of 0.0379 in AUC and a 17.8\% decrease in false positives compared to baseline models such as feed-forward neural networks. Our results highlight the potential of the FFNN-MHA model as an effective tool for accurate prediction of the need for mechanical ventilation in critical care settings. |
| title | Improving Prediction of Need for Mechanical Ventilation using Cross-Attention |
| topic | Machine Learning Quantitative Methods |
| url | https://arxiv.org/abs/2407.15885 |