arXiv · 2506.06411
CoxNTF: A New Approach for Joint Clustering and Prediction in Survival Analysis
Abstract
The interpretation of the results of survival analysis often benefits from latent factor representations of baseline covariates. However, existing methods, such as Nonnegative Matrix Factorization (NMF), do not incorporate survival information, limiting their predictive power. We present CoxNTF, a novel approach that uses non-negative tensor factorization (NTF) to derive meaningful latent representations that are closely associated with survival outcomes. CoxNTF constructs a weighted covariate tensor in which survival probabilities derived from the Coxnet model are used to guide the tensorization process. Our results show that CoxNTF achieves survival prediction performance comparable to using Coxnet with the original covariates, while providing a structured and interpretable clustering framework. In addition, the new approach effectively handles feature redundancy, making it a powerful tool for joint clustering and prediction in survival analysis.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Paul Fogel, Christophe Geissler, George Luta. 2025-06-06. CoxNTF: A New Approach for Joint Clustering and Prediction in Survival Analysis. https://arxiv.org/abs/2506.06411
Cite the original work for its findings. Save a collection to share your selection of sources.