Distributional Gaussian Process Layers for Outlier Detection in Image Segmentation
We propose a parameter efficient Bayesian layer for hierarchical convolutional Gaussian Processes that incorporates Gaussian Processes operating in Wasserstein-2 space to reliably propagate uncertainty. This directly replaces convolving Gaussian Processes with a distance-preserving affine operator on distributions. Our experiments on brain tissue-segmentation show that the resulting architecture approaches the performance of well-established deterministic segmentation algorithms (U-Net), which has never been achieved with previous hierarchical Gaussian Processes. Moreover, by applying the same segmentation model to out-of-distribution data (i.e., images with pathology such as brain tumors), we show that our uncertainty estimates result in out-of-distribution detection that outperforms the capabilities of previous Bayesian networks and reconstruction-based approaches that learn normative distributions.
Code (0)
등록된 구현이 없습니다.
Tasks
Gaussian ProcessesImage SegmentationOutlier DetectionOut-of-Distribution DetectionSegmentationSemantic SegmentationSimilar Papers 제목 키워드 기반
Unsupervised Outlier Detection using Random Subspace and Subsampling Ensembles of Dirichlet Process Mixtures
Probabilistic mixture models are recognized as effective tools for unsupervised outlier detection owing to their interpretability and global characteristics. Among these, Dirichlet process mixture models stand out as a s…
Outlier DetectionVariational InferenceAdaptive Outlier Detection for Power MOSFETs Based on Gaussian Process Regression
Outlier detection of semiconductor devices is important since manufacturing variation is inherently inevitable. In order to properly detect outliers, it is necessary to consider the discrepancy from underlying trend. Con…
GPROutlier DetectionregressionQuantitative Gaussian Approximation of Randomly Initialized Deep Neural Networks
Given any deep fully connected neural network, initialized with random Gaussian parameters, we bound from above the quadratic Wasserstein distance between its output distribution and a suitable Gaussian process. Our expl…
Non-Parametric Outlier Synthesis
Out-of-distribution (OOD) detection is indispensable for safely deploying machine learning models in the wild. One of the key challenges is that models lack supervision signals from unknown data, and as a result, can pro…
Out-of-Distribution DetectionDistributional Gaussian Processes Layers for Out-of-Distribution Detection
Machine learning models deployed on medical imaging tasks must be equipped with out-of-distribution detection capabilities in order to avoid erroneous predictions. It is unsure whether out-of-distribution detection model…
Gaussian ProcessesOut-of-Distribution DetectionSegmentation