Retrieval-aligned Tabular Foundation Models Enable Robust Clinical Risk Prediction in Electronic Health Records Under Real-world Constraints
Researchers present AWARE, a retrieval-aligned framework for improving clinical risk prediction in electronic health records using tabular foundation models. The method addresses limitations of naive retrieval-augmented approaches in clinical settings, achieving up to 12.2% improvement in AUPRC under extreme class imbalance while maintaining robustness across varying data complexity.