NIFA: Nonlinear IMC enhanced FPGA for efficient ML inference

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Emerging Technologies & Innovation · Depth: Expert, quick

Summary

NIFA is a novel FPGA architecture designed to enhance deep learning inference efficiency, particularly for Transformer-based models. It addresses limitations of conventional in-memory computing (IMC) blocks in FPGAs, which typically handle only static-weight vector-matrix multiplication (VMM) and offload nonlinear and dynamic matrix-matrix multiplication (DIMM) to the FPGA fabric. This prior approach limits IMC benefits for Transformer models and suffers from high area and power consumption by analog-to-digital converters (ADCs). NIFA integrates an ADC-free IMC block, replacing ADCs with analog content-addressable memories (ACAMs) that natively perform nonlinear operations and facilitate DIMM for attention computation. Through FPGA-aware design-space exploration, NIFA achieves up to 40x higher energy efficiency and 4.1x higher area efficiency on CNN benchmarks, and 1.9x higher energy efficiency and 2.5x higher area efficiency on Transformer-based benchmarks, sustaining robust gains across long input sequences.

Key takeaway

For AI Hardware Engineers designing next-generation FPGAs for deep learning inference, NIFA demonstrates a critical architectural shift. You should investigate integrating ADC-free in-memory computing blocks with analog content-addressable memories (ACAMs) to natively handle nonlinear and dynamic matrix-matrix multiplication. This approach significantly boosts energy and area efficiency for both CNNs and Transformer-based workloads, offering up to 40x and 1.9x energy efficiency gains respectively, and is crucial for sustaining performance on long input sequences.

Key insights

NIFA integrates ADC-free IMC with ACAMs into FPGAs for efficient nonlinear and dynamic operations in DL inference.

Principles

Method

Integrate ADC-free IMC with ACAMs, conduct FPGA-aware design-space exploration for optimal crossbar dimensions, and develop efficient mapping for DIMM using ACAMs.

In practice

Topics

Best for: Research Scientist, AI Hardware Engineer, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.