Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

KAPLAN: Kolmogorov-Arnold Prognostic Learnable Activation Networks for Survival Analysis

About

Survival analysis aims to model how covariates and time jointly shape the time-to-event distribution under right censoring. Classical methods such as the Cox model and generalised additive models (GAMs) require interactions and time-varying effects to be manually specified, which is increasingly impractical on rich clinical datasets. We introduce KAPLAN-HR, a B-spline Kolmogorov-Arnold Network (KAN) for nonparametric estimation of the conditional hazard as a joint function of covariates and time. A single-layer KAPLAN-HR model recovers a GAM, while deeper architectures capture interactions and time-varying effects through composition. We establish a convergence rate for the nonparametric KAN hazard estimator that depends only on the smoothness of the underlying KAN representation and not on the covariate dimension, thereby mitigating the curse of dimensionality for KAN-representable targets. In evaluations over six clinical benchmark datasets, KAPLAN-HR matches or exceeds the predictive performance of established statistical and deep learning survival methods.

Stelios Boulitsakis Logothetis, Angela Wood, Pietro Li\`o• 2026

Related benchmarks

TaskDatasetResultRank
Survival PredictionFLCHAIN
IBS0.098
26
Survival AnalysisSUPPORT
Time-dependent C-index0.6706
23
Survival AnalysisNWTCO
Time-dependent C-index0.718
17
Survival AnalysisMETABRIC
D-Calibration Score0.9
17
Survival AnalysisSUPPORT
IBS0.1834
16
Survival PredictionMIMIC-III
C-index81.14
13
Survival AnalysisMIMIC-III
IBS0.1588
12
Survival AnalysisRotGBSG
IBS0.1723
12
Survival AnalysisMETABRIC
IBS0.1658
12
Survival AnalysisMIMIC-III
C-TD0.8114
10
Showing 10 of 28 rows

Other info

Follow for update