Retrieval-aligned Tabular Foundation Models Enable Robust Clinical Risk Prediction in Electronic Health Records Under Real-world Constraints
Organizations: School of Computing, Dublin City University, Dublin, Ireland · ADAPT Centre, Dublin City University, Dublin, Ireland · University of Science, Vietnam National University, Ho Chi Minh City, Vietnam · Department of Otolaryngology, Thong Nhat Hospital, Ho Chi Minh City, Vietnam ·
Abstract
Clinical prediction from structured electronic health records (EHRs) is challenging due to high dimensionality, heterogeneity, class imbalance, and distribution shift. While tabular in-context learning (TICL) and retrieval-augmented methods perform well on generic benchmarks, their behavior in clinical settings remains unclear. We present a multi-cohort EHR benchmark comparing classical, deep tabular, and TICL models across varying data scale, feature dimensionality, outcome rarity, and cross-cohort generalization. PFN-based TICL models are sample-efficient in low-data regimes but degrade under naive distance-based retrieval as heterogeneity and imbalance increase. We propose AWARE, a task-aligned retrieval framework using supervised embedding learning and lightweight adapters. AWARE improves AUPRC by up to 12.2% under extreme imbalance, with gains increasing with data complexity. Our results identify retrieval quality and retrieval-inference alignment as key bottlenecks for deploying tabular in-context learning in clinical prediction.
Figures & tables
| Model | Adaptation | Training | EHR Properties | |||
|---|---|---|---|---|---|---|
| In-context | Retrieval | Pretrained | Task | Feature | Imbalance | |
| Learning | -augmented | -adaptive | Weighting | Handling | ||
| Classical tabular and tree-based models | ||||||
| Logistic Regression | ✗ | ✗ | ✗ | ✓ | ✓ | ✗ |
| k-Nearest Neighbors | ✗ | ✓ | ✗ | ✗ | ✗ | ✗ |
| XGBoost [ 9 ] | ✗ | ✗ | ✗ | ✓ | ✓ | ✗ |
| Method | HIPE | |||
|---|---|---|---|---|
| RDMT | CPE | |||
| AUROC | AUPRC | AUROC | AUPRC | |
| XGB | 0.824_{{\color[rgb]{0.5,0.5,0.5}\pm 0.010}} | 0.432_{{\color[rgb]{0.5,0.5,0.5}\pm 0.020}} | 0.755_{{\color[rgb]{0.5,0.5,0.5}\pm 0.043}} | 0.036_{{\color[rgb]{0.5,0.5,0.5}\pm 0.019}} |
| CatB | 0.827_{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}} | 0.446_{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}} | 0.645_{{\color[rgb]{0.5,0.5,0.5}\pm 0.078}} | 0.027_{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}} |
| LightG | 0.814_{{\color[rgb]{0.5,0.5,0.5}\pm 0.055}} | 0.438_{{\color[rgb]{0.5,0.5,0.5}\pm 0.065}} | 0.719_{{\color[rgb]{0.5,0.5,0.5}\pm 0.092}} | 0.030_{{\color[rgb]{0.5,0.5,0.5}\pm 0.021}} |
| TabNet | 0.827_{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}} | 0.440_{{\color[rgb]{0.5,0.5,0.5}\pm 0.009}} | 0.558_{{\color[rgb]{0.5,0.5,0.5}\pm 0.083}} | {0.111}_{{\color[rgb]{0.5,0.5,0.5}\pm 0.055}} |
| Objective | MIMIC-IV | eICU | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| SEPSIS | VAP | UTI | SEPSIS | UTI | ||||||
| AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | |
| KNN | ||||||||||
| SNNL | 0.795_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | 0.394_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.710_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.233_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.586_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.218_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | 0.803_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.325_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.018}}} | 0.757_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.062_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} |
| kNN L2 | 0.794_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}}} | 0.391_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.026}}} | 0.709_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}}} | 0.213_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.008}}} | 0.569_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}}} | 0.209_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}}} | 0.795_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.023}}} | 0.314_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.027}}} | 0.734_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.041}}} | 0.056_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} |
| Siamese | 0.868_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.453_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.756_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.235_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.014}}} | 0.608_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.012}}} | 0.216_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.009}}} | 0.847_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.010}}} | 0.332_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.017}}} | 0.677_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.010}}} | 0.037_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} |
| Model | MIMIC-IV | eICU | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| SEPSIS | VAP | UTI | SEPSIS | UTI | ||||||
| AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | |
| KNN | ||||||||||
| Full AWARE † | {\textbf{0.903}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | {\textbf{0.507}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | {\textbf{0.788}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} | 0.250_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | {\textbf{0.671}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | {\textbf{0.260}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} | {\textbf{0.971}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | {\textbf{0.711}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | {\textbf{0.928}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | {\textbf{0.155}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} |
| Attention Only | \underline{0.862}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | \underline{0.448}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.021}}} | 0.763_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | \underline{0.257}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | 0.629_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | \underline{0.236}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | \underline{0.911}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.033}}} | \underline{0.492}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.112}}} | 0.810_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.065_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} |
| SNNL Only | 0.795_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | 0.394_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.710_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.233_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.586_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.218_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | 0.803_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.325_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.018}}} | 0.757_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.062_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} |
| Model | MIMIC-IV | eICU | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| SEPSIS | VAP | UTI | SEPSIS | UTI | ||||||
| AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | |
| KNN | ||||||||||
| Full AWARE † | {\textbf{0.903}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | {\textbf{0.507}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | {\textbf{0.788}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} | 0.250_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | {\textbf{0.671}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | {\textbf{0.260}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} | {\textbf{0.971}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | {\textbf{0.711}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | {\textbf{0.928}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | {\textbf{0.155}}^{**}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} |
| w/o Attention | 0.864_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | 0.440_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} | 0.773_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | 0.258_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} | 0.632_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | 0.236_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.861_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} | 0.348_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | 0.827_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} | 0.071_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} |
| w/o SNNL | \underline{0.887}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | \underline{0.486}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | \underline{0.780}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | \textbf{0.266}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | \underline{0.640}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | \underline{0.241}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}}} | \underline{0.946}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | \underline{0.604}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | \underline{0.869}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | \underline{0.091}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} |
| Model | MIMIC-IV | eICU | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| SEPSIS | VAP | UTI | SEPSIS | UTI | ||||||
| AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | |
| KNN | ||||||||||
| Baseline | 0.794_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}}} | 0.391_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.026}}} | 0.709_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}}} | 0.213_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.008}}} | 0.569_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}}} | 0.209_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}}} | 0.795_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.023}}} | 0.314_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.027}}} | 0.734_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.041}}} | 0.056_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} |
| + SNNL | 0.795_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | 0.394_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.710_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.233_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}}} | 0.586_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.218_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | 0.803_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.325_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.018}}} | 0.757_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}}} | 0.062_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} |
| + Attention | 0.843_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | \underline{0.503}_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | 0.736_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}}} | 0.247_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}}} | 0.622_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}}} | 0.246_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}}} | 0.853_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}}} | 0.467_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.009}}} | 0.715_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.030}}} | 0.048_{{{\color[rgb]{0.5,0.5,0.5}\pm 0.014}}} |
| Model | Regression | Classification | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| HF | OXF-PT | TIT | SUPPORT2 | SSMI | DTC | GGCM | AIDS | ILP | COVID19 | Diabetes | Kidney | |
| Classical and Boosting Models | ||||||||||||
| LR | 1.360_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.413_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 1.070_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.836_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.692_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.984_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.883_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.710_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.757_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.727_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.630_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.688_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} |
| KNN | 1.394_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.551_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 1.253_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.782_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.645_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.500_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.500_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.702_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.500_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.692_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.479_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.500_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} |
| CatB | 1.366_{{\color[rgb]{0.5,0.5,0.5}\pm 0.033}} | \underline{0.107}_{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}} | 0.942_{{\color[rgb]{0.5,0.5,0.5}\pm 0.096}} | 0.839_{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}} | 0.690_{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}} | 0.972_{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}} | 0.851_{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}} | 0.714_{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}} | 0.741_{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}} | 0.782_{{\color[rgb]{0.5,0.5,0.5}\pm 0.005}} | \underline{0.676}_{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}} | 0.750_{{\color[rgb]{0.5,0.5,0.5}\pm 0.108}} |
| LightG | 1.360_{{\color[rgb]{0.5,0.5,0.5}\pm 0.032}} | \textbf{0.092}_{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}} | 1.068_{{\color[rgb]{0.5,0.5,0.5}\pm 0.061}} | 0.846_{{\color[rgb]{0.5,0.5,0.5}\pm 0.006}} | 0.691_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.990_{{\color[rgb]{0.5,0.5,0.5}\pm 0.003}} | 0.857_{{\color[rgb]{0.5,0.5,0.5}\pm 0.007}} | 0.711_{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}} | 0.727_{{\color[rgb]{0.5,0.5,0.5}\pm 0.019}} | 0.744_{{\color[rgb]{0.5,0.5,0.5}\pm 0.037}} | \textbf{0.677}_{{\color[rgb]{0.5,0.5,0.5}\pm 0.002}} | 0.854_{{\color[rgb]{0.5,0.5,0.5}\pm 0.110}} |
| Model | MIMIC-IV | eICU | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| SEPSIS | VAP | UTI | SEPSIS | UTI | ||||||
| AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | |
| Classical and Boosting Models | ||||||||||
| LR | 0.886_{{\color[rgb]{0.5,0.5,0.5}\pm 0.012}} | 0.490_{{\color[rgb]{0.5,0.5,0.5}\pm 0.019}} | 0.752_{{\color[rgb]{0.5,0.5,0.5}\pm 0.023}} | 0.218_{{\color[rgb]{0.5,0.5,0.5}\pm 0.020}} | 0.649_{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}} | 0.247_{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}} | 0.940_{{\color[rgb]{0.5,0.5,0.5}\pm 0.016}} | 0.577_{{\color[rgb]{0.5,0.5,0.5}\pm 0.051}} | 0.922_{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}} | 0.126_{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}} |
| KNN | 0.794_{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}} | 0.391_{{\color[rgb]{0.5,0.5,0.5}\pm 0.026}} | 0.709_{{\color[rgb]{0.5,0.5,0.5}\pm 0.015}} | 0.213_{{\color[rgb]{0.5,0.5,0.5}\pm 0.008}} | 0.569_{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}} | 0.209_{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}} | 0.795_{{\color[rgb]{0.5,0.5,0.5}\pm 0.023}} | 0.314_{{\color[rgb]{0.5,0.5,0.5}\pm 0.027}} | 0.734_{{\color[rgb]{0.5,0.5,0.5}\pm 0.041}} | 0.056_{{\color[rgb]{0.5,0.5,0.5}\pm 0.011}} |
| CatB | 0.903_{{\color[rgb]{0.5,0.5,0.5}\pm 0.008}} | 0.527_{{\color[rgb]{0.5,0.5,0.5}\pm 0.025}} | 0.754_{{\color[rgb]{0.5,0.5,0.5}\pm 0.010}} | 0.235_{{\color[rgb]{0.5,0.5,0.5}\pm 0.014}} | 0.659_{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}} | 0.257_{{\color[rgb]{0.5,0.5,0.5}\pm 0.013}} | 0.981_{{\color[rgb]{0.5,0.5,0.5}\pm 0.001}} | 0.756_{{\color[rgb]{0.5,0.5,0.5}\pm 0.008}} | 0.937_{{\color[rgb]{0.5,0.5,0.5}\pm 0.010}} | 0.153_{{\color[rgb]{0.5,0.5,0.5}\pm 0.004}} |
| Model | Lung Cancer | Stroke |
|---|---|---|
| Classical and Ensemble Models | ||
| Logistic Regression | 0.993_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} | 0.992_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} |
| XGBoost | 0.999_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} | 0.999_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} |
| Tabular In-Context Learning Models | ||
| TabPFN | 1.0_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} | 0.999_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} |
| TabDPT | 0.992_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} | 0.978_{{\color[rgb]{0.5,0.5,0.5}\pm 0.00}} |
| Dataset Name | #Rows | #Features | Task | Imbalance Ratio | Cohort | Feature Types |
|---|---|---|---|---|---|---|
| Synthetic EHR datasets | ||||||
| Synthea-Stroke [ 5 ] | 16,055 | 793 | Binary | – | Multi-domain | DEM, VIT, LAB, CODE |
| Synthea-LungCancer [ 5 ] | 4,384 | 773 | Binary | – | Oncology | DEM, VIT, LAB, CODE |
| Small-to-medium real-world EHR datasets | ||||||
| Heart Failure (Platelets Counts) [ 11 ] | 299 | 12 | Regression | – | Cardiovascular | DEM, LAB |
| OXF – Parkinsons Telemonitoring [ 74 ] | 5,875 | 22 | Regression | – | Neurological | DEM, PHYS, TMP |
| Hyperparameter | Range / Type |
|---|---|
| Classical Machine Learning Methods | |
| Logistic Regression | |
| C | Log-uniform |
| penalty | Categorical |
| max_iter | Integer |
| XGBoost | |
Tabular Foundation Models
Tabular foundation models learn reusable representations or in-context prediction skills for structured data such as spreadsheets and databases.
See all 187 papers