arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.07588eess.IVcs.CVcs.LGstat.ML
收藏

基于变分推断的生物医学图像分割不确定性量化

Uncertainty Quantification using Variational Inference for Biomedical Image Segmentation

University of Maryland(马里兰大学)

Abhinav Sagar

AI总结:

针对医学图像深度学习模型需提供预测置信度的需求,提出基于变分推断的编解码架构,在BRATS数据集上以贝叶斯方式融合两类不确定性实现脑肿瘤分割。

摘要:

由卷积神经网络推动的深度学习已在图像分类、图像分割、图像合成等一系列医学成像问题中取得巨大成功。然而,为了实现验证与可解释性,我们不仅需要模型给出的预测结果,还需要了解模型做出这些预测时的置信度,这一点对安全关键型应用获得用户认可至关重要。本研究采用基于变分推断技术的编码器-解码器架构进行脑肿瘤图像分割,在公开的BRATS数据集上开展评估,以戴斯相似系数(DSC)和交并比(IOU)为评价指标。该模型能够以有原则的贝叶斯方式同时考虑偶然不确定性和认知不确定性,完成脑肿瘤分割任务。

英文摘要:

Deep learning motivated by convolutional neural networks has been highly successful in a range of medical imaging problems like image classification, image segmentation, image synthesis etc. However for validation and interpretability, not only do we need the predictions made by the model but also how confident it is while making those predictions. This is important in safety critical applications for the people to accept it. In this work, we used an encoder decoder architecture based on variational inference techniques for segmenting brain tumour images. We evaluate our work on the publicly available BRATS dataset using Dice Similarity Coefficient (DSC) and Intersection Over Union (IOU) as the evaluation metrics. Our model is able to segment brain tumours while taking into account both aleatoric uncertainty and epistemic uncertainty in a principled bayesian manner.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑