arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1808.09679cs.CV
收藏

使用CNN进行肺癌患者的基于图像的生存分析

Image-based Survival Analysis for Lung Cancer Patients using CNNs

RWTH Aachen University(亚琛工业大学)

Christoph Haarburger, Philippe Weitz, Oliver Rippel, Dorit Merhof

摘要:

传统的生存模型,如Cox比例风险模型,通常基于标量或分类的临床特征。随着越来越大的图像数据集的出现,将定量图像特征纳入生存预测已经变得可行。到目前为止,这种分析主要基于放射组学特征,即,先验地在数学上定义的固定特征集。为了捕获高度抽象的信息,期望使用卷积神经网络来学习特征提取。然而,对于断层摄影医学图像,模型训练是困难的,因为一方面,只有很少的3D图像数据样本一次适合一个批次,另一方面,生存损失函数本质上是需要大批次大小的排序度量。在这项工作中,我们表明,通过将生存分析简化为中值生存分类,卷积神经网络可以用小批量进行训练,并学习预测生存的特征,以及端到端的危险预测网络。在公开的肺癌数据集中,我们的方法优于先前的最新技术水平。

英文摘要:

Traditional survival models such as the Cox proportional hazards model are typically based on scalar or categorical clinical features. With the advent of increasingly large image datasets, it has become feasible to incorporate quantitative image features into survival prediction. So far, this kind of analysis is mostly based on radiomics features, i.e. a fixed set of features that is mathematically defined a priori. To capture highly abstract information, it is desirable to learn the feature extraction using convolutional neural networks. However, for tomographic medical images, model training is difficult because on the one hand, only few samples of 3D image data fit into one batch at once and on the other hand, survival loss functions are essentially ordering measures that require large batch sizes. In this work, we show that by simplifying survival analysis to median survival classification, convolutional neural networks can be trained with small batch sizes and learn features that predict survival equally well as end-to-end hazard prediction networks. Our approach outperforms the previous state of the art in a publicly available lung cancer dataset.

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

↑