Skip to main navigation Skip to search Skip to main content

ViM-Q: Scalable Algorithm-Hardware Co-Design for Vision Mamba Model Inference on FPGA

Research output: Chapters, Conference Papers, Creative and Literary Works › RGC 32 - Refereed conference paper (with host publication) › peer-review

Abstract

Vision Mamba (ViM) models offer a compelling efficiency advantage over Transformers by leveraging the linear complexity of State Space Models (SSMs), yet efficiently deploying them on FPGAs remains challenging. Linear layers struggle with dynamic activation outliers that render static quantization ineffective, while uniform quantization fails to capture the weight distribution at low bit-widths. Furthermore, while associative scan accelerates SSMs on GPUs, its memory access patterns are misaligned with the streaming dataflow required by FPGAs. To address these challenges, we present ViM-Q 1, a scalable algorithm-hardware co-design for end-to-end ViM inference on the edge. We introduce a hardware-aware quantization scheme combining dynamic per-token activation quantization and perchannel smoothing to mitigate outliers, alongside a custom 4-bit per-block Additive Power-of-Two (APoT) weight quantization. The models are deployed on a runtime-parameterizable FPGA accelerator featuring a linear engine employing a Lookup-Table (LUT) unit to replace multiplications with shift-add operations, and a fine-grained pipelined SSM engine that parallelizes the state dimension while preserving sequential recurrence. Crucially, the hardware supports runtime configuration, adapting to diverse dimensions and input resolutions across the ViM family. Implemented on an AMD ZCU102 FPGA, ViM-Q achieves an average 4.96× speedup and 59.8× energy efficiency gain over a quantized NVIDIA RTX 3090 GPU baseline for low-batch inference on ViM-tiny. This co-design shows a viable path for deploying ViM models on resource-constrained edge devices. © 2026 IEEE
Original languageEnglish
Title of host publication2026 IEEE 34th Annual International Symposium on Field-Programmable Custom Computing Machines (FCCM)
PublisherIEEE
Pages90-99
Number of pages10
ISBN (Electronic)979-8-3315-5815-4
ISBN (Print)979-8-3315-5816-1
DOIs
Publication statusPublished - 2026
Event34th IEEE International Symposium on Field-Programmable Custom Computing Machines (FCCM 2026) - Atlanta, United States
Duration: 13 May 2026 → 16 May 2026

Publication series

NameProceedings Annual IEEE Symposium on Field-Programmable Custom Computing Machines
ISSN (Print)2576-2613
ISSN (Electronic)2576-2621

Conference

Conference34th IEEE International Symposium on Field-Programmable Custom Computing Machines (FCCM 2026)
Abbreviated titleFCCM 2026
PlaceUnited States
CityAtlanta
Period13/05/26 → 16/05/26

Bibliographical note

Information for this record is supplemented by the author(s) concerned.

Funding

This work was supported by the Research Grants Council of the Hong Kong Special Administrative Region, China (Project No. CityU 11202925, CityU 11202124, and CityU 11201422), the Innovation and Technology Commission of Hong Kong (Project No. MHP/072/23), and City University of Hong Kong (Project 9678187).

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 7 - Affordable and Clean Energy
    SDG 7 Affordable and Clean Energy
  2. SDG 9 - Industry, Innovation, and Infrastructure
    SDG 9 Industry, Innovation, and Infrastructure
  3. SDG 11 - Sustainable Cities and Communities
    SDG 11 Sustainable Cities and Communities
  4. SDG 12 - Responsible Consumption and Production
    SDG 12 Responsible Consumption and Production

Research Keywords

  • Vision Mamba
  • State Space Models
  • FPGA
  • Quantization
  • Algorithm-Hardware Co-Design

RGC Funding Information

  • RGC-funded

Fingerprint

Dive into the research topics of 'ViM-Q: Scalable Algorithm-Hardware Co-Design for Vision Mamba Model Inference on FPGA'. Together they form a unique fingerprint.

Cite this