Oct, 2023

能否利用输入重建直接估计回归U-Net模型的不确定性?-- 应用于头颈癌患者的质子治疗剂量预测

TL;DR我们提出了一种新的直接不确定性估计方法,并将其应用于回归U-Net架构,该方法通过在瓶颈处添加重构输入的分支来估计模型的不确定性,并在头颈部癌症患者的质子疗法剂量预测中进行了验证,与常用方法MCDO和DE相比,我们的方法表现出更高的预测误差相关系数(0.620),同时更容易识别OOD (34.05的Z分数),且在回归任务中同时估计不确定性,因此需要更少的时间或计算资源。