Code for: Artemy Kolchinsky, Brendan D. Tracey, David H. Wolpert, "Nonlinear Information Bottleneck", https://arxiv.org/abs/1705.02436
Tested with Python3. Don't forget to install the dependencies via
pip3 install -r requirements.txtTo see a simple demonstration, see code/mnistdemo.py.
Note that this an updated implementation from 2019. To see the original implementation, take a look at the master-old branch.