【发布时间】:2019-10-07 14:16:21
【问题描述】:
我想用这样的 CT 灰度图像训练一个模型:
对于某些类别的疾病,我的训练集是有限的,例如 2,000 个阳性和 98,000 个阴性。
我正在考虑使用迁移学习来避免过度拟合并提高模型的有效性,但我也意识到我应该微调模型,因为我提供给模型的图像类型与使用的图像类型非常不同我将使用的 Inception 模型已经过训练。
然后我的问题是我不确定我应该保持多少层固定以及多少层可训练。
我当时正在考虑使用跳过连接来应用随机深度,让网络了解真正需要多少层。
所以我正在考虑实现以下架构。
即我将在 Keras (Tensorflow 2.0) 附带的预训练初始模型的层之间添加跳跃连接。
我欢迎就如何实施这些想法提出建议。特别是如何将网络分成三部分,保持第一部分不变(不可训练)并在添加跳过连接后训练第二部分。实现应该在 Keras 中。
【问题讨论】:
-
这里有一些有用的建议。对于考虑提供帮助的读者来说,这对你来说同样重要。 (1) 你可能问的问题太多了。两天内发布八篇帖子意味着您无需等待一件事的答案,然后再跳到下一件事。这将进一步向我表明您没有进行必要的研究。 (2) 您正在添加关于感谢帮助的闲聊材料,尽管许多编辑已从您之前的帖子中删除了此内容。技术写作在这里是一种期望。可根据要求提供元参考。
标签: python tensorflow keras conv-neural-network transfer-learning