Search papers, labs, and topics across Lattice.
This paper introduces Surv-IPTB, an attention-based framework designed to estimate the Individual Probability of Treatment Benefit (IPTB) in survival analysis by reformulating IPTB estimation as a binary classification problem. The model effectively handles right-censored observations using interval-valued probabilities and employs an attention mechanism for flexible aggregation of pairwise patient comparisons. Experimental results demonstrate that Surv-IPTB significantly outperforms traditional meta-learner baselines in complex nonlinear scenarios, establishing it as a robust tool for personalized treatment benefit assessment.
Surv-IPTB outperforms traditional survival analysis methods by accurately estimating treatment benefits in complex, nonlinear patient data.
This work presents a novel attention-based framework for estimating the Individual Probability of Treatment Benefit (IPTB) in survival analysis contexts. The proposed model, called Surv-IPTB, directly quantifies the probability that a specific patient will experience extended survival time under treatment versus control. We reformulate IPTB estimation as a binary classification problem, leveraging pairwise patient comparisons across treatment and control cohorts. The framework incorporates a principled handling of right-censored observations through imprecise probability representations, where uncertain treatment effects are characterized by interval-valued probabilities. An attention mechanism with learnable query-key transformations enables flexible, data-driven aggregation of pairwise comparisons, while simultaneously learning soft class probabilities for censored cases. Through extensive experiments on synthetic datasets with complex nonlinear structures, including spiral, bell-shaped, and circular feature spaces, we demonstrate that our approach maintains robust performance across varying censoring rates and treatment effect strengths. The model consistently outperforms meta-learner baselines (T-learner and S-learner) equipped with random survival forests, Cox proportional hazards, and Beran estimators, particularly in challenging nonlinear scenarios where conventional methods exhibit significant degradation. The results establish the proposed attention-based framework as a scalable and statistically principled solution for personalized treatment benefit assessment in survival settings. The code implementing the model is publicly available.
