CNN入门实战:我如何把准确率从86% 提高到99%(中)

本文介绍了在医疗图像识别任务中,使用预训练的卷积神经网络(CNN)进行微调,以提高准确率从86%到99%的过程。通过与医生交流获取额外信息,利用Dicom数据提取特征,以及研究超声波数据,作者展示了模型优化的不同阶段,包括特征提取、模型融合和规则基础方法的应用。尽管数据量有限,但模型的性能得到了显著提升。
摘要由CSDN通过智能技术生成

 

蒋竺波

蒋竺波

新加坡研究院 AI Department 公众号:follow_bobo

 

首发于专栏:https://zhuanlan.zhihu.com/c_141391545

个人公众号:follow_bobo

能不能点个赞,就是那种看起来写的还不错的样子


我们先来总结一下上一期的数据情况,顺便补充一点上一期没有讲到的:

目标类别一共20类,每类数量400-50不等

单从数据的情况,数据量少,并且不平衡,这个时候一个最基础的判断是用Pretrained model 实现两个Solution:

  1. Fine tune with Pretrained model

对于Google自己家的模型,比如Inception系列,我们用Tensorflow来fine tune,,毕竟自己家的模型,总会特别优化一点, 对于其他Pretrained model, 主要用Keras。

可能有人会问,Pretrained model 都是用ImageNet 训练的,但是ImageNet 里面并没有医疗数据呀?

首先,我们的数据量比较少,不建议Train from scratch.

第二,你可以理解为用Pretrained model 的参数来初始化model 的参数,而这些Pretrained model 的参数 的是经过专家们精确调参出来,效果肯定比随机生成参数好很多。最后也证明Fine tune with Pretrained model 在20个epochs 时就收敛了,并且有90%以上的Accuracy。

2. Extracting features with Pretrained model + Xg boost

这是在Kaggle 比赛中很常见的一种做法,特别是针对数据集比较少的时候。在这里最后效果不如 Fine tune,并且考虑到日后我们的数据越来越多,暂时不用这个方法。

如果有读者想有此法,我提醒一下。在提取特征之前,需要对model进行Domain Transfer, 什么意思,举个例子

  • 26
    点赞
  • 81
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值