如何在多任务CNN中调整减肥

Question

我开发了一个具有 6 个损失的多任务 CNN。我使用 BCEWithLogits Loss 或 MSE Loss 来定义这些损失。我通过最小-最大特征缩放将数据标准化为 0 到 1 之间的值。

某些损失的训练和验证损失不像其他损失那样收敛，并且误差很高。我尝试使用随机权重来提高性能，但得到了类似的结果。

 w = nn.functional.softmax(torch.randn(6), dim=-1)
 loss = w[0]*loss1 + w[1]*loss2 + w[2]*loss3   + w[3]*loss4 + w[4]*loss5 + w[5]*loss6

如何调整重量来改善效果？