BiDense: Binarization for Dense Prediction

Rui Yin, Haotong Qin, Yulun Zhang, Wenbo Li, Yong Guo +3 more
11/15/2024
cs.CV

No official code found. Best implementation: frickyinn/BiDense (5 stars, Apache-2.0, last commit 2024-11-21).

Abstract

Dense prediction is a critical task in computer vision. However, previous methods often require extensive computational resources, which hinders their real-world application. In this paper, we propose BiDense, a generalized binary neural network (BNN) designed for efficient and accurate dense prediction tasks. BiDense incorporates two key techniques: the Distribution-adaptive Binarizer (DAB) and the Channel-adaptive Full-precision Bypass (CFB). The DAB adaptively calculates thresholds and scaling factors for binarization, effectively retaining more information within BNNs. Meanwhile, the CFB facilitates full-precision bypassing for binary convolutional layers undergoing various channel size transformations, which enhances the propagation of real-valued signals and minimizes information loss. By leveraging these techniques, BiDense preserves more real-valued information, enabling more accurate and detailed dense predictions in BNNs. Extensive experiments demonstrate that our framework achieves performance levels comparable to full-precision models while significantly reducing memory usage and computational costs.

View on arXivView PDF

Code Implementations(1)

frickyinn/BiDenseCommunity100%
Apache-2.0

PyTorch implementation of "BiDense: Binarization for Dense Prediction," A binary neural network for dense prediction tasks.

51PythonNov 14, 20241 years agoApache-2.0
model-compressionmodel-quantization

Similar Papers

Cite this paper

@article{yin2024bidense,
  title  = {BiDense: Binarization for Dense Prediction},
  author = {Rui Yin and Haotong Qin and Yulun Zhang and Wenbo Li and Yong Guo and Jianjun Zhu and Cheng Wang and Biao Jia},
  year   = {2024},
  eprint = {2411.10346},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.10346v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2411.10346.svg)](https://codeofpaper.com/paper/2411.10346)

Discussion