atlas

Research map / Architectures and efficiency / Quantization

PTMQ: Post-training Multi-Bit Quantization of Neural Networks

AAAI 2024 · Architectures and efficiency › Quantization · approach: Ultra-low-bit LLM quantization · 10 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes PTMQ, a post-training (no retraining) multi-bit quantization framework that uses block-wise reconstruction with a Multi-bit Feature Mixer and Group-wise Distillation Loss to support real-time bit-width switching and mixed precision for CNNs and ViTs.

Problem

Multi-bit (arbitrary bit-width) quantization normally needs costly QAT (hundreds of GPU hours), and switching bit-widths requires recalculating norm-layer statistics, which prevents real-time switching.

Approach

Extends rounding-based PTQ (BRECQ/QDrop style block reconstruction) to multi-bit settings by sharing rounding values and activation step sizes across bit-width groups, mixing quantized features from different bit groups as block inputs (MFM), and distilling high-bit group outputs into lower-bit group outputs (GD-Loss); norm layers are folded so bit switching needs no recalculation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use PTMQ instead of QDrop at 3-bitResNet-18 · ImageNet, 3-bittop-1 accuracy: 66.26% -> 64.92%Hurt
Use PTMQ instead of QDrop at 4-bitResNet-18 · ImageNet, 4-bittop-1 accuracy: 69.54% -> 67.57%Hurt
Use PTMQ instead of QDrop at 3-bitResNet-50 · ImageNet, 3-bittop-1 accuracy: 71.07% -> 69.67%Hurt
Use PTMQ instead of QDrop at 4-bitResNet-50 · ImageNet, 4-bittop-1 accuracy: 74.98% -> 73.93%Hurt
Use PTMQ instead of QDrop at 3-bitRegNetX-600MF · ImageNet, 3-bittop-1 accuracy: 64.53% -> 59.29%Hurt
Use PTMQ instead of QDrop at 4-bitRegNetX-600MF · ImageNet, 4-bittop-1 accuracy: 70.62% -> 68.84%Hurt
Use PTMQ instead of QDrop at 4-bitMobileNet-V2 · ImageNet, 4-bittop-1 accuracy: 67.89% -> 64.94%Hurt
Use PTMQ instead of PTQ4ViTViT-S/224/16 · ImageNet, weights/activations 4/6top-1 accuracy: 71.41% -> 71.67%Helped
Use PTMQ instead of PTQ4ViTViT-B/224/16 · ImageNet, weights/activations 4/6top-1 accuracy: 75.43% -> 75.00%Hurt
Use PTMQ instead of PTQ4ViTDeiT-S/224/16 · ImageNet, weights/activations 4/6top-1 accuracy: 74.35% -> 77.20%Helped
Use PTMQ instead of PTQ4ViTDeiT-B/224/16 · ImageNet, weights/activations 4/6top-1 accuracy: 77.95% -> 80.00%Helped
Add mixed-precision quantization rather than uniform 4-bit quantizationPTMQ on ResNet-18 · ImageNet, ResNet-18top-1 accuracy: +2%Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet

Benchmarks: ImageNet

Metrics: top-1 accuracy GPU hours

Baselines: AdaRound BRECQ QDrop RobustQuant AnyPrecision MultiQuant PTQ4ViT HAWQ-V3 OMPQ LSQ

Builds on: BRECQ QDrop AdaRound MultiQuant

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy