mlr3proba: An R Package for Machine Learning in Survival Analysis

TL;DR

mlr3proba is an R package that extends machine learning capabilities specifically for survival analysis, integrating with mlr3 for comprehensive model tuning and benchmarking in various scientific fields.

Contribution

It introduces a dedicated R package for survival analysis that enhances existing machine learning tools with specialized support and systematic evaluation features.

Findings

Provides a unified interface for survival models in R.

Enables systematic benchmarking and tuning of survival algorithms.

Fills a gap in R's machine learning ecosystem for survival analysis.

Abstract

As machine learning has become increasingly popular over the last few decades, so too has the number of machine learning interfaces for implementing these models. Whilst many R libraries exist for machine learning, very few offer extended support for survival analysis. This is problematic considering its importance in fields like medicine, bioinformatics, economics, engineering, and more. mlr3proba provides a comprehensive machine learning interface for survival analysis and connects with mlr3's general model tuning and benchmarking facilities to provide a systematic infrastructure for survival modeling and evaluation.