arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.04251eess.IVcs.CV

2.75D:通过将3D医学影像表示为2D特征来提升小数据学习

2.75D: Boosting learning by representing 3D Medical imaging to 2D features for small data

Netherlands Cancer Institute(荷兰癌症研究所);Erasmus Medical Center Rotterdam(鹿特丹伊拉斯姆斯医学中心);Radboud University Medical Center(拉德堡德大学医学中心);Southern University of Science and Technology(南方科技大学);Aberystwyth University(阿伯里斯特威斯大学)

Xin Wang, Ruisheng Su, Weiyi Xie, Wenjin Wang, Yi Xu, Ritse Mann, Jungong Han, Tao Tan

AI总结:

提出2.75D螺旋旋转表示,将3D医学影像压缩为2D特征,并结合2.75Dx3多视图与预训练2D CNN,在小数据分类中取得优于或可比3D方法的性能且显著降低计算时间。

摘要:

在医学数据驱动的学习中,3D卷积神经网络(CNNs)已开始在众多深度学习任务中表现出优于2D CNNs的性能,证明了3D空间信息在特征表示中的附加价值。然而,收集更多训练样本以实现收敛的困难、更多的计算资源以及更长的执行时间,使这种方法应用较少。此外,由于缺乏公开可用的预训练3D模型,在3D CNN上应用迁移学习具有挑战性。为解决这些问题,我们提出了一种新颖的体数据2D策略表示,即2.75D。在本研究中,3D图像的空间信息通过螺旋旋转技术在单个2D视图中被捕获。因此,2D CNN网络也可用于学习体信息。此外,我们能够充分利用预训练2D CNNs来处理下游视觉问题。我们还探索了多视图2.75D策略,即2.75D 3通道(2.75Dx3),以增强2.75D的优势。我们在三个具有不同模态或器官的公开数据集(肺部CT、乳腺MRI和前列腺MRI)上评估了所提出的方法,并在分类任务中与对应的2D、2.5D和3D方法进行比较。结果表明,当所有方法在肺部数据集上从头开始训练时,所提出的方法显著优于其他对应方法。在使用迁移学习或训练数据有限的情况下,这种性能提升更加明显。我们的方法在其他数据集上也取得了相当的性能。此外,与2.5D或3D方法相比,我们的方法在训练和推理的时间消耗上实现了大幅降低。

英文摘要:

In medical-data driven learning, 3D convolutional neural networks (CNNs) have started to show superior performance to 2D CNNs in numerous deep learning tasks, proving the added value of 3D spatial information in feature representation. However, the difficulty in collecting more training samples to converge, more computational resources and longer execution time make this approach less applied. Also, applying transfer learning on 3D CNN is challenging due to a lack of publicly available pre-trained 3D models. To tackle these issues, we proposed a novel 2D strategical representation of volumetric data, namely 2.75D. In this work, the spatial information of 3D images is captured in a single 2D view by a spiral-spinning technique. As a result, 2D CNN networks can also be used to learn volumetric information. Besides, we can fully leverage pre-trained 2D CNNs for downstream vision problems. We also explore a multi-view 2.75D strategy, 2.75D 3 channels (2.75Dx3), to boost the advantage of 2.75D. We evaluated the proposed methods on three public datasets with different modalities or organs (Lung CT, Breast MRI, and Prostate MRI), against their 2D, 2.5D, and 3D counterparts in classification tasks. Results show that the proposed methods significantly outperform other counterparts when all methods were trained from scratch on the lung dataset. Such performance gain is more pronounced with transfer learning or in the case of limited training data. Our methods also achieved comparable performance on other datasets. In addition, our methods achieved a substantial reduction in time consumption of training and inference compared with the 2.5D or 3D method.

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

↑