迁移学习——小型数据库

数据挖掘 神经网络 美国有线电视新闻网 迁移学习 神经
2022-02-21 00:42:33

我正在尝试在医学(超声图像)中使用迁移学习。问题是 - 我的图片数据库非常有限 = 400 (360+40)。我正在使用 resnet50(我认为这并不重要,但也许我错了)。Resnet 作为特征提取器 + SVM 不是很好,但归一化的混淆矩阵有点:

1.0 0 0.4 0.6

现在,我想微调 resnet。问题是一开始的CM看起来像:

0.8 0.2 0.6 0.4

是这样的:

1.0 0 0.8 0.2

您可以在下面看到训练 + 测试损失/准确性。

现在我认为它过度拟合(由于速率容量/数据库太大)但有人指出网络可能没有学习。这是什么情况?

训练+测试损失和准确率

1个回答

看到你的图表并没有表明模型没有学习,因为训练的准确性很高。如果模型没有学到任何东西,即使这个值也会很低。此外,您说模型过度拟合是对的,因为它在训练集上表现非常好,而在测试集上表现不佳。

您拥有非常小的数据集这一事实也证实了过度拟合的情况。您可能想要微调 resnet 的所有层,如下所示另外,尝试提前停止和退出以防止模型过度拟合。