Different ouput for combined uncertainty

[abhisheksambyal]

Hi,
I have implemented the Aleatoric, Epistemic, and combined uncertainty for MNIST generation but I got different output.
Your combined uncertainty output showed the complementary behavior but mine did not. Aleatoric unc is related to the digits, and other pixels are black, so uncertainty in the black region is zero. Epistemic unc on the other hand is explained near the digits reconstruction and not on the other area because it is easier for the model to learn about background (black) pixels.

Do share if you have some more insights.
Thanks