Bayesian uncertainty-weighted loss for improved generalisability on polyp segmentation task

arXiv:2309.06807v2 Announce Type: replace
Abstract: While several previous studies have devised methods for segmentation of polyps, most of these methods are not rigorously assessed on multi-center datasets. Variability due to appearance of polyps from one center to another, difference in endoscopic instrument grades, and acquisition quality result in methods with good performance on in-distribution test data, and poor performance on out-of-distribution or underrepresented samples. Unfair models have serious implications and pose a critical challenge to clinical applications. We adapt an …

abstract acquisition arxiv bayesian center cs.ai cs.cv datasets difference grades loss quality replace segmentation studies type uncertainty while