Bayesian uncertainty-weighted loss for improved generalisability on polyp segmentation task
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2023
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866914833991991296 |
|---|---|
| author | Stone, Rebecca S. Chavarrias-Solano, Pedro E. Bulpitt, Andrew J. Hogg, David C. Ali, Sharib |
| author_facet | Stone, Rebecca S. Chavarrias-Solano, Pedro E. Bulpitt, Andrew J. Hogg, David C. Ali, Sharib |
| contents | 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 implicit bias mitigation method which leverages Bayesian predictive uncertainties during training to encourage the model to focus on underrepresented sample regions. We demonstrate the potential of this approach to improve generalisability without sacrificing state-of-the-art performance on a challenging multi-center polyp segmentation dataset (PolypGen) with different centers and image modalities. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2309_06807 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | Bayesian uncertainty-weighted loss for improved generalisability on polyp segmentation task Stone, Rebecca S. Chavarrias-Solano, Pedro E. Bulpitt, Andrew J. Hogg, David C. Ali, Sharib Computer Vision and Pattern Recognition Artificial Intelligence 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 implicit bias mitigation method which leverages Bayesian predictive uncertainties during training to encourage the model to focus on underrepresented sample regions. We demonstrate the potential of this approach to improve generalisability without sacrificing state-of-the-art performance on a challenging multi-center polyp segmentation dataset (PolypGen) with different centers and image modalities. |
| title | Bayesian uncertainty-weighted loss for improved generalisability on polyp segmentation task |
| topic | Computer Vision and Pattern Recognition Artificial Intelligence |
| url | https://arxiv.org/abs/2309.06807 |