使用深度神经网络对肿瘤图像进行分类

简介: 使用 Inception-v3 深度神经网络对可能不适合内存的多分辨率全玻片图像 (WSI) 进行分类。用于肿瘤分类的深度学习方法依赖于数字病理学,其中整个组织切片被成像和数字化。生成的 WSI 具有高分辨率,大约为 200,000 x 100,000 像素。WSI 通常以多分辨率格式存储,以促进图像的高效显示、导航和处理。

一、前言
使用 Inception-v3 深度神经网络对可能不适合内存的多分辨率全玻片图像 (WSI) 进行分类。
用于肿瘤分类的深度学习方法依赖于数字病理学,其中整个组织切片被成像和数字化。生成的 WSI 具有高分辨率,大约为 200,000 x 100,000 像素。WSI 通常以多分辨率格式存储,以促进图像的高效显示、导航和处理。
概述使用基于块的处理来训练大型 WSI 的体系结构。该示例使用迁移学习技术训练基于 Inception-v3 的网络,将各个块分类。
二、准备训练数据
按照预处理训练分类网络的多分辨率图像中的说明准备训练和验证数据。预处理示例将预处理的训练和验证数据存储保存在名为 的文件中。将变量的值设置为文件所在的位置。将定型和验证数据存储加载到变量中。

三、设置用于迁移学习的 Inception-v3 网络层
此示例使用 Inception-v3 网络,这是一个卷积神经网络,在 ImageNet 数据库中的 3 多万张图像上进行训练 [48]。该网络有 1 层深,可以将图像分类为 000,<> 个对象类别,例如键盘、鼠标、铅笔和许多动物。

inceptionv3(深度学习工具箱)函数返回一个预先训练的 Inception-v3 网络。Inception-v3 需要 Inception-v3 Network 的深度学习工具箱模型支持包。如果未安装此支持包,则该函数将提供下载链接。

网络的卷积层提取图像特征。最后一个可学习层和最后一个分类层使用图像特征对输入图像进行分类。这两个图层包含有关如何将要素组合到类概率、损失值和预测标签中的信息。要重新训练预训练网络以对新图像进行分类,请将这两个层替换为适应新数据集的新层。
此示例的目标是在两个类,为两个类创建新的全连接图层。将最终的全连接层替换为新层。为两个类创建新的分类图层。将最终分类图层替换为新图层。

三、指定训练选项
使用均方根传播 (RMSProp) 优化训练网络。使用深度学习工具箱指定 RMSProp 的超参数设置。
减少到一个小数字,因为大量的训练数据使网络能够更快地实现收敛。根据您的可用 GPU 内存指定。虽然较大的小批量大小可以使训练更快,但较大的大小会降低网络的泛化能力。设置为 以防止完全读取训练数据以计算规范化统计信息。

四、训练网络或下载预训练网络
默认情况下,此示例使用帮助程序函数下载已训练分类网络的预训练版本。预训练网络可用于运行整个示例,而无需等待训练完成。

五、下载测试数据
Camelyon16测试数据集由130个WSI组成。这些图像具有正常组织和肿瘤组织。每个文件的大小约为 2 GB。要下载测试数据,请访问Camelyon17网站,然后单击第一个“CAMELYON16数据集”链接。打开“测试”目录,然后按照以下步骤操作。

六、预处理测试数据
创建阻止的映像对象以管理测试映像。获取测试映像的文件名。然后,创建管理测试图像的阻止映像对象数组。每个对象都指向磁盘上的相应映像文件。
创建组织面膜,为了有效地处理WSI数据,请为每个测试图像创建一个组织掩模。此过程与用于预处理正常训练图像的过程相同。组织面膜只有一个级别,并且足够小以适合记忆。使用帮助程序功能在图像浏览器应用程序中显示组织蒙版。
1.png

预处理肿瘤地面真相图像。指定肿瘤掩膜的分辨率级别。

七、预测肿瘤概率的热图
使用经过训练的分类网络预测每个测试图像的热图。热图给出了每个块属于该类的概率分数。该示例对每个测试映像执行以下操作以创建热图:
可视化热图,选择要显示的测试图像。在图形的左侧,使用帮助程序函数将地面实况边界坐标显示为手绘 ROI。此帮助程序函数作为支持文件附加到示例。正常区域(用绿色边界显示)可以发生在肿瘤区域(用红色边界显示)内。
在图的右侧,显示测试图像的热图。链接轴并放大到感兴趣区域。
2.png

八、在特定阈值下对测试图像进行分类
要将块分类,请对热图概率值应用阈值。选择一个阈值概率,高于该阈值的块被分类为 .理想情况下,您将使用验证数据集上的受试者工作特征 (ROC) 或精度-召回率曲线来计算此阈值。对每个测试图像中的块进行分类,并使用应用函数和帮助程序函数定义的处理操作计算混淆矩阵。
计算所有测试图像的全局混淆矩阵。显示归一化全局混淆矩阵的混淆图。WSI图像中的大多数块都是正常组织的,导致高比例的真阴性预测。

3.png

可视化分类结果
将真实 ROI 边界坐标与分类结果进行比较。在图形的左侧,将地面实况边界坐标显示为手绘 ROI。在图的右侧,显示测试图像,并根据混淆矩阵在每个块上叠加颜色。将真阳性显示为红色,将误报显示为青色,将漏报显示为黄色,将真阴性显示为无色。

假阴性和假阳性出现在肿瘤区域的边缘,这表明网络难以对具有部分类的块进行分类。
4.png

注意:为了减少肿瘤周边的分类误差,您可以使用较少的同质块重新训练网络。预处理训练数据集的块时,请减小名称-值参数的值。

九、使用 AUC-ROC 曲线量化网络预测
使用帮助程序函数计算不同阈值下的 ROC 曲线值。计算曲线下面积 (AUC) 指标。该指标返回 [0, 1] 范围内的值,其中 1 表示完美的模型性能。此数据集的 AUC 接近 1。您可以使用 AUC 微调训练过程。
5.png

目录
相关文章
|
1月前
|
机器学习/深度学习 自然语言处理 算法
【从零开始学习深度学习】49.Pytorch_NLP项目实战:文本情感分类---使用循环神经网络RNN
【从零开始学习深度学习】49.Pytorch_NLP项目实战:文本情感分类---使用循环神经网络RNN
|
20天前
|
算法 定位技术 网络架构
网络的分类与性能指标
可以分为广域网(WAN)、城域网(MAN)、局域网(LAN)、个人区域网(PAN)。
30 4
|
20天前
|
机器学习/深度学习 自然语言处理 搜索推荐
深度学习之分类网络
深度学习的分类网络(Classification Networks)是用于将输入数据分配到预定义类别的神经网络。它们广泛应用于图像分类、文本分类、语音识别等任务。以下是对深度学习分类网络的详细介绍,包括其基本概念、主要架构、常见模型、应用场景、优缺点及未来发展方向。
47 4
|
2月前
|
机器学习/深度学习 JSON PyTorch
图神经网络入门示例:使用PyTorch Geometric 进行节点分类
本文介绍了如何使用PyTorch处理同构图数据进行节点分类。首先,数据集来自Facebook Large Page-Page Network,包含22,470个页面,分为四类,具有不同大小的特征向量。为训练神经网络,需创建PyTorch Data对象,涉及读取CSV和JSON文件,处理不一致的特征向量大小并进行归一化。接着,加载边数据以构建图。通过`Data`对象创建同构图,之后数据被分为70%训练集和30%测试集。训练了两种模型:MLP和GCN。GCN在测试集上实现了80%的准确率,优于MLP的46%,展示了利用图信息的优势。
42 1
|
2月前
|
机器学习/深度学习 监控 自动驾驶
【传知代码】从零开始搭建图像去雾神经网络-论文复现
本文介绍了基于集成学习的双分支非均匀去雾神经网络的复现,该网络由迁移学习子网和数据拟合子网组成,分别处理全局表示和数据拟合。网络使用Res2Net作为编码器,并结合通道和像素注意力模块。代码可在提供的链接下载。网络在交通监控、自动驾驶、航海和目标跟踪等领域有广泛应用,通过提升图像质量来提高系统性能。实验在O-Haze、I-Haze和NH-Haze数据集上进行,展示了网络在去除雾霾方面的效果,尽管存在细节模糊和色彩饱和度低的问题。
|
1月前
|
机器学习/深度学习 算法 PyTorch
【从零开始学习深度学习】50.Pytorch_NLP项目实战:卷积神经网络textCNN在文本情感分类的运用
【从零开始学习深度学习】50.Pytorch_NLP项目实战:卷积神经网络textCNN在文本情感分类的运用
|
2月前
|
人工智能 数据可视化
【数据分享】维基百科Wiki负面有害评论(网络暴力)文本数据多标签分类挖掘可视化
【数据分享】维基百科Wiki负面有害评论(网络暴力)文本数据多标签分类挖掘可视化
|
2月前
|
数据挖掘 开发工具 Android开发
R语言对git安卓包分类统计、聚类、复杂网络可视化分析
R语言对git安卓包分类统计、聚类、复杂网络可视化分析
|
2月前
|
机器学习/深度学习 并行计算 算法
MATLAB|【免费】概率神经网络的分类预测--基于PNN的变压器故障诊断
MATLAB|【免费】概率神经网络的分类预测--基于PNN的变压器故障诊断
|
2月前
|
机器学习/深度学习 PyTorch 算法框架/工具
Python用GAN生成对抗性神经网络判别模型拟合多维数组、分类识别手写数字图像可视化
Python用GAN生成对抗性神经网络判别模型拟合多维数组、分类识别手写数字图像可视化

相关实验场景

更多