Timezone: »
We study the problem of training deep networks while quantizing parameters and activations into low-precision numeric representations, a setting central to reducing energy consumption and inference time of deployed models. We propose a method that learns different precisions, as measured by bits in numeric representations, for different weights in a neural network, yielding a heterogeneous allocation of bits across parameters. Learning precisions occurs alongside learning weight values, using a strategy derived from a novel framework wherein the intractability of optimizing discrete precisions is approximated by training per-parameter noise magnitudes. We broaden this framework to also encompass learning precisions for hidden state activations, simultaneously with weight precisions and values. Our approach exposes the objective of constructing a low-precision inference-efficient model to the entirety of the training process. Experiments show that it finds highly heterogeneous precision assignments for CNNs trained on CIFAR and ImageNet, improving upon previous state-of-the-art quantization methods. Our improvements extend to the challenging scenario of learning reduced-precision GANs.
Author Information
Pedro Savarese (TTIC)
Xin Yuan (University of Chicago)
Yanjing Li (University of Chicago)
Michael Maire (University of Chicago)
More from the Same Authors
-
2022 : On Convexity and Linear Mode Connectivity in Neural Networks »
David Yunis · Kumar Kshitij Patel · Pedro Savarese · Gal Vardi · Jonathan Frankle · Matthew Walter · Karen Livescu · Michael Maire -
2023 Poster: Accelerated Training via Incrementally Growing Neural Networks using Variance Transfer and Learning Rate Adaptation »
Xin Yuan · Pedro Savarese · Michael Maire -
2021 Poster: Online Meta-Learning via Learning with Layer-Distributed Memory »
Sudarshan Babu · Pedro Savarese · Michael Maire -
2020 Poster: Winning the Lottery with Continuous Sparsification »
Pedro Savarese · Hugo Silva · Michael Maire -
2020 Poster: Self-Supervised Visual Representation Learning from Hierarchical Grouping »
Xiao Zhang · Michael Maire -
2020 Spotlight: Self-Supervised Visual Representation Learning from Hierarchical Grouping »
Xiao Zhang · Michael Maire