Paper Reading AI Learner

FraQAT: Quantization Aware Training with Fractional bits

2025-10-16 16:01:08
Luca Morreale, Alberto Gil C. P. Ramos, Malcolm Chadwick, Mehid Noroozi, Ruchika Chavhan, Abhinav Mehrotra, Sourav Bhattacharya

Abstract

State-of-the-art (SOTA) generative models have demonstrated impressive capabilities in image synthesis or text generation, often with a large capacity model. However, these large models cannot be deployed on smartphones due to the limited availability of on-board memory and computations. Quantization methods lower the precision of the model parameters, allowing for efficient computations, \eg, in \INT{8}. Although aggressive quantization addresses efficiency and memory constraints, preserving the quality of the model remains a challenge. To retain quality in previous aggressive quantization, we propose a new fractional bits quantization (\short) approach. The novelty is a simple yet effective idea: we progressively reduce the model's precision from 32 to 4 bits per parameter, and exploit the fractional bits during optimization to maintain high generation quality. We show that the \short{} yields improved quality on a variety of diffusion models, including SD3.5-Medium, Sana, \pixart, and FLUX.1-schnell, while achieving $4-7\%$ lower FiD than standard QAT. Finally, we deploy and run Sana on a Samsung S25U, which runs on the Qualcomm SM8750-AB Snapdragon 8 Elite Hexagon Tensor Processor (HTP).

Abstract (translated)

最先进的(SOTA)生成模型在图像合成或文本生成方面表现出令人印象深刻的能力,通常使用的是大规模模型。然而,由于手机上的内存和计算资源有限,这些大型模型无法部署到智能手机上。量化方法通过降低模型参数的精度来实现高效的计算,例如采用INT8格式。尽管激进的量化方法可以解决效率和内存限制问题,但保持模型质量仍是一个挑战。为了在之前的激进量化中保留质量,我们提出了一种新的分数位量化(简称FracBits)方法。其新颖之处在于一个简单而有效的想法:我们逐步将模型精度从32位减少到每参数4位,并利用分数位进行优化以保持高质量的生成效果。我们展示了FracBits在多种扩散模型上的改进,包括SD3.5-Medium、Sana、PixArt和FLUX.1-schnell,在FiD指标上比标准QAT低4-7%。最后,我们在三星S25U设备(搭载高通SM8750-AB骁龙8 Elite Hexagon Tensor Processor (HTP))上部署并运行了Sana模型。 这一段翻译概述了一种新的分数位量化方法,旨在解决大型生成模型在手机等资源受限的设备上的部署问题。该方法通过逐步减少模型参数的精度,并利用分数位进行优化,在保持高质量的同时实现了更高效的计算。

URL

https://arxiv.org/abs/2510.14823

PDF

https://arxiv.org/pdf/2510.14823.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot