Stronger Baseline Models -- A Key Requirement for Aligning Machine Learning Research with Clinical Utility
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_ | 1866917779387449344 |
|---|---|
| author | Wolfrath, Nathan Wolfrath, Joel Hu, Hengrui Banerjee, Anjishnu Kothari, Anai N. |
| author_facet | Wolfrath, Nathan Wolfrath, Joel Hu, Hengrui Banerjee, Anjishnu Kothari, Anai N. |
| contents | Machine Learning (ML) research has increased substantially in recent years, due to the success of predictive modeling across diverse application domains. However, well-known barriers exist when attempting to deploy ML models in high-stakes, clinical settings, including lack of model transparency (or the inability to audit the inference process), large training data requirements with siloed data sources, and complicated metrics for measuring model utility. In this work, we show empirically that including stronger baseline models in healthcare ML evaluations has important downstream effects that aid practitioners in addressing these challenges. Through a series of case studies, we find that the common practice of omitting baselines or comparing against a weak baseline model (e.g. a linear model with no optimization) obscures the value of ML methods proposed in the research literature. Using these insights, we propose some best practices that will enable practitioners to more effectively study and deploy ML models in clinical settings. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2409_12116 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Stronger Baseline Models -- A Key Requirement for Aligning Machine Learning Research with Clinical Utility Wolfrath, Nathan Wolfrath, Joel Hu, Hengrui Banerjee, Anjishnu Kothari, Anai N. Machine Learning Computers and Society Machine Learning (ML) research has increased substantially in recent years, due to the success of predictive modeling across diverse application domains. However, well-known barriers exist when attempting to deploy ML models in high-stakes, clinical settings, including lack of model transparency (or the inability to audit the inference process), large training data requirements with siloed data sources, and complicated metrics for measuring model utility. In this work, we show empirically that including stronger baseline models in healthcare ML evaluations has important downstream effects that aid practitioners in addressing these challenges. Through a series of case studies, we find that the common practice of omitting baselines or comparing against a weak baseline model (e.g. a linear model with no optimization) obscures the value of ML methods proposed in the research literature. Using these insights, we propose some best practices that will enable practitioners to more effectively study and deploy ML models in clinical settings. |
| title | Stronger Baseline Models -- A Key Requirement for Aligning Machine Learning Research with Clinical Utility |
| topic | Machine Learning Computers and Society |
| url | https://arxiv.org/abs/2409.12116 |