tensorflow / probability

Probabilistic reasoning and statistical analysis in TensorFlow
https://www.tensorflow.org/probability/
Apache License 2.0
4.25k stars 1.1k forks source link

Variational inference for Conv2D layers #1813

Open haniehmsv opened 3 months ago

haniehmsv commented 3 months ago

Hello, I have been trying to use variational inference to the MLP and CNN (conv2D) layers for my research. Unlike a comprehensive discussion and examples for probabilistic MLP layers, I could not find any resources implying whether tensorflow_probability has also released VI for the conv2D layers. My network is built of a combination of MLP-CNN layers and I need to define probabilistic layers for all types of layers in my code. I would appreciate it if you could guide me through finding VI for CNN.