A Novel Method for Medical Predictive Models in Small Data Using Out-of-Distribution Data and Transfer Learning

Inyong Jeong, Yeongmin Kim, Nam-Jun Cho, Hyo-Wook Gil and Hwamin Lee ()
Additional contact information
Inyong Jeong: Department of Biomedical Informatics, Korea University College of Medicine, Seoul 02708, Republic of Korea
Yeongmin Kim: Department of Biomedical Informatics, Korea University College of Medicine, Seoul 02708, Republic of Korea
Nam-Jun Cho: Department of Internal Medicine, Soonchunhyang University Cheonan Hospital, Cheonan 31151, Republic of Korea
Hyo-Wook Gil: Department of Internal Medicine, Soonchunhyang University Cheonan Hospital, Cheonan 31151, Republic of Korea
Hwamin Lee: Department of Biomedical Informatics, Korea University College of Medicine, Seoul 02708, Republic of Korea

Mathematics, 2024, vol. 12, issue 2, 1-26

Abstract: Applying deep learning to medical research with limited data is challenging. This study focuses on addressing this difficulty through a case study, predicting acute respiratory failure (ARF) in patients with acute pesticide poisoning. Commonly, out-of-distribution (OOD) data are overlooked during model training in the medical field. Our approach integrates OOD data and transfer learning (TL) to enhance model performance with limited data. We fine-tuned a pre-trained multi-layer perceptron model using OOD data, outperforming baseline models. Shapley additive explanation (SHAP) values were employed for model interpretation, revealing the key factors associated with ARF. Our study is pioneering in applying OOD and TL techniques to electronic health records to achieve better model performance in scenarios with limited data. Our research highlights the potential benefits of using OOD data for initializing weights and demonstrates that TL can significantly improve model performance, even in medical data with limited samples. Our findings emphasize the significance of utilizing context-specific information in TL to achieve better results. Our work has practical implications for addressing challenges in rare diseases and other scenarios with limited data, thereby contributing to the development of machine-learning techniques within the medical field, especially regarding health inequities.

Keywords: transfer learning; out-of-distribution data; machine learning; limited medical data; acute respiratory failure; acute pesticide poisoning (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/2/237/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/2/237/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:12:y:2024:i:2:p:237-:d:1317149

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().