arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.06992cs.LGstat.AP
收藏

ResSurv:基于残差网络的癌症生存分析预测模型

ResSurv: Cancer Survival Analysis Prediction Model Based on Residual Networks

Dalian Maritime University(大连海事大学)

Wankang Zhai

AI总结:

针对 Cox 深度学习模型在高通量基因组数据上的过拟合与网络退化问题,提出融合残差结构和 CPH 偏似然损失的 ResSurv,实验表明其达到 SOTA。

摘要:

生存预测是癌症预后分析的一个重要分支。通过 TCGA 基因组学数据预测生存风险的模型能够发现与癌症相关的基因,并根据患者特征提供诊断和治疗建议。我们发现,基于 Cox 比例风险的深度学习模型在处理高通量数据时经常出现过拟合。此外,我们发现,随着网络层数增加,实验结果并不会变得更好,反而会出现网络退化。针对这一问题,我们提出了一种基于深度残差学习的新框架,将 Cox 比例风险与残差思想相结合,并将其命名为 ResSurv。首先,ResSurv 是一个由多个基本 ResNet Block 堆叠而成的前馈深度学习网络。在每个 ResNet Block 中,我们加入归一化层,以防止梯度消失和梯度爆炸。其次,对于神经网络的损失函数,我们继承了 Cox 比例风险方法,将 CPH 模型的半参数方法应用于神经网络,并结合偏似然模型建立损失函数,进而执行反向传播和梯度更新。最后,我们比较了不同深度的 ResSurv 网络,发现其能够有效提取高维特征。消融实验和对比实验证明,我们的模型在深度学习领域达到了 SOTA(state of the art)水平,并且该网络能够有效提取深层信息。

英文摘要:

Survival prediction is an important branch of cancer prognosis analysis. The model that predicts survival risk through TCGA genomics data can discover genes related to cancer and provide diagnosis and treatment recommendations based on patient characteristics. We found that deep learning models based on Cox proportional hazards often suffer from overfitting when dealing with high-throughput data. Moreover, we found that as the number of network layers increases, the experimental results will not get better, and network degradation will occur. Based on this problem, we propose a new framework based on Deep Residual Learning. Combine the ideas of Cox proportional hazards and Residual. And name it ResSurv. First, ResSurv is a feed-forward deep learning network stacked by multiple basic ResNet Blocks. In each ResNet Block, we add a Normalization Layer to prevent gradient disappearance and gradient explosion. Secondly, for the loss function of the neural network, we inherited the Cox proportional hazards methods, applied the semi-parametric of the CPH model to the neural network, combined with the partial likelihood model, established the loss function, and performed backpropagation and gradient update. Finally, we compared ResSurv networks of different depths and found that we can effectively extract high-dimensional features. Ablation experiments and comparative experiments prove that our model has reached SOTA(state of the art) in the field of deep learning, and our network can effectively extract deep information.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑