为皮肤癌病变图像分割提供可量化不确定性的深度学习框架,提升临床可信度。
Uncertainty Quantified Deep Learning and Regression Analysis Framework for Image Segmentation of Skin Cancer Lesions
- 引入蒙特卡洛丢弃与贝叶斯反向传播,实现像素级不确定性估计。
- 首次提出单张图像多区域不确定性映射,与分割误差高度相关(皮尔逊相关,p<0.05)。
- 构建四个轻量回归模型,仅用少量计算即可预测分割质量(Dice得分)。
深度学习模型(DLMs)在医学图像肿瘤分割与分类中表现优异,但在处理未见图像时缺乏对分割机制的反馈,如骰子系数和置信度,导致临床应用中信任度不足。本文针对公开数据集ISIC-19的皮肤镜图像,训练两个DLMs(一个从零开始,一个基于迁移学习),并结合蒙特卡洛丢弃或贝叶斯反向传播方法,首次实现像素级不确定性估计。提出一种新方法,从单张皮肤镜图像中生成多个临床区域的图像级不确定性图,结果显示不完美分割区域与高不确定性区域显著对应。此外,首次报告了四个线性回归模型,可利用常数与不确定性指标(来自病灶、组织结构及非组织像素区域)联合预测分割性能(骰子系数),具有统计显著性(斯皮尔曼相关,p < 0.05),适用于低算力环境下的不确定性评估流程。
原文摘要 · Abstract (English)
Deep learning models (DLMs) frequently achieve accurate segmentation and classification of tumors from medical images. However, DLMs lacking feedback on their image segmentation mechanisms, such as Dice coefficients and confidence in their performance, face challenges when processing previously unseen images in real-world clinical settings. Uncertainty estimates to identify DLM predictions at the cellular or single-pixel level that require clinician review can enhance trust. However, their deployment requires significant computational resources. This study reports two DLMs, one trained from scratch and another based on transfer learning, with Monte Carlo dropout or Bayes-by-backprop uncertainty estimations to segment lesions from the publicly available The International Skin Imaging Collaboration-19 dermoscopy image database with cancerous lesions. A novel approach to compute pixel-by-pixel uncertainty estimations of DLM segmentation performance in multiple clinical regions from a single dermoscopy image with corresponding Dice scores is reported for the first time. Image-level uncertainty maps demonstrated correspondence between imperfect DLM segmentation and high uncertainty levels in specific skin tissue regions, with or without lesions. Four new linear regression models that can predict the Dice performance of DLM segmentation using constants and uncertainty measures, either individually or in combination from lesions, tissue structures, and non-tissue pixel regions critical for clinical diagnosis and prognostication in skin images (Spearman's correlation, p < 0.05), are reported for the first time for low-compute uncertainty estimation workflows.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。