CNN手把手维修攻略:你的网络不好好训练 需要全面体检
消息来源:baojiabao.com 作者: 发布时间:2024-05-29
作者 Zeyad Emam
方栗子 郭一璞 编译
量子位 出品 | 公众号 QbitAI
您吃了么?
你的CNN还健在么?
Github用户Zeyad Emam贡献了一篇CNN故障排除攻略,详细介绍了CNN的常见故障和调教方法,量子位编译了中文版。
欢迎收藏并食用~
介绍
本文是一篇给卷积神经网络排查故障的攻略,主要来自于作者此前的经验和包括斯坦福CS231n课程笔记在内的线上资源。
本文主要针对使用深度神经网络进行的监督学习。虽然本攻略假设你用的是TensorFlow和Python3.6,不过本文内容编程语言无关,你可以当成是一篇通用的指南。
首先,假设我们现在有一个CNN,测试后发现它的表现比我们预想的差很多。于是,你就可以按照本攻略的步骤,一步一步来完成故障排除,之后你的神经网络可能就更合你心意了。
咦?这张图为什么点不开
第一部分是故障排除开始之前的必备事项,后面的则是针对具体问题的解决办法,问题主要集中在相对更为常见的方面,每个部分会优先给出针对该问题最容易实现的解决方法。
前菜:故障排除前
首先,给大家列一下完成深度学习算法时要遵循的最佳实践。
1.使用适当的日志和有意义的变数名。在TensorFlow中,你可以通过名称来跟踪不同的变数,并在TensorBoard中可视化图形。最重要的是,在每个训练步骤中,你都能记录相关的值,比如:step_number、accuracy、loss、learning_rate,甚至有时候还包括一些更具体的值,比如mean_intersection_over_union。之后,就可以画出每一步的损失曲线。
2.确保您的网络连接正确。使用TensorBoard或其他debug技术确保图中的每个操作的输入和输出都准确无误,还要确保在将数据和标签送入网络之前对其进行适当的预处理和配对。
3.实施数据增强技术。虽然这一点并不是对所有情况都适用,不过如果你在搞图像相关的神经网络,用简单的数据增强技术处理一下图像,例如镜像、旋转、随机裁剪和重新缩放、添加噪声、弹性变形等,大部分时候出来的效果都有巨大提升。
而且,TensorFlow内置了大多数基本的图像处理功能,十分良心了。
4.对所有层使用权重初始化和正则化。不要把权重初始化为相同的值,当然你要是把它们都初始化成0……那就更糟了,这可能会引入对称性,并且导致梯度消失,大多数时候都会导致糟糕的结果。
一般情况下,如果你在权重初始化时遇到问题,你可以考虑在神经网络中添加批量标准化层(Batch Normalization Layer)。关于批量标准化层,可以看这篇名为《Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift》的论文,链接:arxiv.org/abs/1502.03167
5.确保正则化条款不会压倒损失函数中的其他项。关闭正则化,找出损失的数量级,然后适当调整正则化权重。确保在增加正则化强度时,损失也在增加。
6.尝试过拟合一个小数据集。关闭正则化/丢失/数据增强,拿出训练集的一小部分,让神经网络练它几个世纪,确保可以实现零损失,不然就很可能是错误的。
在某些情况下,将损失驱动为零非常具有挑战性,例如,如果您的损失涉及每个像素的softmax-ed logits和ground truth labels之间的交叉熵,那么在语义分割中可能真的难以将其降低到0。相反,你应该争取达到接近100%的准确度。
7.在过拟合上述小数据集的同时,找到合理的学习率。Yoshua Bengio的论文中给到了结论:最佳学习率通常接近最大学习率的一半,不会引起训练标准的差异,这个观察结果是设置学习率的启发。例如,从较大的学习率开始,如果训练标准发散,就用最大学习率除以3再试试,直到观察不到发散为止。
8.执行梯度检查。如果您在图表中使用自定义操作,则梯度检查尤其重要。斯坦福CS231n中介绍了梯度检查的方法。
故障排除前的步骤主要来自于下面三篇资料,需要的朋友可复制链接查看:
·斯坦福CS231n中数据预处理部分
cs231n.github.io/neural-networks-2
·斯坦福CS231n中训练神经网络部分
cs231n.github.io/neural-networks-3
·Practical Recommendations for Gradient-Based Training of Deep Architectures
Yoshua Bengio
arxiv.org/pdf/1206.5533v2.pdf
现在,开始进食主菜:
损失没有改善怎么办?
如果,训练了好几个Epoch,损失还是没有变小,甚至还越来越大,就要:
1.确认你用的损失函数是合适的,你优化的张量也是对的。常用损失函数列表传送门:t.cn/RkZXji1。
2.用个好点的优化器。这里也有常见优化器的列表:t.cn/RDKwbNA。
3.确认变数真的在训练。要检查这个,就得看张量板的直方图。
或者写个脚本,在几个不同的训练实例 (training instances) 中,算出每个张量的范数。
如果变数没在训练,请看下节,“变数没在训练怎么办?”。
4.调整初始学习率,实施适当的学习率计划。
如果损失越来越大,可能是初始学习率太大;如果损失几乎不变,可能是初始学习率太小。
不管怎样,一旦找到好用的初始学习率,就要进行学习率衰减计划。
像AMA这样的优化器,内部就有学习率衰减机制。但它可能衰减得不够激烈,还是自己做一个比较好。
5.确认没有过拟合。做个学习率 vs 训练步数的曲线,如果像是抛物线,可能就过拟合了。解决方法参见下文“过拟合怎么办?”章节
变数没在训练怎么办?
像上文说的,看了张量板直方图,或者写了脚本来算每个张量的范数之后,把范数是常数的那些张量拎出来。
如果有个变数不训练:
1.确认TF把它看做可训练 (trainable) 的变数。详情可以查看TF GraphKeys:t.cn/R18Do6Y。
2.确认没发生梯度消失。
如果下游变数(更靠近output的变数) 训练正常,而上游变数不训练,大概就是梯度消失了。
解决方案见下文,“梯度消失/梯度爆炸”章节。
3.确认ReLU (线性整流函数) 还在放电。
如果大部分神经元电压都保持在零了,可能就要改变一下权重初始化策略了:尝试用一个不那么激烈的学习率衰减,并且减少权重衰减正则化。
梯度消失/梯度爆炸
1.考虑用个好点的权重初始化策略。尤其是在,训练之初梯度就不怎么更新的情况下,这一步尤为重要。
2.考虑换一下激活函数。比如ReLU,就可以拿Leaky ReLU或者MaxOut激活函数来代替。
3.如果是RNN (递归神经网络) 的话,就可以用LSTM block。详情可参照此文:t.cn/RI6Qe7t
过拟合怎么办?
过拟合,就是神经网络记住训练数据了。如果网络在训练集和验证集上,准确度差别很大,可能它就过拟合了。详情可见 (Train/Val accuracy) :t.cn/RAkUzJP。
1.做个数据扩增。可上翻至本文第一节。
2.做个Dropout。在训练的每一步,都抛弃一些神经元。详情请见:t.cn/RkZodZo。
3.增加正则化。
4.做个批量归一化。详情请见:t.cn/RNunyfR。
5.做个早停止 (early stopping)。因为,过拟合可能是训练了太多Epoch造成的。详情可见:t.cn/RkZKjEQ。
6.还不行的话,就用个小一点的网络吧。不过,没到万不得已,还是别这样。
还能调些什么?
1.考虑用个带权重的损失函数。
比如,在图像的语义分割中,神经网络要给每一个像素归类。其中一些类别,可能很少有像素属于它。
如果,给这些不常被光顾的类别,加个权重,mean_iou这项指标就会好一些。
2.改变网络架构。之前的网络,可能太深,可能太浅了。
3.考虑把几个模型集成起来用。
4.用跨步卷积 (strided convolution)来代替最大池化/平均池化。
5.做个完整的超参数搜索。
6.改变随机种子 (random seeds)。
7.上面的步骤全都不管用的话,还是再去多找点数据吧。
最后,祝大家的CNN都能吃喝不愁,健康成长。
https://gist.github.com/zeyademam/0f60821a0d36ea44eef496633b4430fc#before-troubleshooting
—完—
加入社群
量子位AI社群19群开始招募啦,欢迎对AI感兴趣的同学,在量子位公众号(QbitAI)对话界面回复关键字“交流群”,获取入群方式;
此外,量子位专业细分群(自动驾驶、CV、NLP、机器学习等)正在招募,面向正在从事相关领域的工程师及研究人员。
进专业群请在量子位公众号(QbitAI)对话界面回复关键字“专业群”,获取入群方式。(专业群审核较严,敬请谅解)
诚挚招聘
量子位正在招募编辑/记者,工作地点在北京中关村。期待有才气、有热情的同学加入我们!相关细节,请在量子位公众号(QbitAI)对话界面,回复“招聘”两个字。
相关文章
- 中兴受美国制裁事件 被罚了20亿美元过程事件始末 中兴被制裁后公司现状
2023-11-02 22:12:46
- B站怎么炸崩了哔哩哔哩服务器今日怎么又炸挂了?技术团队公开早先原因
2023-03-06 19:05:55
- 苹果iPhoneXS/XR手机电池容量续航最强?答案揭晓
2023-02-19 15:09:54
- 华为荣耀两款机型起内讧:荣耀Play官方价格同价同配该如何选?
2023-02-17 23:21:27
- google谷歌原生系统Pixel3 XL/4/5/6 pro手机价格:刘海屏设计顶配版曾卖6900元
2023-02-17 18:58:09
- 科大讯飞同传同声翻译软件造假 浮夸不能只罚酒三杯
2023-02-17 18:46:15
- 华为mate20pro系列手机首发上市日期价格,屏幕和电池参数配置对比
2023-02-17 18:42:49
- 小米MAX4手机上市日期首发价格 骁龙720打造大屏标准
2023-02-17 18:37:22
- 武汉弘芯遣散!结局是总投资1280亿项目烂尾 光刻机抵押换钱
2023-02-16 15:53:18
- 谷歌GoogleDrive网云盘下载改名“GoogleOne” 容量提升价格优惠
2023-02-16 13:34:45
- 巴斯夫将裁员6000人 众化工巨头裁员潮再度引发关注
2023-02-13 16:49:06
- 人手不足 韵达快递客服回应大量包裹派送异常没有收到
2023-02-07 15:25:20
- 资本微念与李子柒销声匿迹谁赢? 微念公司退出子柒文化股东
2023-02-02 09:24:38
- 三星GalaxyS8 S9 S10系统恢复出厂设置一直卡在正在检查更新怎么办
2023-01-24 10:10:02
- 华为Mate50 RS保时捷最新款顶级手机2022多少钱?1.2万元售价外观图片吊打iPhone14
2023-01-06 20:27:09
- 芯片常见的CPU芯片封装方式 QFP和QFN封装的区别?
2022-12-02 17:25:17
- 华为暂缓招聘停止社招了吗?官方回应来了
2022-11-19 11:53:50
- 热血江湖手游:长枪铁甲 刚猛热血 正派枪客全攻略技能介绍大全
2022-11-16 16:59:09
- 东京把玩了尼康微单相机Z7 尼康Z7现在卖多少钱?
2022-10-22 15:21:55
- 苹果iPhone手机灵动岛大热:安卓灵动岛App应用下载安装量超100万次
2022-10-03 22:13:45