Erick F, Müller J, Li Z, Kainz B (2026)
Publication Type: Conference contribution
Publication year: 2026
Publisher: Springer Science and Business Media Deutschland GmbH
Book Volume: 15967 LNCS
Pages Range: 278-287
Conference Proceedings Title: Lecture Notes in Computer Science
Event location: Daejeon, KOR
ISBN: 9783032049834
DOI: 10.1007/978-3-032-04984-1_27
Developing robust machine learning algorithms is of utmost importance for their applications to biomedical imaging applications. This issue is non-trivial, as networks are generally trained with datasets taken from relatively homogeneous samples dominated by statistically more probable disease classes, leading to unbalanced class distributions. One possible solution is to resolve the intrinsic biases towards certain dominating classes in the training datasets through more data collection with a more diverse sample, which is often prohibitively expensive. Another solution is to directly implement established uncertainty estimation measures for more robust predictions, which are nevertheless computationally demanding and insensitive to class imbalance. To address this issue, we propose a novel class-aware and uncertainty-aware pseudocoreset framework consisting of the following components: 1) An efficient framework with last layer Laplacian approximation 2) Class-aware calibration with error-based regularization, and 3) a Wasserstein distance-based regularization which explicitly imposes uncertainty-awareness. We evaluate our method for In-Distribution calibration, Out-of-Distribution inference, and class balance evaluations in two public skin cancer datasets taken from samples from different geographical location with differing skin colors. Our method outperforms various baseline uncertainty quantification and Bayesian pseudocoreset methods.
APA:
Erick, F., Müller, J., Li, Z., & Kainz, B. (2026). Last Layer Laplacian Pseudocoresets for Robust Medical Image Analysis. In James C. Gee, Jaesung Hong, Carole H. Sudre, Polina Golland, Daniel C. Alexander, Juan Eugenio Iglesias, Archana Venkataraman, Jong Hyo Kim (Eds.), Lecture Notes in Computer Science (pp. 278-287). Daejeon, KOR: Springer Science and Business Media Deutschland GmbH.
MLA:
Erick, Franciskus, et al. "Last Layer Laplacian Pseudocoresets for Robust Medical Image Analysis." Proceedings of the 28th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2025, Daejeon, KOR Ed. James C. Gee, Jaesung Hong, Carole H. Sudre, Polina Golland, Daniel C. Alexander, Juan Eugenio Iglesias, Archana Venkataraman, Jong Hyo Kim, Springer Science and Business Media Deutschland GmbH, 2026. 278-287.
BibTeX: Download