Uncertainty-aware segmentation quality prediction via deep learning Bayesian Modeling: Comprehensive evaluation and interpretation on skin cancer and liver segmentation
Open accessAbstract
score of 85.03 and a Pearson correlation of 65.02, demonstrating cross-modality robustness. Additionally, we propose an aggregation strategy that combines multiple uncertainty estimates into a single score per image, offering a more robust and comprehensive assessment of segmentation quality compared to evaluating each measure independently. The proposed uncertainty-aware segmentation quality prediction network is interpreted using gradient-based methods such as Grad-CAM and feature embedding analysis through UMAP. These techniques provide insights into the model's behavior and reliability, helping to assess the impact of incorporating uncertainty into the segmentation quality prediction pipeline. The code is available at: https://github.com/sikha2552/Uncertainty-Aware-Segmentation-Quality-Prediction-Bayesian-Modeling-with-Comprehensive-Evaluation-.
AI snapshot
Written from the paper’s full text.
TL;DR
The paper proposes uncertainty-aware segmentation quality prediction without test-time ground truth, using Bayesian approximations (Monte Carlo Dropout, ensembles, test-time augmentation) and uncertainty maps to regress Dice scores. Across HAM10000 skin lesion and liver CT datasets, confidence maps and three-input CNNs improve prediction, with best HAM10000 R² 93.99/PCC 97.01 and liver R² 85.03/PCC 65.02.
Key findings
- For the two-way CNN using uncertainty estimates plus predicted segmentation maps, MCD + confidence achieved the best HAM10000 result: R² 93.25 and Pearson correlation 96.58; Ensemble + confidence reached R² 86.07/PCC 92.83, while TTA + confidence reached R² 71.7/PCC 85.74. On liver, Ensemble + confidence reached R² 72.94/PCC 85.88 and MCD + confidence R² 61.63/PCC 78.81.
- The three-way CNN incorporating original image, uncertainty map, and predicted segmentation map improved performance: HAM10000 MCD + entropy reached R² 93.99/PCC 97.01, TTA + entropy R² 89.13/PCC 94.49, and Ensemble + entropy R² 86.49/PCC 93.2. For liver, TTA + entropy reached R² 85.03/PCC 65.02.
- Confidence maps consistently improved two-way predictions across uncertainty models, while entropy maps showed better correlation in the three-way setup. The proposed aggregate uncertainty score combines confidence, entropy, mutual information, and expected pairwise KL divergence with weights 0.4, 0.2, 0.2, and 0.2, respectively, and was used to flag ambiguous or noisy labels.
- In RMSE comparison on skin cancer segmentation, Ensemble-Confidence achieved the lowest RMSE (0.132 ± 0.039), followed by TTA-Confidence (0.153 ± 0.049), LCE (0.167 ± 0.019), Max Probability (0.168 ± 0.014), HCNN (0.196 ± 0.023), No Uncertainty (0.198 ± 0.011), MCD-Confidence (0.209 ± 0.071), Quality Net (0.213 ± 0.009), and RCA (0.438 ± 0.007); the paper states uncertainty reduced RMSE by up to 0.066.
Methods
- The framework uses a fixed segmentation backbone, uncertainty models, uncertainty estimates, and a CNN regressor for segmentation quality. Backbones were FPN with ResNet50 for HAM10000 skin lesion segmentation and SwinUNet for liver segmentation. Uncertainty models were Monte Carlo Dropout (10 passes), ensembles (reported as 3 models in the Method and 5 models in Results), and test-time augmentation (10 random rotations). Uncertainty estimates were confidence map, entropy (Renyi with α = 2), mutual information, and expected pairwise KL divergence.
- Data: HAM10000 contains 2500 dermatoscopic images, with 50% used to train the segmentation model and 50% reserved for testing. The liver dataset contains 130 CT scans from Kaggle/LiTS17, with tumor labels merged with liver labels for binary segmentation.
- The quality prediction models were trained supervised on Dice scores using either two input branches (uncertainty estimates plus predicted segmentation maps, or uncertainty estimates plus original images) or three input branches (original image, uncertainty map, predicted segmentation map). Evaluation used R², Pearson correlation, and RMSE, with Grad-CAM and UMAP for interpretation.
Limitations
- The paper does not include a dedicated limitations section.
- Evident from the text: evaluation is limited to two datasets/modalities (2D skin lesion and 3D liver CT), and the quality prediction network is trained using ground-truth Dice scores from the training set, so ground truth is still required during training even though not at test time.
- Evident from the text: TTA performance was variable and sometimes lower, liver results showed high R² but lower Pearson correlation (65.02), and the paper contains reporting inconsistencies, including ensemble size (3 vs. 5 models) and reversed backbone-dataset assignments in the Conclusion.
Model-written summary. Check it against the paper before relying on it.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
6 of 14 cited sources checked · 1 supported · 0 flagged · accuracy not yet measured
- Premisecitation§ 1. Introduction
Traditional approaches to assessing segmentation quality heavily depend on the availability of ground truth data.
“Traditional approaches to assessing segmen- tation quality, such as boundary-based, region-based, and hy- brid metrics, heavily depend on the availability of ground truth data [7], [8], [9].”
Quote in the paperSubstantial claim1 of 3 sources checked- cites [7] DeVries, 2018
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- cites [8] DeVries, 2018
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- cites [9] Esteva, 2017, Dermatologist-level classification of skin cancer with deep neural networks, Nature source
The Abstract discusses lesion segmentation as preprocessing used by earlier classification methods, but does not discuss how segmentation quality is assessed. A search for boundary-based, region-based, and hybrid metrics and segmentation ground truth found no passage addressing the claimed dependence on ground-truth data. (LLM, conf=0.76)
Experimental — accuracy not yet measured
- Findingown data
Traditional methods like RCA exhibit higher RMSE than uncertainty-aware approaches on the skin cancer segmentation dataset.
“In contrast, traditional methods like RCA [3] exhibit higher RMSE, underscoring the advantages of uncertainty-aware approaches.”
Quote in the paperSubstantial claim1 of 1 sources checked- cites [3] Bilic, 2019 source
The Results (Meta analysis and Results of individual challenges; page not provided) concern liver tumor segmentation and report Dice and ASD scores, not RMSE comparisons on a skin cancer dataset. Searches for RCA, RMSE, skin lesions, and uncertainty-aware methods found no passage addressing the attributed comparison. (LLM, conf=0.76)
Experimental — accuracy not yet measured
- Findingown data§ Abstract
The framework achieves a Pearson correlation of 96.58 on the HAM10000 dataset.
“Pearson correlation of 96.58 on the HAM10000 dataset”
Quote in the paperSubstantial claim - Findingown data§ 4. Expected pairwise Kullback-Leibler (KL) diver-
In the authors' experiments, confidence maps consistently showed strong correlation with the prediction task across different models and datasets.
“In our experiments, confidence maps consistently showed strong correlation with the prediction task across different models and datasets.”
Quote in the paperSubstantial claim - Findingown data§ Abstract
For 3D liver segmentation, Test Time Augmentation with entropy achieves a Pearson correlation of 65.02.
“a Pearson correla- tion of 65.02”
Quote in the paperSubstantial claim
- Findingown data§ Abstract
For 3D liver segmentation, Test Time Augmentation with entropy achieves an R2 score of 85.03.
“For 3D liver segmentation, Test Time Augmentation with entropy achieves an R² score of 85.03”
Quote in the paperNot a substantial claim - Findingown data
For samples with the highest Dice score predictions, the Grad-CAMs from Sub-branch one focus on the predicted foreground, which corresponds well with the lesion area.
“Corresponding Grad-CAMs from Sub-branch one (3rd column) focuses on the predicted foreground, which corresponds well with the lesion area.”
Quote in the paperNot a substantial claim - Findingown data
For the two-way CNN using UE and original images as inputs, TTA + Entropy achieved the highest R2 score (71.12) and Pearson correlation (84.71) on the HAM10000 dataset.
“The results, evaluated on the HAM10000 and Liver datasets, show that TTA + Entropy achieved the highest R² score (71.12) and Pearson correlation (84.71) on HAM10000”
Quote in the paperSubstantial claim - Findingown data
For samples with the lowest Dice score predictions, the predicted segmentation maps clearly fail to accurately capture the lesion regions.
“For samples with the lowest Dice scores, column.1 shows the predicted segmentation maps which clearly fail to accurately capture the lesion regions.”
Quote in the paperNot a substantial claim - Findingown data
On the HAM10000 dataset, Monte Carlo Dropout (MCD) with confidence map achieved the highest performance with an R2 score of 93.25 and a PCC of 96.58.
“while Monte Carlo Dropout (MCD) with confidence map achieved the highest performance with an R² score of 93.25 and a PCC of 96.58.”
Quote in the paperSubstantial claim - Findingown data
For the two-way CNN using UE and original images as inputs, Ensemble + Confidence delivered the best R2 score (63.72) on the Liver dataset.
“while Ensemble + Confidence delivered the best R² score (63.72) on the Liver dataset.”
Quote in the paperNot a substantial claim - Findingown data
On the Liver dataset, ensemble models with confidence map produced an R2 score of 72.94 and a PCC of 85.88.
“On the Liver dataset, ensemble models with confidence map produced an R² score of 72.94 and a PCC of 85.88”
Quote in the paperSubstantial claim - Findingown data
On the HAM10000 dataset, the ensemble model combined with confidence maps achieved an R2 score of 86.07 and a PCC of 92.83.
“For the HAM10000 dataset, ensemble models combined with confidence maps yielded an R² score of 86.07 and a PCC of 92.83”
Quote in the paperNot a substantial claim - Findingcitation
Among uncertainty estimation techniques, HCNN provides the least performance gain.
“Among uncertainty estimation techniques, HCNN provides the least performance gain”
Quote in the paperNot a substantial claim - Interpretationreasoning§ 1. Introduction
In the absence of ground truth for individual samples, the applicability of traditional segmentation quality metrics is limited.
“In the absence of ground truth for individual samples, the applicability of these metrics is limited.”
Quote in the paperSubstantial claim - Premisecitation§ 1. Introduction
Prior segmentation quality prediction approaches do not incorporate uncertainty measures.
“However, these ap- proaches do not incorporate uncertainty measures, which are crucial for understanding model confidence and improving seg- mentation reliability [13].”
Quote in the paperSubstantial claim0 of 1 sources checked- cites [13] Huttenlocher, 1993, Comparing images using the Hausdorff distance, IEEE Trans. Pattern Anal. Mach. Intell. source
- Interpretationnone given§ 1. Introduction
The proposed framework functions in the absence of ground truth during test time.
“this paper proposes an uncertainty-aware segmentation quality prediction framework that functions in the absence of ground truth during test time.”
Quote in the paperNot a substantial claim- Rests on an assumption: Ground truth annotations are often impractical to obtain for large medical image datasets.
- Interpretationown data§ Abstract
The framework demonstrates cross-modality robustness.
“demonstrating cross-modality robustness.”
Quote in the paperSubstantial claim- Rests on an assumption: The HAM10000 and liver datasets are sufficiently representative of different modalities to demonstrate cross-modality robustness.
- Interpretationnone given§ Abstract
The proposed framework predicts segmentation quality without requiring ground truth annotations during test time.
“we pro- pose a novel framework for predicting segmentation quality without requiring ground truth annotations during test time.”
Quote in the paperSubstantial claim- Rests on an assumption: Ground truth annotations are often impractical to obtain for large medical image datasets.
- Interpretationown data§ 4. Expected pairwise Kullback-Leibler (KL) diver-
The aggregate score effectively integrates different aspects of uncertainty, offering a more robust evaluation of segmentation performance compared to considering each measure in isolation.
“This aggregate score effectively integrates different aspects of uncertainty, offering a more robust evaluation of the segmentation performance compared to considering each measure in isolation.”
Quote in the paperSubstantial claim - Interpretationown data§ 7. Conclusion
Confidence maps consistently improved segmentation quality predictions across various uncertainty models by clearly identifying reliable predictions.
“Confidence maps consistently improved segmentation quality predictions across various UMs by clearly identifying reliable predictions.”
- Also stated in: § 7. Conclusion · boosted
“Comprehensive evaluation showed that Confidence maps consistently improved segmentation quality predictions across various UMs by clearly identifying reliable predictions.”
Quote in the paperSubstantial claim - Also stated in: § 7. Conclusion · boosted
- Interpretationreasoning
Among uncertainty models, ensemble models consistently performed well across both datasets, particularly when combined with confidence maps.
“Among UMs, ensemble models consistently performed well across both datasets, particularly when combined with confidence maps.”
Quote in the paperSubstantial claim - Interpretationown data§ 7. Conclusion
The aggregated uncertainty score has potential value in flagging images that require revisiting or revision by annotators, thereby improving the overall quality and reliability of the dataset used for both training and evaluation.
“This suggests its potential value in flagging images that require revisiting or revision by annotators, thereby improving the overall quality and reliability of the dataset used for both training and evaluation.”
Quote in the paperSubstantial claim
- Classified as probably not a substantial claim. 13
- The quote only partly states this claim. 2
- May combine two claims. 12
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 1. IntroductionRobinson et al. proposed a CNN-based regression model to predict Dice Similarity Coefficients using original images and segmentation maps.
“Robinson et al. [10] proposed a CNN-based regression model to predict Dice Similarity Co- efficients (DSC) using original images and segmentation maps”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [10] Gal, 2016, Dropout as a bayesian approximation: Representing model uncertainty in deep learning source
Section 5.2 (p. 6) describes a CNN used to assess classification confidence on MNIST, while Section 5.1 (p. 5) describes regression experiments on CO₂ data. Searches for Robinson, Dice Similarity Coefficients, and segmentation maps found no passage addressing the attributed model. (LLM, conf=0.76)
Experimental — accuracy not yet measured
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 2. Ensemble models: Ensemble models involve com-Ensemble models combine predictions from multiple individual models to improve overall performance.
“Ensemble models involve com- bining predictions from multiple individual models to improve overall performance[32].”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [32] Tang, Y., Yang, D., Li, W., Roth, H.R., Landman, B., Xu, D., Nath, V., Hatamizadeh, A., 2022. Self-supervised pre-training of swin transformers for 3d medical image analysis. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 20730–20740. source
Classified as probably not a substantial claim.
Premisecitation§ 1. Monte Carlo Dropout (MCD): Monte CarloMonte Carlo Dropout estimates model uncertainty in neural networks by enabling dropout during test time.
“Monte Carlo Dropout (MCD) estimate model uncertainty in neu- ral networks [21] by enabling dropout during test time.”
Quote in the paperSubstantial claim1 of 1 sources checked- cites [21] Liu, 2019, Accurate uncertainty estimation and decomposition in ensemble learning, Adv. Neural Inf. Process. Syst. source
The Introduction (pp. 1–2) discusses epistemic uncertainty, and the Experiments section (p. 8) describes a deep-ensemble comparator, but neither explains Monte Carlo Dropout. Searches for “dropout” and “test time” found no passage saying that enabling dropout at test time estimates model uncertainty. (LLM, conf=0.75)
Experimental — accuracy not yet measured
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 5. Datasets and Segmentation BackbonesThe HAM10000 dataset contains 2500 dermatoscopic images of pigmented skin lesions.
“The HAM10000 dataset [34] contains 2500 dermatoscopic images of pigmented skin lesions.”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [34] Valindria, 2017, Reverse classification accuracy: predicting segmentation performance in the absence of ground truth, IEEE Trans. Med. Imaging source
The Introduction (p. 1) discusses medical-image segmentation, and the Discussion (p. 8) describes a liver-image example. Searches for HAM10000, 2500 dermatoscopic images, and skin-lesion datasets found no passage in our copy addressing the claimed dataset or image count. (LLM, conf=0.76)
Experimental — accuracy not yet measured
The quote only partly states this claim.
Findingown data§ AbstractThe framework achieves an R2 score of 93.25 on the HAM10000 dataset.
“Our framework achieves an R² score of 93.25”
Quote in the paperSubstantial claimMay combine two claims.
Findingown dataThe proposed aggregated uncertainty score effectively identifies and quantifies discrepancies between the ground truth and model predictions.
“The proposed aggregated uncertainty score effectively identifies and quantifies these discrepancies between the ground truth and model predictions.”
- Also stated in: § 7. Conclusion · boosted
“the results demonstrated that the aggregated uncertainty score accurately identifies and quantifies discrepancies between the ground truth and model predictions.”
Quote in the paperSubstantial claim- Also stated in: § 7. Conclusion · boosted
May combine two claims.
Findingown dataOn the Liver dataset, MCD with confidence map had an R2 score of 61.63 and a Pearson correlation of 78.81.
“while MCD with confidence map had an R² score of 61.63 and a Pearson correlation of 78.81.”
Quote in the paperNot a substantial claimMay combine two claims.
Findingown dataFor samples with the lowest Dice score predictions, the Grad-CAMs from the second sub-branch concentrate on very small regions inside the lesions, reflecting high uncertainty across most pixels.
“The 4th column, representing the Grad-CAM from the second sub-branch, concentrates on very small regions inside the lesions, reflecting high uncertainty across most pixels as seen in the uncertainty map.”
Quote in the paperSubstantial claimMay combine two claims.
Findingown dataThe top three images with the highest aggregated uncertainty scores have ambiguous or noisy labels, where a significant portion of pixels are labeled as lesions, and the model predictions are notably inaccurate.
“It is evident from the images that all three have ambiguous or noisy labels, where a significant portion of pixels are labeled as lesions, and the model predictions are notably inaccurate.”
Quote in the paperSubstantial claimMay combine two claims. The quote only partly states this claim.
Findingown dataThe 3-way CNN UMAP embedding shows the highest degree of feature separation among the three configurations, followed by the 2-way CNN with segmentation map and uncertainty map, then the 2-way CNN with original image and uncertainty map.
“In Figure.10 column (a), the 2-way CNN embedding with the original image and uncertainty map shows a moderate level of feature separation”
- Also stated in: § 6.3. Visualization and Interpretation of Grad-CAMs and · boosted
“Column (c) demonstrates the UMAP embedding for the 3-way CNN that includes the original image, segmentation map, and uncertainty map. It shows the highest degree of feature separation among the three configurations.”
Quote in the paperNot a substantial claim- Also stated in: § 6.3. Visualization and Interpretation of Grad-CAMs and · boosted
Probably not a substantial claim: may not pass S4 (based).
Premisereasoning§ 3. MethodFull Bayesian inference in deep learning models is computationally infeasible due to the complexity of calculating the posterior distribution.
“However, in practice, performing full Bayesian inference in deep learning models is compu- tationally infeasible due to the complexity of calculating the posterior distribution.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 3. Test Time Augmentation (TTA): TTA involves ap-Test Time Augmentation involves applying data augmentation transformations to the input image during test time and generating predictions for each augmented version.
“TTA involves ap- plying data augmentation transformations, such as rotations, flips, or color adjustments, to the input im- age during test time and generating predictions for each augmented version [24].”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [24] Movahedi, 2010, Design and perceptual validation of performance measures for salient object segmentation source
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 1. IntroductionZhou et al. combined deep reconstruction networks with regression models for DSC prediction.
“Zhou et al. [12] combined deep reconstruction networks with regression models for DSC prediction.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [12] Huang, 2016, Qualitynet: Segmentation quality evaluation with deep convolutional networks source
Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Premisecitation§ 3. MethodFPN with ResNet50 has been widely used for skin cancer segmentation, with Esteva et al. and Codella et al. highlighting its effectiveness in capturing fine-grained details in dermoscopic images.
“Similarly, FPN with ResNet50 has been widely used for skin cancer segmentation, with Esteva et al. [30] and Codella et al. [31] highlighting its effective- ness in capturing fine-grained details in dermoscopic im- ages.”
Quote in the paperNot a substantial claim0 of 2 sources checkedProbably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Premisecitation§ 1. IntroductionDeVries et al. introduced a DCNN-based model that uses uncertainty maps alongside original images and segmentation maps to predict DSC scores and flag cases for expert review.
“De- Vries et al.[14] introduced a DCNN-based model that uses uncertainty maps alongside original images and segmentation maps to predict DSC scores and flag cases for expert review.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [14] Kendall, 2017, What uncertainties do we need in bayesian deep learning for computer vision?, Adv. Neural Inf. Process. Syst. source
Probably not a substantial claim: may not pass S4 (based).
Premisenone given§ 1. IntroductionNo studies have systematically evaluated and benchmarked multiple uncertainty models alongside various uncertainty estimates in a multimodal setting.
“No studies have systematically eval- uated and benchmarked multiple uncertainty models (such as Monte Carlo Dropout, Ensembles, and Test Time Augmenta- tion) alongside various uncertainty estimates (including confi- dence maps, entropy, mutual information, and expected pair- wise Kullback-Leibler divergence) in a multimodal setting.”
Quote in the paperNot a substantial claimMay combine two claims.
Interpretationnone given§ AbstractThe aggregation strategy combines multiple uncertainty estimates into a single score per image, offering a more robust and comprehensive assessment of segmentation quality compared to evaluating each measure independently.
“Additionally, we propose an aggregation strategy that combines mul- tiple uncertainty estimates into a single score per image, offering a more robust and comprehensive assessment of segmenta- tion quality compared to evaluating each measure independently.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Interpretationreasoning§ 5. Datasets and Segmentation BackbonesSwinUnet achieves superior segmentation performance on complex volumetric medical datasets, effectively balancing broad contextual understanding with fine-grained local delineation.
“This synergy between global context captured by the Swin Transformer and the precise spatial details from UNet en- ables SwinUnet to achieve superior segmentation performance on complex volumetric medical datasets, effectively balancing broad contextual understanding with fine-grained local delin- eation.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
InterpretationreasoningThe reduced effectiveness of TTA may be due to noise or irrelevant variations introduced by augmentations, suggesting that TTA's effectiveness is highly dependent on the dataset and augmentation type.
“The reduced effectiveness may be due to noise or irrelevant variations introduced by augmentations, suggesting that TTA's effectiveness is highly dependent on the dataset and augmentation type.”
Quote in the paperNot a substantial claimMay combine two claims.
InterpretationreasoningThe improved clustering in the 3-way CNN signifies that it is better equipped to push the Dice scores further apart, thereby enhancing its predictive capabilities and robustness in segmentation tasks.
“The improved clustering in (c) signifies that the 3-way CNN is better equipped to push the Dice scores further apart, thereby enhancing its predictive capabilities and robustness in segmentation tasks.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisereasoning§ 5. Datasets and Segmentation BackbonesThe liver dataset is part of the Liver Tumor Segmentation Challenge (LiTS17).
“This dataset is part of the Liver Tu- mor Segmentation Challenge (LiTS17), organized in conjunc- tion with ISBI 2017 and MICCAI 2017.”
Quote in the paperNot a substantial claimMay combine two claims.
Interpretationown dataEnsemble models and MCD, which address epistemic uncertainty, significantly benefit from confidence maps, whereas TTA, capturing aleatoric uncertainty, showed variable performance due to the potential introduction of noise or artifacts.
“Ensemble models and MCD, which address epistemic uncertainty, significantly benefit from confidence maps, whereas TTA, capturing aleatoric uncertainty, showed variable performance due to the potential introduction of noise or artifacts.”
Quote in the paperSubstantial claim
The paper relies on this without support.
Ground truth annotations are often impractical to obtain for large medical image datasets.
“particularly in medical image analysis, where obtaining ground truth for large datasets is often impractical.”
The authors state this as an assumption.
Test Time Augmentation can be viewed as a Bayesian mixture model for uncertainty estimation.
“While not explicitly Bayesian, TTA can be viewed as a Bayesian mixture model for uncertainty es- timation.”
The paper relies on this without support.
The HAM10000 and liver datasets are sufficiently representative of different modalities to demonstrate cross-modality robustness.
“For 3D liver segmentation, Test Time Augmentation with entropy achieves an R² score of 85.03 and a Pearson correla- tion of 65.02, demonstrating cross-modality robustness.”
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
Each contradiction between two passages listed here was confirmed by an independent check.
Ensemble size changes from three to five models
MajorThe methods specify an ensemble of three models, but the results describe the evaluated ensemble as having five models.
“In our experiment, we used an ensemble of 3 models.”
“The UMs included: ensemble (5 models), Monte”
Conclusion reverses the segmentation backbones assigned to the datasets
MajorThe methods section assigns FPN to HAM10000 and Swin UNET to the liver dataset, while the conclusion assigns them the other way around.
“SwinUNet for the skin lesion dataset and FPN with ResNet50 for the liver dataset,”
“the Feature Pyramid Network (FPN) for the HAM10000 dataset and the Swin UNET for the liver dataset.”
The stated highest HAM10000 three-input R2 is below another reported result
MajorThe text calls Ensemble + Entropy's R2 of 86.49 the highest for HAM10000, then reports 93.99 for MCD + Entropy in the same three-input comparison.
“achieved the highest R² score of 86.49”
“tropy followed closely with an R² score of 93.99”
The segmentation model is described as requiring a label at input
MinorThe stated segmentation function takes both image x and label y as inputs, although the framework is said to operate without ground truth at test time. As written, the function would require the unavailable label to generate a test-time segmentation.
“The segmentation model (S) takes the pair (x, y) as inputs and generates (Ŷ) as the output”
“functions in the absence of ground truth during test time.”
The stated best liver R2 in Table 2 is not the largest
MinorThe narrative identifies Ensemble + Confidence at 63.72 as the best liver R2 for this input setup, whereas Table 2 reports 63.75 for TTA + Entropy.
“Ensemble + Confidence delivered the best R² score (63.72) on the Liver dataset.”
“TTA + Entropy 71.12 84.71 63.75 56.50”
Entropy is not the best-correlating three-input estimate for every model
MinorThe text says entropy maps showed better correlation with all uncertainty models, but Table 3 reports a higher liver Pearson correlation for Ensemble + MI (85.31) than for Ensemble + Entropy (84.76).
“for a 3-way CNN entropy maps showed better correlation in combination with all UMs.”
“Ensemble +MI 84.2 91.93 71.06 85.31”
TTA is not uniformly worse on the liver two-input task
MinorThe narrative says TTA performed comparatively worse with all uncertainty estimates on the liver dataset. In Table 1, TTA + Confidence has a higher liver R2 (71.6) than MCD + Confidence (61.63), with a higher PCC as well.
“TTA combined with all uncertainty es- timations (UEs) showed comparatively lower performance on the Liver dataset.”
“TTA + Confidence 71.7 85.74 71.6 85.4”
Cites
Feature Pyramid Networks for Object Detection
Lin, T.-Y., Dollar, P., Girshick, R., He, K., Hariharan, B., & Belongie, S. (2017) · 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2017.106 · Cited by 30,939 · cited 3× · see citing works
“We used SwinUNet [27] and Feature Pyramid Network (FPN) with Resnet50 [28] as the segmentation backbones.”
Dermatologist-level classification of skin cancer with deep neural networks
Esteva, A., Kuprel, B., Novoa, R. A., Ko, J., Swetter, S. M., Blau, H. M., & Thrun, S. (2017) · Nature · doi:10.1038/nature21056 · Cited by 14,362 · cited 1× · see citing works
“Similarly, FPN with ResNet50 has been widely used for skin cancer segmentation, with Esteva et al. [30] and Codella et al. [31] highlighting its e ff ective-ness in capturing fine-grained details in dermoscopic images.”
Dropout as a bayesian approximation: Representing model uncertainty in deep learning
Gal, Y., & Ghahramani, Z. (2015) · arXiv · doi:10.48550/arxiv.1506.02142 · Cited by 6,471 · see citing works
Swin-Unet: Unet-like pure transformer for medical image segmentation
Cao, H., Wang, Y., Chen, J., Jiang, D., Zhang, X., Tian, Q., & Wang, M. (2023) · Lecture Notes in Computer Science · doi:10.1007/978-3-031-25066-8_9 · Cited by 5,640 · see citing works
Comparing images using the Hausdorff distance
Huttenlocher, D. P., Klanderman, G. A., & Rucklidge, W. J. (1993) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/34.232073 · Cited by 4,545 · see citing works
Bayesian Learning for Neural Networks
Neal, R. M. (1996) · Lecture Notes in Statistics · doi:10.1007/978-1-4612-0745-0 · Cited by 4,276 · see citing works
Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles
Lakshminarayanan, B., Pritzel, A., & Blundell, C. (2016) · arXiv · doi:10.48550/arxiv.1612.01474 · Cited by 3,666 · see citing works
Tschandl, P., Rosendahl, C., & Kittler, H. (2018) · Scientific Data · doi:10.1038/sdata.2018.161 · Cited by 3,628 · see citing works
What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
Kendall, A., & Gal, Y. (2017) · arXiv · doi:10.48550/arxiv.1703.04977 · Cited by 3,088 · see citing works
A review of uncertainty quantification in deep learning: Techniques, applications and challenges
Abdar, M., Pourpanah, F., Hussain, S., Rezazadegan, D., Liu, L., Ghavamzadeh, M., Fieguth, P., Cao, X., Khosravi, A., Acharya, U. R., et al. (2021) · Information Fusion · doi:10.1016/j.inffus.2021.05.008 · Cited by 2,720 · cited 2× · see citing works
“However, these approaches do not incorporate uncertainty measures, which are crucial for understanding model confidence and improving segmentation reliability [13].”
2013 IEEE 10th International Symposium on Biomedical Imaging
(2013) · doi:10.1109/isbi20048.2013 · Cited by 1,637 · see citing works
The Liver Tumor Segmentation Benchmark (LiTS)
Bilic, P., Christ, P., Li, H. B., Vorontsov, E., Ben-Cohen, A., Kaissis, G., Szeskin, A., Jacobs, C., Mamani, G. E. H., Chartrand, G., et al. (2023) · Medical Image Analysis · doi:10.1016/j.media.2022.102680 · Cited by 1,412 · see citing works
The Liver Tumor Segmentation Benchmark (LiTS)
Bilic, P., Christ, P., Li, H. B., Vorontsov, E., Ben-Cohen, A., Kaissis, G., Szeskin, A., Jacobs, C., Mamani, G. E. H., Chartrand, G., et al. (2019) · arXiv · doi:10.48550/arxiv.1901.04056 · Cited by 1,412 · see citing works
Codella, N., Rotemberg, V., Tschandl, P., Celebi, M. E., Dusza, S., Gutman, D., Helba, B., Kalloo, A., Liopyris, K., Marchetti, M., et al. (2019) · arXiv · doi:10.48550/arxiv.1902.03368 · Cited by 984 · see citing works
Self-Supervised Pre-Training of Swin Transformers for 3D Medical Image Analysis
Tang, Y., Yang, D., Li, W., Roth, H. R., Landman, B., Xu, D., Nath, V., & Hatamizadeh, A. (2022) · 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr52688.2022.02007 · Cited by 849 · see citing works
Nock, R., & Nielsen, F. (2004) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/tpami.2004.110 · Cited by 822 · see citing works
Wang, G., Li, W., Aertsen, M., Deprest, J., Ourselin, S., & Vercauteren, T. (2019) · Neurocomputing · doi:10.1016/j.neucom.2019.01.103 · Cited by 659 · see citing works
Feature Pyramid Networks for Object DetectionPreprint
Lin, T.-Y., Dollár, P., Girshick, R., He, K., Hariharan, B., & Belongie, S. (2016) · arXiv (Cornell University) · doi:10.48550/arxiv.1612.03144 · Cited by 645 · see citing works
Design and perceptual validation of performance measures for salient object segmentation
Movahedi, V., & Elder, J. H. (2010) · 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition - Workshops · doi:10.1109/cvprw.2010.5543739 · Cited by 469 · see citing works
Learning Confidence for Out-of-Distribution Detection in Neural NetworksPreprint
DeVries, T., & Taylor, G. W. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1802.04865 · Cited by 354 · see citing works
Multiplicative Normalizing Flows for Variational Bayesian Neural NetworksPreprint
Louizos, C., & Welling, M. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1703.01961 · Cited by 254 · see citing works
Abdominal multi-organ segmentation with organ-attention networks and statistical fusion
Wang, Y., Zhou, Y., Shen, W., Park, S., Fishman, E. K., & Yuille, A. (2019) · Medical Image Analysis · doi:10.1016/j.media.2019.04.005 · Cited by 247 · see citing works
Automatic Fuzzy Clustering Framework for Image Segmentation
Lei, T., Liu, P., Jia, X., et al. (2019) · IEEE Transactions on Fuzzy Systems · doi:10.1109/tfuzz.2019.2930030 · Cited by 175 · see citing works
Reverse Classification Accuracy: Predicting Segmentation Performance in the Absence of Ground Truth
Valindria, V. V., Lavdas, I., Bai, W., Kamnitsas, K., Aboagye, E. O., Rockall, A. G., Rueckert, D., & Glocker, B. (2017) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2017.2665165 · Cited by 152 · see citing works
Roy, A. G., Conjeti, S., Navab, N., & Wachinger, C. (2019) · NeuroImage · doi:10.1016/j.neuroimage.2019.03.042 · Cited by 137 · see citing works
Leveraging Uncertainty Estimates for Predicting Segmentation QualityPreprint
DeVries, T., & Taylor, G. W. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1807.00502 · Cited by 117 · see citing works
New benchmark for image segmentation evaluation
Ge, F., Wang, S., & Liu, T. (2007) · Journal of Electronic Imaging · doi:10.1117/1.2762250 · Cited by 108 · see citing works
Evaluating Segmentation Error without Ground Truth
Kohlberger, T., Singh, V., Alvino, C., Bahlmann, C., & Grady, L. (2012) · Lecture Notes in Computer Science · doi:10.1007/978-3-642-33415-3_65 · Cited by 102 · see citing works
IEEE International Symposium on Biomedical Imaging
(2017) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2017.2730381 · Cited by 81 · see citing works
Current Trends on Deep Learning Models for Brain Tumor Segmentation and Detection – A Review
Somasundaram, S., & Gobinath, R. (2019) · 2019 International Conference on Machine Learning, Big Data, Cloud and Parallel Computing (COMITCon) · doi:10.1109/comitcon.2019.8862209 · Cited by 68 · see citing works
Accurate Uncertainty Estimation and Decomposition in Ensemble Learning
Liu, J. Z., Paisley, J., Kioumourtzoglou, M.-A., & Coull, B. (2019) · arXiv · doi:10.48550/arxiv.1911.04061 · Cited by 59 · see citing works
Estimating Uncertainty in Neural Networks for Cardiac MRI Segmentation: A Benchmark Study
Ng, M., Guo, F., Biswas, L., Petersen, S. E., Piechnik, S., Neubauer, S., & Wright, G. (2020) · IEEE Transactions on Biomedical Engineering · doi:10.1109/tbme.2022.3232730 · Cited by 59 · see citing works
Uncertainty Estimation via Stochastic Batch Normalization
Atanov, A., Ashukha, A., Molchanov, D., Neklyudov, K., & Vetrov, D. (2019) · Lecture Notes in Computer Science · doi:10.1007/978-3-030-22796-8_28 · Cited by 50 · see citing works
QualityNet: Segmentation quality evaluation with deep convolutional networks
Huang, C., Wu, Q., & Meng, F. (2016) · 2016 Visual Communications and Image Processing (VCIP) · doi:10.1109/vcip.2016.7805585 · Cited by 40 · see citing works
Evaluation of cell segmentation methods without reference segmentations
Chen, H., & Murphy, R. F. (2022) · Molecular Biology of the Cell · doi:10.1091/mbc.e22-08-0364 · Cited by 40 · see citing works
A Novel Quality Control Algorithm for Medical Image Segmentation Based on Fuzzy Uncertainty
Qiao, L., Chen, X., Chen, C., & Garibaldi, J. M. (2022) · IEEE Transactions on Fuzzy Systems · doi:10.1109/tfuzz.2022.3228332 · Cited by 32 · see citing works
Visual Communications and Image Processing
(2023) · Visual Communications and Image Processing · doi:10.23977/vcip · Cited by 24 · see citing works
Learning Active Learning from Real and Synthetic Data
Konyushkova, K., Sznitman, R., & Fua, P. (2017) · arXiv.org · doi:10.48550/arxiv.1703.03365 · Cited by 14 · see citing works
Robust Image Segmentation Quality AssessmentPreprint
Zhou, L., Deng, W., & Wu, X. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1903.08773 · Cited by 12 · see citing works
Subject-level Prediction of Segmentation Failure using Real-Time Convolutional Neural Nets
Robinson, R. H., Oktay, O., Bai, W., Valindria, V. V., Sanghvi, M. M., Aung, N., Paiva, J. M., Zemrak, F., Fung, K., Lukaschuk, E. I., et al. (2018) · Cited by 11 · see citing works
Molchanova, N., Raina, V., Malinin, A., Rosa, F. L., Müller, H., Gales, M., Granziera, C., Graziani, M., & Cuadra, M. B. (2023) · IEEE International Symposium on Biomedical Imaging (ISBI) · doi:10.1109/isbi53787.2023.10230563 · Cited by 9 · see citing works
ValUES: A Framework for Systematic Validation of Uncertainty Estimation in Semantic SegmentationPreprint
Kahl, K.-C., Lüth, C. T., Zenk, M., et al. (2024) · arXiv (Cornell University) · doi:10.48550/arxiv.2401.08501 · Cited by 8 · see citing works
Uncertainty Aware Segmentation Quality Assessment in Medical Images
O K, S., Galdran, A., Riera-Marin, M., García, J., Rodríguez-Comas, J., Piella, G., & González Ballester, M. A. (2024) · 2024 IEEE International Symposium on Biomedical Imaging (ISBI) · doi:10.1109/isbi56570.2024.10635509 · Cited by 4 · see citing works
Self-citation
Flagged first, then by citations
Research works printed but not linked: 1
- Konyushkova, 2017, Learning active learning from data, Adv. Neural Inf. Process. Syst., 30
Cited by
Sorted by citations
Thomas, N. R., Anitha, J., Popîrlan, C., Popirlan, C.-I., & Hemanth, D. J. (2025) · Computers, materials & continua/Computers, materials & continua (Print) · doi:10.32604/cmc.2025.070689 · Cited by 3 · see citing works
HAMF-Net: hybrid attention multi-stage fusion network for skin cancer detection
Sanmorino, A. (2025) · International Journal of Information Technology · doi:10.1007/s41870-025-02997-w · Cited by 2 · see citing works
Pierson, H. E., Shand, Z., Invik, J., Wang, Y., Ohal, V., Smith, K., Patel, D., & Pierson, R. A. (2025) · Discover Imaging. · doi:10.1007/s44352-025-00019-w · Cited by 1 · see citing works
He, P., Zhu, F., Kumar, R., et al. (2025) · International Symposium on Medical Information Processing and Analysis (SIPAIM) · doi:10.1109/sipaim67325.2025.11283433
Geshvadi, M., So, G., Chlorogiannis, D., et al. (2026) · SPIE Medical Imaging: Image Processing · doi:10.1117/12.3086577
Wang, X., Yu, W., Wang, G., et al. (2026) · Computerized Medical Imaging and Graphics · doi:10.1016/j.compmedimag.2026.102760
Luan, H., Lyu, Y., Dong, J., & Pan, L. (2026) · Lecture notes in computer science · doi:10.1007/978-3-032-16271-7_3
Alfken, P., Bereska, J. I., Hemke, R., Marquering, H. A., & Lisowska, A. (2026) · Lecture notes in computer science · doi:10.1007/978-3-032-30710-1_16
U-Net Based Framework for Automatic Skin Lesion Segmentation
Bhanuprathap, S., Rashmi, S., Hemachandran, K., et al. (2026) · Procedia Computer Science · doi:10.1016/j.procs.2026.06.309
Hua, S., Chen, T., Mi, Y., Shen, Y., Li, Y., Zhang, J., & Ji, Y. (2026) · Frontiers in Cell and Developmental Biology · doi:10.3389/fcell.2026.1940436
Kaur, H., & Singh, H. (2026) · International Conference on Natural Language Processing and Computer Vision (ICNPCV) · doi:10.1109/icnpcv69434.2026.11709005
Uncertainty-Aware Deep Learning for Skin Cancer Diagnosis
Mohammed, A. J., & Hassan, M. M. (2026) · Dasinya Journal for Engineering and Informatics · doi:10.65542/djei.v2i4.97
- Authors
- O.K., S., Riera-Marín, M., Galdran, A., López, J. G., Rodríguez-Comas, J., Piella, G., & Ballester, M. A. G.
- Affiliations
- Universitat Pompeu Fabra (ES)
- Venue
- Computerized Medical Imaging and Graphics
- Publisher
- Elsevier BV
- Published
- 2025-07
- Volume
- 123
- Pages
- 102547
- ISSN
- 0730-4862 (ISSN-L)
- 0895-6111
- OpenAlex
- W4409432992
- Semantic Scholar
- ceb476c2f3c016b379ea6b5873428231105dca11
- Cited by
- 11
- References
- 43
- Citations by year
- 2026: 7 · 2025: 4
- FWCI
- 3.95
- Type
- journal article
- Language
- en
- Open access
- hybrid
- doi.org
- Licence
- Locations
- Computerized Medical Imaging and Graphics (publishedVersion, cc-by-nc-nd)
- PubMed (publishedVersion)
- Repositori digital de la UPF (Universitat Pompeu Fabra) (publishedVersion, cc-by-nc-nd)
- arXiv (Cornell University) (submittedVersion)
- TECNALIA Publications (Fundación TECNALIA Research & Innovation) (submittedVersion, cc-by-nc-nd)
- arXiv (Cornell University) (cc-by)
- Funders
- Spain Ministry of Science and Innovation (CPP2023-010970)
- Topics
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Medical Image Segmentation Techniques
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
- Physical Sciences › Computer Science › Artificial Intelligence › AI in cancer detection
- Keywords
- uncertainty estimation, Bayesian deep learning, liver segmentation, uncertainty aggregation, Grad-CAM, interpretability
- MeSH terms
- Deep Learning, Bayes Theorem, Humans, Image Interpretation, Computer-Assisted, Image Processing, Computer-Assisted, Liver Neoplasms, Skin Neoplasms, Reproducibility of Results, Uncertainty
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-10-22.
Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-10-22.
Cited sources are checked by gpt-6-sol.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review