Optimal Transport for Information Bottleneck

A framework based on Optimal Transport (OT) with posterior probability for the computation of Information Bottleneck (IB) problem.

The Generalized Alternating Sinkhorn (GAS) algorithm is presented in "Information Bottleneck Revisited: Posterior Probability Perspective with Optimal Transport" https://ieeexplore.ieee.org/abstract/document/10206826

The Code for the OT type algorithm

https://github.com/wushitong1234/CommOT.git (AS in CommOT)

https://github.com/wushitong1234/Constrained-BA.git (CBA)

The Code detail

IB_OT_main for the OT version, corresponding to RD.

CBA_IB_RI for the CBA version, corresponding to RD.

CBA_IB_IR for the CBA version, corresponding to DR.