使用不确定性进行多任务学习-Keras实现

Question

“使用不确定性权衡场景几何和语义损失的多任务学习”论文基本上总结了多任务损失函数可以采用以下形式：

因此，在上文中，L1是与输出1相关的损耗，L2是与输出2相关的损耗。alpha项是可以学习的！它们使用梯度下降进行了修改。 如何在Keras中实现呢？据我所知，自定义损失函数不能包含可学习的变量。