SIFMAP: Parallel CNN Inference using Sub-Input-Feature Maps in PIM

Tong Wu

doi:10.1109/tc.2026.3727031

SIFMAP: Parallel CNN Inference using Sub-Input-Feature Maps in PIM

2026-08-24

0

PRE

AI

DOI：10.1109/tc.2026.3727031

Original

Original request for help

Save

Abstract

En 中文

Processing-in-memory (PIM) accelerators perform in-situ dot-product operations between input feature maps (IFMs) and convolutional neural network (CNN) weights to accelerate inference. The efficiency of these computations is heavily influenced by the weight mapping strategy. However, existing approaches often adopt IFM-oriented mapping schemes that provide limited parallelism, leading to high latency and low energy efficiency. To address this issue, we propose SIFMAP, a subIFM-oriented weight mapping framework designed to enable highly parallel CNN inference. SIFMAP leverages the observation that different regions within an IFM contribute unevenly to computation. It first partitions the IFM into fine-grained sub-maps, or subIFMs, based on their computational characteristics. Then, it applies a reinforcement learning (RL) technique to determine optimal weight mapping strategies tailored to each subIFM, optimizing global performance. Furthermore, SIFMAP introduces a subIFM-oriented resource allocation scheme to further boost computational parallelism across subIFMs. Experimental results show that SIFMAP reduces inference latency by 4.8× and improves energy efficiency by 5.2× compared to state-of-the-art mapping methods under equivalent resource constraints.

Keywords:

Pattern-aware

Shift and duplicate kernel

Reinforcement Learning

Processing-in-memory

SIFMAP: Parallel CNN Inference using Sub-Input-Feature Maps in PIM

Abstract

Journal

Organization

Cited Papers