如何处理keras的单输出多重损失？

Posted 2021-05-04

我的模型有一个输出。但我想将损失分成3个不同的组件，以便我可以跟踪每个组件的进度。有没有办法用keras做到这一点？

也许我可以定义与指标相同的损失组件，但是有更优雅的方式吗？

答案

您可以将损失定义为Keras图层，然后您可以手动添加所有损失和指标（如果需要）。

You can see a complete tutorial about this topic here

TL; DR：

• 定义计算损失的图层
• 编写自己的compile（）函数，添加优化器，损失和指标
• 在model.compile（optimizer =“adam”，loss = ...）中添加None作为损失

这是在代码中手动添加损失的方式：

loss_layer_names = {"my_loss", ...}

# Adding losses
for name in loss_layer_names:
    layer = model.get_layer(name)
    loss = (tf.reduce_mean(layer.output, keepdims=True))
    model.add_loss(loss)

# Adding metrics
for name in loss_layer_names:
    layer = model.get_layer(name)
    loss = (tf.reduce_mean(layer.output, keepdims=True))
    model.metrics_names.append(name)
    model.metrics_tensors.append(loss)

model.compile(optimizer="adam", loss=[None] * len(model.outputs))

model是Keras模型