udeepam/vib

Theory and PyTorch implementation of Deep Variational Information Bottleneck

Deep Variational Information Bottleneck (VIB)

This repository contains theory and pytorch code for building and training deep variational information bottleneck (VIB).

Overview

Models that have been implemented with corresponding theory:

• Deep VIB [Paper] [Notebook]