Research & Papers

KAPLAN: New AI method improves survival analysis with Kolmogorov-Arnold networks

KAPLAN beats deep learning on clinical benchmarks without manual feature engineering.

Deep Dive

Classical survival analysis methods like the Cox model and generalised additive models (GAMs) require manual specification of interactions and time-varying effects—a major bottleneck for rich clinical datasets. Now, researchers from Cambridge and NeurIPS 2026 present KAPLAN-HR, a novel architecture based on Kolmogorov-Arnold Networks (KANs) that nonparametrically estimates the hazard function as a joint function of covariates and time. A single-layer KAPLAN-HR recovers a GAM, while deeper architectures automatically learn complex interactions and time-varying effects through function composition.

KAPLAN-HR also offers a theoretical guarantee: its convergence rate depends only on the smoothness of the underlying KAN representation, not on the covariate dimension—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. The paper (9 pages + 13 supplementary) is submitted to NeurIPS 2026 and available on arXiv.

Key Points
  • Automatically captures interactions and time-varying effects without manual specification.
  • Convergence rate depends only on smoothness of KAN representation, not covariate dimension.
  • Matches or exceeds performance of established methods on six clinical benchmark datasets.

Why It Matters

KAPLAN automates complex hazard modeling, improving risk prediction in healthcare with less manual effort and better accuracy.