Skip to main content

Research timeline

Related research and updates

Public articles linked to the same research event.

BMC Medical Informatics and Decision Making

CleanSurvival uses reinforcement learning to auto-select preprocessing pipelines for survival analysis, improving predictive performance over simple baselines on real-world benchmarks

The work presents CleanSurvival, a reinforcement-learning (Q-learning) based automated data-preprocessing framework tailored to time-to-event (survival analysis) models for censored data: it selects among combinations of data imputation, outlier detection, and feature extraction techniques to optimize performance for a Cox, random forest, neural network, or user-supplied time-to-event model; on real-world dataset benchmarks, this Q-learning-based preprocessing improved predictive performance relative to simple baselines, runtime behavior was condition-dependent and most clearly interpretable in the best-covered benchmark cells, and a simulation study showed effectiveness across different types and levels of missingness and noise.