AAAIDec, 2021

通过多任务学习改进证据深度学习

TL;DR提出了一种多任务学习框架 MT-ENet,通过将 Lipschitz 修正均方误差(MSE)损失函数添加到负对数边缘似然 (NLL) 损失中,动态调整 Lipschitz 常数,以解决 Evidential regression network (ENet) 的梯度收缩问题,并提高其预测精度,同时保持其良好的不确定性估计能力,该方法在药物 - 靶标亲和力 (DTA) 回归等实际基准测试中表现出了出色的校准和超出分布检测能力。