/Equal-Bits-BNN

Implementation of accepted AAAI 2022 paper "Equal Bits: Enforcing Equally Distributed Binary Network Weights"

Primary LanguagePython

Equal Bits: Enforcing Equally Distributed Binary Network Weights

This is the PyTorch implementation of accepted AAAI 2022 paper: Equal Bits: Enforcing Equally Distributed Binary Network Weights

Illustrative 2D example

Full binary network: combinations/solutions (4096/76) Bi-half subnetwork: combinations/solutions (924/66) Solution space vs bit-ratios
Decision Boundaries

Citation

If you find the code in this repository useful for your research consider citing it.

@article{li2022equal,
  title={Equal Bits: Enforcing Equally Distributed Binary Network Weights},
  author={Li, Yunqiang and Pintea, Silvia L and van Gemert, Jan C},
  journal={AAAI},
  year={2022}
}

Contact

If you have any problem about our code, feel free to contact