【问题标题】:I am getting really high training loss, but the model has 100% accuracy of testing data我的训练损失非常高,但该模型的测试数据准确度为 100%
【发布时间】:2023-02-26 07:10:30
【问题描述】:

我正在使用来自 kaggle (https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia) 的数据集对 CNN 模型进行肺炎检测。我得到了 100% 的准确率,但我不确定是否有问题,因为我听说不太可能获得 100% 的准确率。

这是我的模型结构:

model = Sequential([
    Conv2D(input_shape=(150, 150, 1), filters=32, kernel_size=(3,3), strides=(1,1), padding="same", activation="relu"),
    BatchNormalization(),
    MaxPool2D(pool_size=(2,2),strides=(2,2)),
    
    Conv2D(filters=64, kernel_size=(3,3), strides=(1,1), padding="same", activation="relu"),
    BatchNormalization(),
    MaxPool2D(pool_size=(2,2),strides=(2,2)),
    
    
    Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding="same", activation="relu"),
    BatchNormalization(),
    MaxPool2D(pool_size=(2,2),strides=(2,2)),
    
    Flatten(),
    Dense(units=128,activation="relu"),
    Dropout(0.3),
    Dense(units=2, activation="softmax")
    ])

我使用 0.001 的学习率和分类交叉熵损失函数。我正在训练 12 个时期的模型,批量大小为 32。

这是我得到的用于训练和验证的模型性能: (https://i.stack.imgur.com/ryJK6.png)

如上图所示,训练损失值猛增,训练准确率稳定在 50% 左右。然而,当我使用 testig 数据评估我的模型时,我得到了 100% 的准确率,但损失值非常高。我尝试将学习率更改为 0.01 和 0.0001,但模型的准确率仍然为 100%。

然而,当我将损失函数更改为二元交叉熵并将最后一个密集层的激活函数更改为 sigmoid 时,我得到了 87.50% 的准确率。

我不确定为什么我的模型会以这种方式运行。

【问题讨论】:

    标签: deep-learning computer-vision conv-neural-network tensorflow2.0


    【解决方案1】:

    在二进制分类挑战中达到 100% 的准确率是不寻常的。因此,您的模型或训练方法可能存在一些问题。

    以下是一些需要考虑的事项:

    1. 分类交叉熵损失函数适用于多类 分类,而不是二元分类。对于二进制分类 像这样的问题,利用二元交叉熵损失并改变 最终激活函数为sigmoid而不是softmax

    2. 您的模型可能过度拟合,这可以解释为什么训练 几个时代后损失飙升。辍学和 L1 或 L2 正则化可以减少过度拟合。

    3. 为避免过度拟合,请添加训练数据。更多的数据通常会有所改善 模型泛化并减少过度拟合。

    4. 数据集可能包含类别不平衡。如果一个班级多得多 比另一个常见,模型可能会更频繁地预测它。使用 类别加权或过采样以纠正类别不平衡 训练和测试数据。

    5. 最后,检查精度、召回率和 F1 分数以及 准确性。对于不平衡的数据集,这些措施提供了更多 模型性能的细微差别。

    【讨论】:

      猜你喜欢
      • 2018-10-08
      • 2020-07-22
      • 2023-04-03
      • 1970-01-01
      • 2020-11-06
      • 1970-01-01
      • 2020-09-05
      • 1970-01-01
      • 2020-10-16
      相关资源
      最近更新 更多