arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2108.13661q-bio.QMeess.IV

使用卷积神经网络对乳腺癌图像进行分类

Using convolutional neural networks for the classification of breast cancer images

Eric Bonnet

AI总结:

本研究比较了12种CNN和4种硬件加速设备在两个公开乳腺癌组织图像数据集上的分类表现,发现增加模型深度可提升性能但非线性增加训练时间,完全重训优于迁移学习,微调超参数使最佳模型达到SOTA水平。

摘要:

乳腺癌分期的一个重要部分是评估前哨腋窝淋巴结是否有肿瘤扩散的早期迹象。然而,病理学家进行这种评估并不总是容易的,回顾性调查常常会重新判定高比例前哨淋巴结的状态。卷积神经网络是一类深度学习算法,在最具挑战性的视觉分类任务中表现优异,并在医学图像领域有众多应用。在本研究中,我比较了十二种不同的CNN和不同的硬件加速设备,用于从乳腺癌组织的显微图像中检测乳腺癌。卷积模型在两个公开数据集上进行训练和测试。第一个数据集由来自乳腺癌患者的超过300,000张前哨淋巴结组织图像组成,而第二个数据集包含超过220,000张来自浸润性乳腺癌组织(最常见的乳腺癌形式之一)的图像。使用四种不同的硬件加速卡和现成的深度学习框架。测试了迁移学习和超参数微调的影响。硬件加速设备的性能可以将训练时间缩短5到12倍,具体取决于所使用的模型。另一方面,增加卷积深度会使训练时间增加4到6倍,具体取决于所使用的加速设备。增加模型的深度和复杂度通常能提升性能,但两者关系并非线性,且也取决于模型的架构。与完全重新训练模型相比,迁移学习的性能总是更差。微调模型的超参数可以改善结果,其中最佳模型的性能与state-of-the-art模型相当。

英文摘要:

An important part of breast cancer staging is the assessment of the sentinel axillary node for early signs of tumor spreading. However, this assessment by pathologists is not always easy and retrospective surveys often requalify the status of a high proportion of sentinel nodes. Convolutional Neural Networks (CNNs) are a class of deep learning algorithms that have shown excellent performances in the most challenging visual classification tasks, with numerous applications in medical imaging. In this study I compare twelve different CNNs and different hardware acceleration devices for the detection of breast cancer from microscopic images of breast cancer tissue. Convolutional models are trained and tested on two public datasets. The first one is composed of more than 300,000 images of sentinel lymph node tissue from breast cancer patients, while the second one has more than 220,000 images from inductive breast carcinoma tissue, one of the most common forms of breast cancer. Four different hardware acceleration cards were used, with an off-the-shelf deep learning framework. The impact of transfer learning and hyperparameters fine-tuning are tested. Hardware acceleration device performance can improve training time by a factor of five to twelve, depending on the model used. On the other hand, increasing convolutional depth will augment the training time by a factor of four to six times, depending on the acceleration device used. Increasing the depth and the complexity of the model generally improves performance, but the relationship is not linear and also depends on the architecture of the model. The performance of transfer learning is always worse compared to a complete retraining of the model. Fine-tuning the hyperparameters of the model improves the results, with the best model showing a performance comparable to state-of-the-art models.

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

↑