kchua/handful-of-trials

Could you explain the implementation of modeling variance?

TomorrowIsAnOtherDay opened this issue · 2 comments

https://github.com/kchua/handful-of-trials/blob/master/dmbrl/modeling/models/BNN.py#L414-L415
Intuitively we can directly model the Gaussian variance using the output of last layer, but you created max_logvar&min_logvar. Could you explain this kind of implementation for variance?

kchua commented

max_logvar and min_logvar were included as part of the implementation to ensure prediction stability. In practice, we found that the variance predictions tended to be quite large away from data, rendering trajectory sampling methods unstable, so these two quantities were included to help alleviate the problem.

Got it ! Thanks for your detailed explanation.