Surv-IPTB：用注意力机制预测个体治疗获益概率

近两年临床预测模型和生存分析工具越来越被关注大家已经不满足于只算“一组病人的平均风险”而是想回答一个更具体的问题面前这个病人用A方案还是B方案谁获益更大传统生存分析通常给出的是风险比HR、中位生存时间这类群体层面的结论。但因为个体异质性群体平均获益并不等于每个个体获益。一个真实的临床场景是两项药物试验的HR都是0.75看起来疗效一致但实际上试验1里几乎所有病人都受益试验2里只有少数病人有强响应、多数人无效甚至受损。如果只报HR临床决策就可能走偏。Surv-IPTB要解决的正是这个痛点它用注意力机制在生存数据上估计个体治疗获益概率IPTB把“这个治疗对TA到底有没有用”这件事从群体统计推向个体预测。本文会从问题背景、核心原理、数据要求、代码实现、结果验证到常见坑完整拆解一遍。需要说明的是Surv-IPTB是学术研究型模型公开版本和实现细节会随论文版本、复现仓库更新而变化。因此本文不纠结某个固定版本号而是围绕模型设计思想和可落地的实践思路展开让你能理解它、复现它再迁移到你自己的生存数据项目中。1. 这篇文章真正要解决的问题1.1 群体疗效估计的局限在随机对照试验RCT中我们经常用Cox比例风险模型或KM曲线比较两组生存差异。最终结果通常给一个HR和95%置信区间。但这个HR代表的是“平均处理效应ATE”它隐含了一个假设处理效应在整个样本中相对一致。这个假设在现实中经常不成立。同一个化疗方案对某些基因亚型有效对另一些亚型无效甚至有害。同一个靶向药在携带特定突变的人群里获益很大在无突变的群体里可能毫无作用。如果数据中这些亚型比例不同最终HR就会被稀释或夸大。1.2 个体治疗获益为什么难估计要在单个病人层面估计治疗获益至少遇到三个困难反事实缺失每个病人只能观察到一种治疗下的结局另一种治疗结局永远缺失。个体获益本质上是一个反事实推断问题。删失数据生存数据普遍存在右删失很多人随访结束时还未发生事件这给推断带来额外不确定性。高维异质性病人的基线特征、生物标志物、病史等维度很高关键修饰因子可能是某个特征组合而不是单一变量。传统方法受限于手工指定交互项很难自动发现这种高维异质性。机器学习模型虽然有拟合能力但很多模型只输出预测风险不能直接回答“治疗获益概率”。1.3 Surv-IPTB的回答方式Surv-IPTB把问题建模为一个注意力机制驱动的个体治疗效果预测模型。它不直接输出“会获益”或“不会获益”的二分类结论而是估计每个个体从治疗中获得益处的概率。这样临床医生可以结合概率阈值做决策模型也保留了不确定性信息。从材料看这个模型的核心竞争力在于使用注意力机制自动识别与治疗获益相关的特征而不是人工指定交互项。直接面向生存数据可以处理删失。输出个体获益概率而不是群体平均效应。更通俗地说过去我们问“这个药对这类病人平均有效吗”现在Surv-IPTB让我们更有机会回答“这个药对眼前这个具体病人有效的概率是多少”。2. 基础概念与核心原理2.1 IPTB个体治疗获益概率IPTBIndividual Probability of Treatment Benefit是模型的目标输出。定义为在给定协变量 ( x ) 的条件下接受治疗 ( T1 ) 相比不治疗 ( T0 ) 能获得更好结局的概率。如果用生存结局来定义“获益”常见有两种方式在一定时间点 ( t_0 ) 上治疗组的生存概率高于对照组。治疗组的期望限制平均生存时间RMST更长。Surv-IPTB 中的“获益”设计需要看具体论文定义但一般逻辑是[ IPTB(x) P(S_1(t_0|x) S_0(t_0|x)) ]其中 ( S_1(t|x) ) 是治疗组在 ( t ) 时刻的生存函数( S_0(t|x) ) 是对照组的生存函数。对于单个病人模型推断出他的预测生存曲线之后再比较治疗和对照两条曲线从而得到获益概率。2.2 注意力机制在生存分析里的作用注意力机制最早流行于自然语言处理后来在表格数据和时间序列模型中被大量使用。它的核心思想是在聚合输入信息时不是把每个特征同等对待而是通过学习为每个特征或每个样本赋予一个权重。在Surv-IPTB场景中注意力可以作用于两个层面特征层面判断哪些特征对判断治疗获益更重要。比如年龄、肿瘤分期、生物标志物各自应该占多少权重。样本层面判断当前病人与训练集中哪些历史样本更相似用相似样本的结局来推测当前样本的反事实结局。这种机制的价值在于不同个体可能有不同的“重要特征集合”。一个病人的获益主要由基因突变决定另一个病人的获益主要由合并症状态决定。注意力机制可以让模型针对每个个体动态调整关注点。2.3 与常规生存模型的区别维度Cox模型DeepSurv等深度学习生存模型Surv-IPTB输出风险比/风险函数个体风险个体获益概率异质性需手工加交互项自动学习部分交互通过注意力自动关注获益相关特征删失处理支持支持支持决策支持群体层面个体风险个体治疗选择Cox模型回答“风险高低”DeepSurv回答“这个人的风险函数是什么”Surv-IPTB进一步回答“这个人用了治疗以后有多大可能比不用更好”。这不是替代关系而是递进关系。2.4 模型的训练逻辑从研究思路推断Surv-IPTB的训练流程大概分为几步构造治疗组和对照组的生存数据。使用带注意力的网络结构分别学习两个潜在结果下的生存分布。对每个样本同时预测其在“接受治疗”和“不接受治疗”两种状态下的生存曲线。通过比较两条曲线得到个体治疗获益概率。设计损失函数将生存似然函数和治疗获益预测的监督信号结合起来训练。关键点在于模型不是预测所有样本同一个效果而是让每个样本都有自己的一组生存曲线预测通过注意力机制从数据中提取个体化的获益信号。3. 环境准备与前置条件3.1 运行环境复现Surv-IPTB需要Python环境建议版本3.8以上。核心依赖包括PyTorch深度学习框架建议1.10以上lifelines生存分析常用库用于KM曲线、Cox模型对比pandas、numpy数据处理scikit-learn数据划分与评估matplotlib可视化如果使用GPU建议CUDA版本与PyTorch版本匹配。没有GPU也能跑小规模示例但训练速度会明显变慢。3.2 数据要求Surv-IPTB要求的数据结构为治疗字段0或1表示是否接受治疗。生存时间字段事件发生或删失的时间。事件字段0表示删失1表示事件发生。多个协变量字段可以是数值型、类别型或生物标志物。数据中必须同时包含治疗组和对照组样本否则无法估计治疗获益。3.3 环境搭建示例# 创建虚拟环境 python -m venv surv-iptb-env source surv-iptb-env/bin/activate # 安装基础依赖 pip install torch --index-url https://download.pytorch.org/whl/cu118 pip install pandas numpy scikit-learn lifelines matplotlib在安装PyTorch时请根据你的CUDA版本选择合适的index-url如果你只用CPU直接执行pip install torch安装完成后可以用下面的命令检查核心依赖是否可用import torch import lifelines import pandas as pd print(PyTorch version:, torch.__version__) print(lifelines version:, lifelines.__version__) print(pandas version:, pd.__version__)4. 核心流程拆解4.1 数据预处理生存分析的最大特点是时间-事件对不能简单丢缺失值也不能直接回归时间。预处理阶段要做的事情包括将类别变量编码为数值可以用one-hot或embedding。对连续变量做标准化或归一化。构造成批次数据每个批次包含特征矩阵、治疗标签、时间、事件。这里的“事件”字段很关键。如果你的数据里删失比例过高模型训练会不稳定需要先做描述性统计。4.2 注意力网络构造模型的主体是一个带有注意力模块的全连接网络。输入经过多个隐藏层后进入注意力层最后分别输出两组结果治疗条件下的潜在结局、对照条件下的潜在结局。注意力层可以设计成简单的softmax加权[ \alpha_i \frac{\exp(f(x_i))}{\sum_j \exp(f(x_j))} ]其中 ( f ) 是一个学习映射( \alpha_i ) 是第 ( i ) 个特征或样本的注意力权重。在具体实现中要注意区分“对特征做注意力”和“对样本做注意力”。特征注意力适合维度较高的表格数据样本注意力则类似记忆网络适合样本量较大的场景。Surv-IPTB具体实现以论文源码为准但两种思路都不复杂。4.3 生存分布建模在获得两组潜在结果表示后需要估计生存函数。常见做法有离散时间模型把时间轴分成多个区间每个区间预测一个条件风险概率。连续时间模型用Cox部分的log-risk函数输出风险再结合基线生存函数。如果编码时使用离散时间模型最终的生存概率可以累乘得到[ S(t_k) \prod_{j1}^{k} (1 - h_j) ]其中 ( h_j ) 是第 ( j ) 个时间区间的条件风险概率。Surv-IPTB 的核心思想之一就是让模型对每个个体输出两条生存曲线一条来自治疗状态一条来自对照状态。比较这两条曲线就能得到每个个体的获益概率。4.4 训练损失函数训练时需要同时优化两类目标生存预测准确性让预测的生存曲线尽量拟合真实时间-事件分布。可以使用基于似然的损失例如负对数部分似然损失。治疗获益预测一致性如果训练集中存在某些个体已知治疗获益方向可以加入相应的排序损失或二分类损失。但这里有个很容易犯错的地方生存数据中我们不知道同一个体“未接受治疗”的结局。因此反事实部分的损失只能通过平衡两组样本的分布来间接优化而不是直接监督。更稳妥的训练策略是采用潜在结果框架把治疗组和对照组分别建模但在低维表示层共享参数。使用对抗训练或平衡权重来降低两组特征分布的差异。最终预测时对同一 ( x )同时送入治疗分支和对照分支。4.5 预测与评估预测阶段对每个样本 ( x )计算治疗分支的生存曲线 ( S_1(t|x) )。计算对照分支的生存曲线 ( S_0(t|x) )。在指定时间点 ( t_0 ) 比较生存概率或者比较RMST。输出获益概率或获益评分。评估时不能只看训练集的AUC还要验证在删失数据下的稳定性。比较常用的评估指标包括治疗组和对照组的预后C-index。校准曲线。在验证集上按预测获益概率分组的生存曲线是否分离良好。5. 完整示例与代码实现由于Surv-IPTB不同版本结构可能不同这里用一个最小可运行的演示工程来展示从数据构造到模型训练、预测的全部流程。这个示例使用了模拟数据重点在于让你理解模型如何组织、训练和验证而不是替换原论文实现。5.1 模拟生存数据# 文件路径generate_demo_data.py import numpy as np import pandas as pd np.random.seed(42) n_samples 2000 n_features 8 # 随机生成特征 X np.random.randn(n_samples, n_features) # 随机分配治疗 treatment np.random.binomial(1, 0.5, sizen_samples) # 制造一个与治疗获益相关的特征特征0与性别/基因型相关 # 假设当 feature0 0 时治疗能降低事件风险 benefit_flag (X[:, 0] 0).astype(int) # 治疗组的风险系数低风险者获益更大 base_risk 0.5 * X[:, 1] 0.3 * X[:, 2] treatment_effect -0.8 * benefit_flag log_risk base_risk treatment_effect * treatment # 生成事件时间指数分布 time np.random.exponential(scale1.0 / np.exp(log_risk), sizen_samples) # 生成删失随访截止时间 censor_time np.random.uniform(0.5, 3.0, sizen_samples) event (time censor_time).astype(int) observed_time np.minimum(time, censor_time) df pd.DataFrame(X, columns[fx{i} for i in range(n_features)]) df[treatment] treatment df[time] observed_time df[event] event df.to_csv(demo_survival_data.csv, indexFalse) print(df.head()) print(删失比例, 1 - df[event].mean())这个模拟数据中x0 0的个体治疗获益更大x1、x2影响基础风险。你可以用这个数据验证模型能否学到“只有特定人群获益”的异质性。5.2 数据加载与统一编码# 文件路径preprocess.py import pandas as pd from sklearn.preprocessing import StandardScaler def load_data(pathdemo_survival_data.csv): df pd.read_csv(path) feature_cols [c for c in df.columns if c.startswith(x)] treatment_col treatment time_col time event_col event scaler StandardScaler() X scaler.fit_transform(df[feature_cols]) treatment df[treatment_col].values.astype(np.float32) time df[time_col].values.astype(np.float32) event df[event_col].values.astype(np.float32) return X, treatment, time, event, scaler if __name__ __main__: X, treatment, time, event, _ load_data() print(特征矩阵大小:, X.shape) print(治疗样本比例:, treatment.mean())标准化很重要因为网络中的注意力权重对特征尺度敏感。如果不做标准化数值范围大的特征会主导注意力权重导致模型学到错误的“重要性”。5.3 定义注意力生存网络下面是核心模型定义包含特征注意力模块和两个潜在结果分支。# 文件路径model.py import torch import torch.nn as nn import torch.nn.functional as F class AttentionSurvivalNet(nn.Module): def __init__(self, n_features, n_time_bins10): super().__init__() self.n_time_bins n_time_bins # 共享编码器 self.encoder nn.Sequential( nn.Linear(n_features, 64), nn.ReLU(), nn.Linear(64, 32), nn.ReLU(), ) # 注意力打分网络 self.attention nn.Sequential( nn.Linear(32, 16), nn.Tanh(), nn.Linear(16, 1), ) # 治疗分支 self.treatment_head nn.Sequential( nn.Linear(32, 16), nn.ReLU(), nn.Linear(16, n_time_bins), ) # 对照分支 self.control_head nn.Sequential( nn.Linear(32, 16), nn.ReLU(), nn.Linear(16, n_time_bins), ) def forward(self, x, treatmentNone): h self.encoder(x) # (batch, 32) # 特征注意力对每个特征计算权重 att_scores self.attention(h) # (batch, 1) att_weights torch.softmax(att_scores, dim1) # (batch, 1) # 用注意力权重调制编码表示 h_att h * att_weights # 分别预测治疗和对照的离散风险 treat_logits self.treatment_head(h_att) # (batch, n_time_bins) control_logits self.control_head(h_att) # 转成条件风险概率 treat_hazard torch.sigmoid(treat_logits) control_hazard torch.sigmoid(control_logits) return treat_hazard, control_hazard def predict_survival(self, x): with torch.no_grad(): treat_hazard, control_hazard self.forward(x) treat_surv torch.cumprod(1 - treat_hazard, dim1) control_surv torch.cumprod(1 - control_hazard, dim1) return treat_surv, control_surv在这个示例中注意力权重是在特征维度上计算的作用是强化对当前样本影响更大的特征信号。实际论文版本可能还会加入时间维度的注意力或者样本级别的注意力这里提供一个最小可运行的结构。5.4 训练循环训练时需要把时间离散化成区间。这里把时间分位数作为区间边界每个样本根据观察时间落在哪个区间来计算离散时间的条件概率损失。# 文件路径train.py import numpy as np import torch import torch.nn as nn from torch.utils.data import DataLoader, TensorDataset from preprocess import load_data from model import AttentionSurvivalNet def make_time_bins(time, n_bins10): # 使用事件时间分位数作为边界 boundaries np.quantile(time[time 0], np.linspace(0, 1, n_bins1)[1:-1]) return np.unique(boundaries) def time_to_bin(time, bins): # 找时间区间索引 return np.searchsorted(bins, time, sideright) X, treatment, time, event, _ load_data() bins make_time_bins(time, n_bins10) time_bin time_to_bin(time, bins) # 转换为张量 X_t torch.tensor(X, dtypetorch.float32) treat_t torch.tensor(treatment, dtypetorch.float32) time_t torch.tensor(time_bin, dtypetorch.long) event_t torch.tensor(event, dtypetorch.float32) dataset TensorDataset(X_t, treat_t, time_t, event_t) loader DataLoader(dataset, batch_size128, shuffleTrue) model AttentionSurvivalNet(n_featuresX.shape[1], n_time_binslen(bins)1) optimizer torch.optim.Adam(model.parameters(), lr1e-3) def discrete_survival_loss(treat_hazard, control_hazard, treatment, time_bin, event): # 选择实际治疗对应的风险 hazard torch.where(treatment.unsqueeze(1) 0.5, treat_hazard, control_hazard) # 计算该样本在观测时间之前的生存概率和观测时点的风险 prob_uncensored 0 # 简化处理用离散近似 surv torch.cumprod(1 - hazard, dim1) # 对于事件样本我们希望观测区间风险高删失样本只希望之前生存率高 # 这里用一个近似损失事件样本最大化生存到区间前的概率 * 区间风险 # 删失样本最大化生存到删失区间的概率 loss 0 for i in range(len(time_bin)): t time_bin[i].item() if event[i] 0: p surv[i, t-1] * hazard[i, t] if t 0 else hazard[i, 0] else: p surv[i, t] loss - torch.log(p 1e-8) return loss / len(time_bin) for epoch in range(20): total_loss 0 model.train() for xb, tb, timeb, evb in loader: optimizer.zero_grad() treat_h, control_h model(xb) loss discrete_survival_loss(treat_h, control_h, tb, timeb, evb) loss.backward() optimizer.step() total_loss loss.item() print(fEpoch {epoch1}, loss: {total_loss/len(loader):.4f})注意这里为了演示损失函数做了简化。真实复现时离散生存模型应使用基于条件风险概率的完整似然函数并且要处理区间右删失、左截断等问题。小规模模拟数据上这个简化版本已经可以跑通梯度传播。5.5 预测个体获益概率训练完成后对每个样本输出治疗组和对照组的生存曲线并计算指定时间点的获益概率。# 文件路径predict.py import torch import numpy as np from preprocess import load_data from model import AttentionSurvivalNet from train import make_time_bins X, treatment, time, event, _ load_data() bins make_time_bins(time, n_bins10) model AttentionSurvivalNet(n_featuresX.shape[1], n_time_binslen(bins)1) # 这里假设你已经保存了训练好的模型权重 # model.load_state_dict(torch.load(surv_iptb_model.pth)) model.eval() with torch.no_grad(): treat_surv, control_surv model.predict_survival(torch.tensor(X, dtypetorch.float32)) # 选择评估时间点例如中位随访时间 eval_time np.median(time) eval_bin time_to_bin(np.array([eval_time]), bins)[0] # 获益定义治疗组生存率 - 对照组生存率0 treat_surv_t treat_surv[:, eval_bin].numpy() control_surv_t control_surv[:, eval_bin].numpy() benefit_prob (treat_surv_t control_surv_t).astype(float) benefit_diff treat_surv_t - control_surv_t print(预测获益个体比例, benefit_prob.mean()) print(平均获益差值, benefit_diff.mean())这里得到的benefit_prob是一个经验概率判断。如果你希望输出更平滑的概率可以基于benefit_diff或两种生存曲线距离做回归校准。6. 运行结果与效果验证6.1 运行命令将上面的代码文件放在同一目录下依次执行python generate_demo_data.py python train.py python predict.py6.2 预期输出generate_demo_data.py会打印前5行数据和删失比例。示例中删失比例一般在30%左右。train.py会输出每个epoch的lossloss整体应呈下降趋势例如Epoch 1, loss: 1.8932 Epoch 2, loss: 1.7411 ... Epoch 20, loss: 1.4325predict.py会输出预测获益个体比例和平均获益差值。因为模拟数据中约一半个体是受益者预测结果应接近0.5附近且平均获益差值为正。6.3 如何判断模型真的学到了个体异质性最简单的方法检查预测获益概率在不同真实获益人群中的分布。在模拟数据中我们已经知道x0 0时治疗效应为负风险降低x0 0时无治疗效应。可以统计两组人群的预测获益均值# 检查模型是否学会异质性 true_benefit (X[:, 0] 0).astype(int) from sklearn.metrics import roc_auc_score auc roc_auc_score(true_benefit, benefit_diff) print(基于获益差值的AUC, auc)如果AUC远大于0.5说明模型成功区分了获益者和非获益者如果接近0.5说明模型没有学到异质性需要检查数据噪声或网络容量。6.4 与Cox模型对比为了说明Surv-IPTB的优势可以训练一个包含“治疗×特征交互项”的Cox模型作为baseline。from lifelines import CoxPHFitter # 在df中添加交互项 import pandas as pd df pd.read_csv(demo_survival_data.csv) df[trt_x0] df[treatment] * df[x0] cph CoxPHFitter() cph.fit(df[[time, event, treatment, x0, x1, x2, trt_x0]], duration_coltime, event_colevent) print(cph.summary)看交互项trt_x0的系数是否显著且为负。它代表x0越大的病人治疗带来的风险下降越多。如果Cox模型也能正确发现这个交互项说明问题相对简单如果你的真实数据交互项是“多特征组合才有效”Cox模型就很可能漏掉而注意力模型更有机会捕获。6.5 验证失败时排查顺序如果训练后AUC接近0.5按以下顺序排查数据是否有足够事件数时间离散化是否把有效信息丢掉了注意力层是否退化成均匀权重学习率是否过大导致梯度不稳定训练轮数是否太少7. 常见问题与排查思路问题现象可能原因排查方式解决方案训练loss不下降学习率过大或网络结构问题打印梯度范数尝试降低学习率使用Adam默认学习率增加归一化层预测获益比例接近0或1类别不平衡或模型过拟合检查训练集获益比例观察验证集效果增加数据使用早停降低网络容量注意力权重几乎相等注意力打分网络退化打印注意力权重统计增加注意力网络复杂度或使用温度参数训练完成但C-index低生存曲线预测不准分别评估治疗组和对照组的C-index增加时间区间数调整隐藏层维度手动指定时间点后获益概率不稳定生存曲线在该点附近波动大画多条样本的生存曲线改用RMST作为获益定义或对曲线做平滑删失比例过高模型无法收敛事件信息不足查看事件比例分布考虑改用条件风险模型降低时间区间数7.1 关于删失数据的误区很多初学者在训练时直接丢弃删失样本这是最常见的错误。删失样本虽然“事件时间未知”但它提供了“至少存活到某个时间点”的信息对估计生存曲线非常重要。Surv-IPTB这类模型在设计时就考虑了删失因此不要为了方便而删除删失样本。7.2 关于反事实推断的局限性无论模型多复杂都不可能完全消除反事实推断的固有缺陷。观测数据中的治疗分配可能存在选择偏差病情重的病人可能更多接受治疗这时即使治疗有效治疗组生存率也可能低于对照组。在实际应用中要结合倾向评分加权、逆概率加权等方法先做数据平衡再训练Surv-IPTB。8. 最佳实践与工程建议8.1 数据层面治疗字段必须是明确的二值变量不要用“实际用药时长”代替。协变量要统一标准化。事件定义要一致尽量避免竞争风险混入。如果样本量少建议使用交叉验证代替单一划分。8.2 模型层面不要盲目堆深度。表格数据上2到3层隐藏层往往足够。注意力权重是解释性工具要定期检查是否出现病态权重。在最终评估时既要看整体AUC也要看按风险分层后的获益概率分布。对生存曲线的不确定性做区间估计可以使用Dropout或深度集成。8.3 业务落地层面个体获益概率预测模型进入临床应用前必须经过外部验证。仅在一个数据集上表现好不能保证在另一个医院、另一种人群上表现稳定。建议先以“辅助筛选高风险获益人群”为目标做回顾性研究再用真实世界数据做前瞻性验证。并且模型输出的是概率不是确定性结论。在风险较高或治疗成本较高的场景要设置更保守的获益概率阈值。8.4 工程实现层面将数据读取、特征工程、模型定义、训练、评估拆分成独立模块。用配置文件管理超参数方便复现。保存模型时同时保存特征标准化器。记录训练数据的时间范围、事件定义、删失比例便于后续审计。一个推荐的配置文件示例# config.yaml data: path: demo_survival_data.csv features: x0,x1,x2,x3,x4,x5,x6,x7 treatment: treatment time: time event: event model: hidden_dim: 64 attention_dim: 32 n_time_bins: 10 dropout: 0.2 train: batch_size: 128 epochs: 50 lr: 0.001 weight_decay: 1e-5 eval: time_point: median benefit_threshold: 0.5使用配置文件后调参时不需要改代码直接修改yaml即可这在复现学术项目时特别重要。8.5 安全与伦理提醒个体治疗获益预测涉及医疗决策存在隐私和伦理边界。使用真实患者数据时要确保数据获取和使用符合相关法律法规和伦理审查要求不能在未经授权的情况下将模型用于临床决策。本文所述代码只用于技术学习和模拟数据演示不能直接作为医疗诊断依据。9. 总结与后续学习方向Surv-IPTB 把“注意力机制”和“生存数据”结合目标是把疗效评估从群体平均数推进到个体获益概率。本文重点解释了它解决的问题、模型结构、数据格式、训练思路和验证方法并给出了一个可运行的最小示例。这个示例虽然简化了损失函数但足够帮你理解完整流程构建数据、设计注意力网络、输出两组潜在结果生存曲线、比较生存曲线得到获益概率。后续你可以往这几个方向继续深入阅读Surv-IPTB论文原文确认作者使用的损失函数、注意力结构和评估指标。将模型替换为更成熟的离散生存模型损失函数如DeepHit中的事件特定风险函数。在真实生存数据上比较Cox、随机生存森林和Surv-IPTB的个体获益预测能力。引入倾向评分平衡提高观测数据下的反事实估计可靠性。使用可解释性工具如SHAP分析注意力权重与特征重要性的关系。如果你正好在研究个体化治疗决策、药物响应预测或真实世界生存数据分析这个方向值得投入时间。关键在于不要把Surv-IPTB当成一个“能直接给出答案的工具”而是把它看作一个“帮助临床和研究者提出更好问题的框架”。先跑通示例再理解每一步在做什么最后再迁移到自己的数据上这条路径是最稳妥的。