Search papers, labs, and topics across Lattice.
This paper introduces FairQuant, a framework for fairness-aware mixed-precision quantization in medical image classification that optimizes both model accuracy and fairness under bit-rate constraints. FairQuant combines group-aware importance analysis, budgeted mixed-precision allocation, and a learnable Bit-Aware Quantization (BAQ) mode to jointly optimize weights and per-unit bit allocations. Experiments on Fitzpatrick17k and ISIC2019 datasets using ResNet18/50, DeiT-Tiny, and TinyViT architectures demonstrate that FairQuant achieves comparable accuracy to uniform 8-bit quantization while improving worst-group performance relative to uniform 4- and 8-bit baselines under shared bit budgets.
Quantizing your medical image classifier can disproportionately hurt underrepresented groups, but FairQuant offers a way to compress models while boosting worst-group accuracy.
Compressing neural networks by quantizing model parameters offers useful trade-off between performance and efficiency. Methods like quantization-aware training and post-training quantization strive to maintain the downstream performance of compressed models compared to the full precision models. However, these techniques do not explicitly consider the impact on algorithmic fairness. In this work, we study fairness-aware mixed-precision quantization schemes for medical image classification under explicit bit budgets. We introduce FairQuant, a framework that combines group-aware importance analysis, budgeted mixed-precision allocation, and a learnable Bit-Aware Quantization (BAQ) mode that jointly optimizes weights and per-unit bit allocations under bitrate and fairness regularization. We evaluate the method on Fitzpatrick17k and ISIC2019 across ResNet18/50, DeiT-Tiny, and TinyViT. Results show that FairQuant configurations with average precision near 4-6 bits recover much of the Uniform 8-bit accuracy while improving worst-group performance relative to Uniform 4- and 8-bit baselines, with comparable fairness metrics under shared budgets.
