Abstract:The Large Language Models (LLMs) have been popular and widely used in creative ways because of their powerful capabilities. However, the substantial model size and complexity prevent LLMs from being implemented on resourceconstrained computing devices efficiently. Recent works that utilize quantization to boost the on-device efficiency of LLMs show that 8-bit or lower weight quantization is feasible with minimal impact on task performance with full-precision activation. Nevertheless, weight-only quantization does not leverage the potential acceleration on general edge processors, which typically support 16x16 and 8x8 integer multipliers. Additionally, it cannot benefit from specialized devices like Field-Programmable Gate Arrays (FPGAs), which offer reconfigure computing to design custom multipliers of arbitrary bit width. In this paper, we propose HotaQ, a Hardware-oriented token adaptive Quantization framework for general LLMs, and implement end-to-end accelerators on multiple devices. Considering the hardware profiling and activation analysis, we first introduce a basic activation quantization strategy to balance the trade-off of task performance and real inference speed. Then we integrate the activation-aware token pruning workflow to reduce the outliers and the adverse impact on attentivity. For mobile devices, we notably design the SIMD-based 4-bit multiplier and the efficient TRIP matrix multiplication for mobile devices. For FPGAs, we specially utilize the DSP packing techniques for 4/8-bit systolic-array-based multipliers and introduce an analytical model for achieving the best performance with limited on-chip resources. We apply our framework on different scales of LLMs, including LLaMA, OPT, and BLOOM with 4/8-bit for the activation and 4-bit for the weight. Experiments show that HotaQ achieves quantization on both model weights and activations while maintaining task performance comparable to existing weight-only quantization methods or even FP16 models. In the 4/8-bit scenario, HotaQ achieves an on-device speedup of up to 2.5× and 5.2× compared to its FP16 counterparts across edge devices and FPGAs respectively, marking a pioneering advancement in this domain.

Zeroth-Order Fine-Tuning of LLMs with Extreme Sparsity

Zeroth-Order Fine-Tuning of LLMs in Random Subspaces

Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmark

Enhancing Zeroth-order Fine-tuning for Language Models with Low-rank Structures

HotaQ: Hardware Oriented Token Adaptive Quantization for Large Language Models

Simultaneous Computation and Memory Efficient Zeroth-Order Optimizer for Fine-Tuning Large Language Models

Variance-reduced Zeroth-Order Methods for Fine-Tuning Language Models

AdaZeta: Adaptive Zeroth-Order Tensor-Train Adaption for Memory-Efficient Large Language Models Fine-Tuning

Second-Order Fine-Tuning without Pain for LLMs:A Hessian Informed Zeroth-Order Optimizer

Second-Order Fine-Tuning Without Pain for LLMs: A Hessian Informed Zeroth-Order Optimizer.

QFT: Quantized Full-parameter Tuning of LLMs with Affordable Resources

Full Parameter Fine-tuning for Large Language Models with Limited Resources

Enabling Efficient On-Device Fine-Tuning of LLMs Using Only Inference Engines

Quantized Side Tuning: Fast and Memory-Efficient Tuning of Quantized Large Language Models

Low-Rank Quantization-Aware Training for LLMs

One QuantLLM for ALL: Fine-tuning Quantized LLMs Once for Efficient Deployments

FineQuant: Unlocking Efficiency with Fine-Grained Weight-Only Quantization for LLMs

DPZero: Private Fine-Tuning of Language Models without Backpropagation

On the Convergence of Zeroth-Order Federated Tuning for Large Language Models

VPTQ: Extreme Low-bit Vector Post-Training Quantization for Large Language Models

PocketLLM: Enabling On-Device Fine-Tuning for Personalized LLMs