arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2006.08997cs.LGstat.ML
收藏

离散时间Cox模型的联合生存分析

Federated Survival Analysis with Discrete-Time Cox Models

Owkin Inc.(Owkin公司);Hospital Israelita Albert Einstein(爱因斯坦以色列医院);French Red Cross Innovation(法国红十字会创新中心)

Mathieu Andreux, Andre Manoel, Romuald Menuet, Charlie Saillard, Chloé Simpson

摘要:

使用联邦学习(FL)从位于不同中心的分散数据集构建机器学习模型是一种很有前途的方法,可以在保护隐私的同时规避本地数据稀缺性。然而,用于生存分析的显著Cox比例风险(PH)模型不适合FL框架,因为其损失函数相对于样本不可分离。绕过这种不可分离性的简单方法包括计算每个中心的损失,并最小化它们的总和作为真实损失的近似值。我们表明,由此产生的模型可能会遭受重要的性能损失,在一些不利的设置。相反,我们利用Cox PH模型的离散时间扩展,将生存分析作为一个具有可分离损失函数的分类问题。使用这种方法,我们使用标准FL技术对合成数据以及来自癌症基因组图谱(TCGA)的真实世界数据集进行训练生存模型,显示出与在聚合数据上训练的Cox PH模型相似的性能。与以前的工作相比,所提出的方法通信效率更高、更通用,并且更适合使用隐私保护技术。

英文摘要:

Building machine learning models from decentralized datasets located in different centers with federated learning (FL) is a promising approach to circumvent local data scarcity while preserving privacy. However, the prominent Cox proportional hazards (PH) model, used for survival analysis, does not fit the FL framework, as its loss function is non-separable with respect to the samples. The naïve method to bypass this non-separability consists in calculating the losses per center, and minimizing their sum as an approximation of the true loss. We show that the resulting model may suffer from important performance loss in some adverse settings. Instead, we leverage the discrete-time extension of the Cox PH model to formulate survival analysis as a classification problem with a separable loss function. Using this approach, we train survival models using standard FL techniques on synthetic data, as well as real-world datasets from The Cancer Genome Atlas (TCGA), showing similar performance to a Cox PH model trained on aggregated data. Compared to previous works, the proposed method is more communication-efficient, more generic, and more amenable to using privacy-preserving techniques.

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

↑