时间戳

首页 > 解决方案 > 深度学习的损失分析

parameters - 深度学习的损失分析

问题描述

我是深度学习的新手，我已经建立了一个图卷积网络。我使用了 5 折交叉验证。在绘制了平均值之后train_loss (blue)，validate_loss (orange)我得到了这个孩子。

MSE损失

如您所见，从validate_loss的曲线趋势来看，我的网络似乎学到的东西很少。（我猜是数据？GCN 框架？学习率？）

你们能具体帮我找出错误吗？

我会很感激的！如果你不明白我的意思，请告诉我。

class Scorer(nn.Module):
    """
    Three conv_layers and two fc_layers with Dropout
    """
    def __init__(self):
        super(Scorer, self).__init__()
        self.conv_layer1 = GraphConvNet(5, 64)
        self.conv_layer2 = GraphConvNet(64, 128)
        self.conv_layer3 = GraphConvNet(128, 256) # (I have tried delete conv_layer3)
        self.fc_layer1 = nn.Linear(256, 128)
        self.drop_layer1 = nn.Dropout(0.5)
        self.fc_layer2 = nn.Linear(128, 64)
        self.drop_layer2 = nn.Dropout(0.5)
        self.out_layer = nn.Linear(64, 1)


    def forward(self, NormLap, feat):
        h = self.conv_layer1(NormLap, feat)
        h = F.leaky_relu(h)
        h = self.conv_layer2(NormLap, h)
        h = F.leaky_relu(h)
        h = self.conv_layer3(NormLap, h)
        h = F.leaky_relu(h)
        h = self.fc_layer1(h)
        h = self.drop_layer1(h)
        h = F.leaky_relu(h)
        h = self.fc_layer2(h)
        h = self.drop_layer2(h)
        h = F.leaky_relu(h)
        h = self.out_layer(h)
        h = F.leaky_relu(h)

以下是我的网络和参数：

    # parameter setting
    learning_rate = 0.001 # (I have tried 1e-1, 1e-2)
    weight_decay = 1e-3  # (I have tried 1e-4)
    epochs = 500
    batch_size = 50  # (I have tried 30)

    model = Scorer()
    loss_func = nn.MSELoss()
    optimizer = th.optim.Adam(model.parameters(), lr=learning_rate, weight_decay=weight_decay)

标签： parametersdeep-learningpytorchdata-loss

解决方案

这几乎就是训练和验证损失应该做的事情。损失随着时间的推移而下降；这就是优化器正在尝试做的事情。在 valid_loss 趋于平稳或平稳后 train_loss 继续下降，表明模型在 epoch ~100 左右后开始过度拟合。MSE 0.3 对您的应用程序是好是坏完全取决于应用程序，但是是的，优化器优化得很好。

请查看此资源以了解如何解释损失曲线：https ://machinelearningmastery.com/learning-curves-for-diagnosing-machine-learning-model-performance/

“从 validate_loss 的曲线趋势来看，我的网络似乎学到的东西很少” - 如果您解释（非常详细：）您为什么这么认为，以获得更好的答案，这将有所帮助。除了你看到的，你还期待看到什么？我看着同一张图，在我看来，您的网络正在学习对数据进行建模并预测您要预测的任何内容。

推荐阅读