AITOP

MC Dropout

共 2 条相关 AI 资讯
6月18日
10:56
10:56官方账号arXiv cs.LG@Xin Ci Wong, Duygu Sarikaya, Kieran Zucker, Marc De Kamps, Nishant Ravikumar
该论文评估了蒙特卡洛丢弃法（MC Dropout）在脑肿瘤分割中识别错误的能力。在126例BraTS21患者上测试了SegResNet和UNet-Res两个模型，MC Dropout保持了分割精度（|ΔDice|<0.01），且不确定性-误差对齐AUROC（熵H）约0.97。但全局对齐掩盖了区域差异：UNet-Res在增强肿瘤亚区域的熵仅为0.054，期望校准误差（ECE）达0.915，Dice仅0.714，显示严重误校准。标准Dice和AUROC无法检测这一失败模式。
论文MC Dropout不确定性估计脑肿瘤分割模型校准BraTS
推荐理由：这篇论文用具体数据告诉你，模型Dice高不一定安全，得看关键区域的校准情况。做医疗AI必读。
原文
5月25日
10:03
10:03官方账号arXiv cs.LG@Rouaa Hoblos, Noura Dridi, Noureddine Zerhouni, Zeina Al Masry
传统神经网络缺乏不确定性估计能力，而贝叶斯神经网络计算复杂。蒙特卡洛 Dropout 通过多次随机前向传播近似贝叶斯推理，但不确定性表示不够精细。本文提出将 Dirichlet 分布框架集成到 MC Dropout 中，利用 Sensoy 等人的方法建模类别概率，从而获得更丰富的不确定性信息。该方法在保持 MC Dropout 计算效率的同时，显著提升了不确定性估计的校准质量。实验表明，该方法能生成良好校准的不确定性估计，为不确定性感知的深度学习提供了实用方案。
论文不确定性估计MC DropoutDirichlet分布贝叶斯神经网络校准
推荐理由：做模型可靠性或安全关键应用的团队，终于有了一个既高效又精准的不确定性量化工具——在 MC Dropout 基础上加 Dirichlet 分布，校准效果更好，值得在分类任务中试试。
原文