Imitation learning for clinical decision support in pediatric ECMO

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Golivand, Fateme, Skinner, Michael, Mathur, Saurabh, Soni, Ameet, Reeder, Phillip, Kersting, Kristian, Raman, Lakshmi, Natarajan, Sriraam
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917500803874816
author Golivand, Fateme
Skinner, Michael
Mathur, Saurabh
Soni, Ameet
Reeder, Phillip
Kersting, Kristian
Raman, Lakshmi
Natarajan, Sriraam
author_facet Golivand, Fateme
Skinner, Michael
Mathur, Saurabh
Soni, Ameet
Reeder, Phillip
Kersting, Kristian
Raman, Lakshmi
Natarajan, Sriraam
contents Pediatric critical care is a dynamic, high-stakes process involving constant monitoring and adjustments in life-saving treatments. Modeling these interventions is crucial for effective decision support. To address the challenges of high complexity and data scarcity in pediatric Extracorporeal Membrane Oxygenation (ECMO), we frame clinical decision-making as learning to act from trajectories, i.e., imitation learning that learns action models from observational data, with a key feature that actions are not directly observed. We consider TabPFN, a recent transformer-based approach for tabular data, and traditional baselines including XGBoost and Multi-Layer Perceptrons(MLPs) on real-world pediatric ECMO data to learn the action models. We find that the TabPFN-based approach consistently outperforms these classical baselines, supporting its use as a strong clinician-behavior baseline for pediatric ECMO decision support.
format Preprint
id arxiv_https___arxiv_org_abs_2605_16175
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Imitation learning for clinical decision support in pediatric ECMO
Golivand, Fateme
Skinner, Michael
Mathur, Saurabh
Soni, Ameet
Reeder, Phillip
Kersting, Kristian
Raman, Lakshmi
Natarajan, Sriraam
Machine Learning
Pediatric critical care is a dynamic, high-stakes process involving constant monitoring and adjustments in life-saving treatments. Modeling these interventions is crucial for effective decision support. To address the challenges of high complexity and data scarcity in pediatric Extracorporeal Membrane Oxygenation (ECMO), we frame clinical decision-making as learning to act from trajectories, i.e., imitation learning that learns action models from observational data, with a key feature that actions are not directly observed. We consider TabPFN, a recent transformer-based approach for tabular data, and traditional baselines including XGBoost and Multi-Layer Perceptrons(MLPs) on real-world pediatric ECMO data to learn the action models. We find that the TabPFN-based approach consistently outperforms these classical baselines, supporting its use as a strong clinician-behavior baseline for pediatric ECMO decision support.
title Imitation learning for clinical decision support in pediatric ECMO
topic Machine Learning
url https://arxiv.org/abs/2605.16175