mlr-org / mlr3learners

Recommended learners for mlr3
https://mlr3learners.mlr-org.com
GNU Lesser General Public License v3.0
91 stars 15 forks source link

RegrKm SE prediction for epistemic uncertainty #173

Open mb706 opened 3 years ago

mb706 commented 3 years ago

RegrKm SE prediction with nugget currently predicts the epistemic uncertainty (uncertainty of the mean prediction) + aleatoric uncertainty (nugget SE, uncertainty that the model sees as random error). E.g. sampling lots of points with SE 0.1 noise and fitting a GP through them gives us

image

Where the GP goes through the points, the epistemic uncertainty is relatively low (it is the mean of a large sample) but the aleatoric uncertainty has SE 0.1 (nugget estimate).

What instead would be interesting would be the epistemic uncertainty alone

image

This is just

sqrt(pmax(p$sd^2 - self$model@covariance@nugget, 0))

I suggest we introduce a hyperparameter that gives the option to predict this, could be interesting for MBO.

mllg commented 3 years ago

PR would be welcome.