arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2305.07883eess.IVcs.CV

基于像素级不确定性估计的可泛化医学图像分割

Towards Generalizable Medical Image Segmentation with Pixel-wise Uncertainty Estimation

Tsinghua University(清华大学);The Hong Kong Polytechnic University(香港理工大学);Southern University of Science and Technology(南方科技大学)

Shuai Wang, Zipei Yan, Daoan Zhang, Zhongsen Li, Sirui Wu, Wenxuan Chen, Rui Li

AI总结:

提出利用不确定性估计突出难分类像素的新框架,提升医学图像分割的泛化能力,在前列腺和眼底数据集上优于现有最先进方法。

摘要:

深度神经网络(DNNs)在独立同分布(IID)假设下能在视觉识别中取得有前景的性能。然而,在众多实际应用中,尤其是在医学图像分析领域,IID假设并不普遍成立。医学图像分割通常被表述为像素级分类任务,即将每个像素归入某一类别。但该表述忽略了难以分类的像素,例如边界区域附近的某些像素,因为它们通常会使DNN产生混淆。本文首先探索了难以分类的像素与高不确定性相关联。基于此,我们提出一种利用不确定性估计来为DNN突出难以分类像素的新颖框架,从而提升其泛化能力。我们在两个流行基准上评估了该方法:前列腺和眼底数据集。实验结果表明,我们的方法优于现有最先进方法。

英文摘要:

Deep neural networks (DNNs) achieve promising performance in visual recognition under the independent and identically distributed (IID) hypothesis. In contrast, the IID hypothesis is not universally guaranteed in numerous real-world applications, especially in medical image analysis. Medical image segmentation is typically formulated as a pixel-wise classification task in which each pixel is classified into a category. However, this formulation ignores the hard-to-classified pixels, e.g., some pixels near the boundary area, as they usually confuse DNNs. In this paper, we first explore that hard-to-classified pixels are associated with high uncertainty. Based on this, we propose a novel framework that utilizes uncertainty estimation to highlight hard-to-classified pixels for DNNs, thereby improving its generalization. We evaluate our method on two popular benchmarks: prostate and fundus datasets. The results of the experiment demonstrate that our method outperforms state-of-the-art methods.

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

↑