OPSurv: Orthogonal Polynomials Quadrature Algorithm for Survival Analysis

Lilian W. Bialokozowicz,Hoang M. Le,Tristan Sylvain,Peter A. I. Forsyth,Vineel Nagisetty,Greg Mori

2024-02-03

Abstract:This paper introduces the Orthogonal Polynomials Quadrature Algorithm for Survival Analysis (OPSurv), a new method providing time-continuous functional outputs for both single and competing risks scenarios in survival analysis. OPSurv utilizes the initial zero condition of the Cumulative Incidence function and a unique decomposition of probability densities using orthogonal polynomials, allowing it to learn functional approximation coefficients for each risk event and construct Cumulative Incidence Function estimates via Gauss--Legendre quadrature. This approach effectively counters overfitting, particularly in competing risks scenarios, enhancing model expressiveness and control. The paper further details empirical validations and theoretical justifications of OPSurv, highlighting its robust performance as an advancement in survival analysis with competing risks.

Machine Learning,Artificial Intelligence,Functional Analysis

What problem does this paper attempt to address?

The paper attempts to address the problem of how to more accurately predict the time of event occurrence in survival analysis, especially in the presence of competing risks. Traditional methods often assume only one risk factor, which is overly simplistic in practical applications and fails to capture multiple potentially interacting risk factors. Additionally, existing models are prone to overfitting when dealing with high noise-to-signal ratio data, limiting their expressiveness and generalization capabilities.

To tackle these challenges, the paper proposes a new method—Orthogonal Polynomial Quadrature Algorithm (OPSurv). This method uses the decomposition of orthogonal polynomials to approximate the probability density function and constructs the cumulative incidence function estimation through the Gauss-Legendre quadrature method. This approach not only effectively prevents overfitting but also enhances the model's expressiveness and control, thereby providing continuous time function outputs in both single-risk and multi-risk scenarios, increasing the model's robustness and flexibility.

Specifically, the main contributions of OPSurv include:
1. **Introduction of a new algorithm**: OPSurv is a new survival analysis method, particularly suitable for situations with competing risks. It adjusts the model's expressiveness through orthogonal polynomial approximation, outputting continuous time functions representing survival probabilities and other important distributions.
2. **Demonstration of strong performance**: The method performs excellently in various scenarios, proving its versatility and effectiveness.
3. **Provision of theoretical support**: The paper not only provides empirical results but also conducts theoretical validation, comprehensively proving the method's effectiveness.

Through these contributions, OPSurv marks a significant advancement in the field of survival analysis, especially in handling complex competing risks.