使用神经网络进行皮肤癌的诊断
点击上方“AI公园”,关注公众号,选择加“星标“或“置顶”
作者:Abhinav Sagar
编译:ronghuaiyang
导读
医疗领域的应用是计算机视觉技术落地的一大方向。
在这个博客中,我们使用卷积神经网络来解决皮肤癌的分类问题。许多癌症病例早期被误诊,导致了患者死亡等严重的后果。在本工作中,我们使用深度神经网络和迁移学习架构来解决上述问题。我们使用公共ISIC数据库来训练和测试我们的网络。我们的模型的准确率为0.935,precision为0.94,recall为0.77,F1 score为0.85,ROC- AUC为0.861,优于之前的最好的方法。
介绍
皮肤癌是世界上被诊断出的最普遍的癌症。可以看出,如果能在早期确诊,并选择适当的治疗方法,存活率是非常高的。因此,尽早了解病人的症状是否与癌症相对应是绝对必要的。传统上,医生一直使用肉眼来检测皮肤癌。尤其是当癌症还处于早期阶段时,即使是专家也难以确认。这就是计算机视觉可以帮助实现整个过程自动化的地方。这种自动化不仅可以在避免假阳性和假阴性方面有更高的效率,而且可以减少时间和手工工作。我们的模型可以应用在缺乏专家医生的地方。
数据集
我们从ISIC网站获得了一个用于皮肤癌分类的公共数据集。我们使用3000张图像进行训练,600张图像进行验证,尺寸为224×224。图像在训练集和验证集之间平均分布,如图1所示。
良性图像与恶性图像
提出的方法
我们使用迁移学习的概念来进行分类。在迁移学习中,模型不是从零开始学习,而是从解决不同问题时学到的模型开始。通过这种方式,模型利用了之前的学习,避免了从零开始。在图像分类中,迁移学习通常通过使用预先训练好的模型来表达。预训练的模型是指在大型基准测试数据集上训练来解决与我们想要解决的问题类似的问题的模型。我们使用五个预先训练好的模型 —— Inception v3,InceptionResNet v2,ResNet50,MobileNet和DenseNet169作为我们工作的预训练的权重。
Inception V3
谷歌的Inception v3架构在我们的数据集上进行了重新训练,通过对所有层进行微调,并用一个平均池化、两个全连接的顶层来替换顶层,最后是softmax层,它可以进行两个诊断类别的分类。
InceptionResNet v2
InceptionResNet v2架构在我们的数据集上进行了重新训练,通过对所有层进行微调,并将顶层替换为一个全局平均池化,一个全连接层,最后是softmax层,它可以进行两个诊断类别的分类。
ResNet50
它使用恒等映射来映射输入。这个恒等映射没有任何参数,只是将上一层的输出添加到前面的层中。层之间的跳过连接将前一层的输出添加到堆叠层的输出中。这使得他们能够训练出比以前更深层次的网络。
MobileNet
MobileNet的基本部分是深度可分离滤波器,称为深度可分离卷积。这些卷积层是一种分解卷积,将标准卷积分解成深度卷积称为pointwise卷积。
DenseNet169
为了解决梯度消失问题,该架构使用简单的连通模式,以确保在前向和后向计算中,各层之间的信息流动最大。这些层以这样一种方式连接,即来自前面所有层的输入通过它自己的特征图传递到后面所有层。
网络结构
网络架构可以用以下几点来解释:
我们将数据集分为两个部分——训练集和测试集,分别使用80%和20%的图像。
我们使用数据增强,如剪切、缩放、翻转和亮度变化,将数据集大小增加到原始数据集大小的两倍以上。
我们尝试使用预训练过的模型,比如Inception v3, InceptionResNet v2, ResNet 50, MobileNet和DenseNet169,然后微调网络的最后几层。
我们使用了50%的dropout和batch normalization层来减少过拟合。
最后我们使用了两个dense层,分别有64个神经元和2个神经元。
最后一层用于分类,softmax作为激活函数。
我们使用二元交叉熵作为损失函数。
除了最后一层之外,整个过程中都使用ReLU作为激活函数。
实验结果
loss vs epochs和accuracy vs epochs图如图2所示:
图2 Loss vs epoch b) Accuracy vs epoch
分类的混淆矩阵如图3:
图3:混淆矩阵
ROC-AUC
通过绘制ROC曲线,用于评价分类器。ROC曲线偏离对角线越远,分类器越好。分类器的ROC-AUC图如图4所示:
图4: ROC-AUC曲线
预训练权重对结果的比较如表5所示。
表6显示了与之前结果的比较。
总结
综上所述,本研究探讨了深度卷积神经网络在皮肤癌良恶性分类中的能力。我们的结果显示,在皮镜图像(3000个训练和600个验证组成的3600个图像)上训练的最先进的深度学习架构优于皮肤科医生。我们展示了使用非常深的卷积神经网络使用迁移学习并在皮镜图像上进行微调,可以达到比专家医生和临床医生更好的诊断准确性。为了帮助皮肤科医生,这些模型可以很容易地在皮肤镜系统或智能手机上实现。
—END—
英文原文:https://towardsdatascience.com/dermatologist-level-skin-cancer-classification-using-neural-network-475f93d7f8c3
请长按或扫描二维码关注本公众号
喜欢的话,请给我个好看吧!
使用神经网络进行皮肤癌的诊断相关推荐
- 基于卷积神经网络的皮肤癌检测系统
摘 要 医疗领域对皮肤癌早期诊断的需求越来越大,因此皮肤癌检测具有重要的研究意义和应用价值.提出的皮肤癌检测系统主要包括皮肤图像预处理.特征提取.模型构建和检测分类等四个功能模块.系统首先采用图像去噪 ...
- 卷积神经网络检测皮肤癌已超越皮肤科专家
内容来源:ATYUN AI平台 研究人员首次发现深度学习卷积神经网络(CNN)比经验丰富的皮肤科医生在检测皮肤癌方面表现更好. 在发表在主要癌症杂志Annals of Oncology [1]上的一项 ...
- 使用卷积神经网络的自动心电图诊断
使用卷积神经网络的自动心电图诊断 Automatic ECG Diagnosis Using Convolutional Neural Network https://www.mdpi.com/207 ...
- 多特征分支卷积神经网络在心梗心电图诊断中的应用
Multiple-feature-branch convolutional neural network for myocardial infarction diagnosis using elect ...
- 诊断皮肤癌,人工智能比医生更出色
一项由多国科学家联合完成的研究表明,基于卷积神经网络的人工智能在诊断皮肤癌方面已达到比人类医生更出色的水准. 研究人员首次证明一种被称之为深度学习卷积神经网络(Convolutional Neural ...
- 斯坦福Nature论文宣布新突破:深度学习皮肤癌诊断达专家水平
为了让人们能获得更好的医疗,斯坦福大学的研究者已经成功训练了一个可以诊断皮肤癌的算法.该研究的相关论文<Dermatologist-level classification of skin ca ...
- 深度学习识别皮肤癌媲美医生,智能手机检测癌症将不再遥远
研究人员希望在不久的将来将这项技术投入实用,让更多人获得更好的医疗服务.试想一下吧:一台皮肤癌检测仪,就在你智能手机里. [导读]斯坦福大学的研究人员开发深度学习算法,识别皮肤癌的准确率与专业的人类医 ...
- 大数据下机械智能诊断的机遇与挑战,阅读文献系列(一)
阅读文献系列一 论文题目:大数据下机械智能诊断的机遇与挑战 1.大数据时代下的故障诊断的挑战: 2.机械大数据的特性: 3.应该怎么做: 4.现有工作: 5.故障诊断三方面研究: 5.1 信号获取 5 ...
- bp 神经网络 优点 不足_【学术论文】基于灰度共生矩阵和BP神经网络的乳腺肿瘤识别...
摘要 乳腺肿瘤是女性病发率极高的一种肿瘤疾病, 但也是一种可以通过早期确诊.提早治疗来降低病死率的一种疾病.提出灰度共生矩阵结合BP神经网络的方法, 提高乳腺肿瘤的识别率.首先将红外乳腺图像进行预处理 ...
最新文章
- TensorRT PoolingLayer
- Python实现固定效应回归模型实现因果关系推断
- Extjs日期格式问题
- matlab bwlabel标记连通区域
- mysql 触发器 本表_MySQL触发器处理本表数据
- linux生产环境下安装anaconda总结
- jsp数据 foreach 循环遍历 (含效果图)- 代码篇
- leetcode53 python3 96ms 最大子序列和 python解法
- 电子邮件地址抓取工具
- 编译Nginx提示gzip module requires the zlib library
- 04-linux下安装neo4j
- 稳定币usda是哪个发行的_usdt稳定币是谁发行的?
- 什么是AsHelper
- 很多男孩不知道:她喜欢你
- SQL Server 使用SQL获取建表语句
- lmdb高效存储图片数据
- 微信商户、支付宝特殊能力开通
- 华为手机序列号前三位_华为Nova2s手机序列号前六位是TPG4C1是什么意思
- 未解——Matlab积分运算int函数
- 火地晉 (易經大意 韓長庚)