如何提高模型的预测精度

数据挖掘 深度学习 图像分类 准确性
2022-03-09 07:49:41

我正在使用 InceptionV3 模型进行训练。这是代码的链接(https://github.com/maxmelnick/tensorflow/blob/no_random/tensorflow/examples/image_retraining/retrain.py)最初我有一个小型数据集。因此,我使用了增强技术来增加数据集的大小。

训练阶段数据集分为训练、验证和测试。在训练阶段,它显示了 11 个类的 96% 的准确率。但是当我预测任何新的输入图像(看不见的数据)时,它给出了 56% 的准确率。模型有什么问题?

我已经使用了 Dropout、Cross-validation、OverSampling 技术,但在新的输入图像上没有取得好的效果。

训练时使用的参数。

训练样本 - 每类 800 张图像

  • 训练样本 - 70%
  • 验证样本 - 20%
  • 测试样品 - 10%

测试样本(训练样本以外的未见数据) - 每个类别 51 张图像

时代 - 10,000

1个回答

听起来您过度拟合了训练数据,因此您应该研究一些神经网络正则化的方法。我看到您使用了 dropout,这是其中之一,但可能还不够。

您还可以考虑将训练数据分为训练集和验证集,并在验证集的性能停止改进时停止训练。这称为提前停止