Saved in:
Bibliographic Details
Main Authors: Shen, Xiaobin, Chen, George H.
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.08063
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908699592753152
author Shen, Xiaobin
Chen, George H.
author_facet Shen, Xiaobin
Chen, George H.
contents We propose an interpretable deep competing risks model called the Deep Kernel Aalen-Johansen (DKAJ) estimator, which generalizes the classical Aalen-Johansen nonparametric estimate of cumulative incidence functions (CIFs). Each data point (e.g., patient) is represented as a weighted combination of clusters. If a data point has nonzero weight only for one cluster, then its predicted CIFs correspond to those of the classical Aalen-Johansen estimator restricted to data points from that cluster. These weights come from an automatically learned kernel function that measures how similar any two data points are. On four standard competing risks datasets, we show that DKAJ is competitive with state-of-the-art baselines while being able to provide visualizations to assist model interpretation.
format Preprint
id arxiv_https___arxiv_org_abs_2512_08063
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Deep Kernel Aalen-Johansen Estimator: An Interpretable and Flexible Neural Net Framework for Competing Risks
Shen, Xiaobin
Chen, George H.
Machine Learning
We propose an interpretable deep competing risks model called the Deep Kernel Aalen-Johansen (DKAJ) estimator, which generalizes the classical Aalen-Johansen nonparametric estimate of cumulative incidence functions (CIFs). Each data point (e.g., patient) is represented as a weighted combination of clusters. If a data point has nonzero weight only for one cluster, then its predicted CIFs correspond to those of the classical Aalen-Johansen estimator restricted to data points from that cluster. These weights come from an automatically learned kernel function that measures how similar any two data points are. On four standard competing risks datasets, we show that DKAJ is competitive with state-of-the-art baselines while being able to provide visualizations to assist model interpretation.
title Deep Kernel Aalen-Johansen Estimator: An Interpretable and Flexible Neural Net Framework for Competing Risks
topic Machine Learning
url https://arxiv.org/abs/2512.08063