AI + Physical IntelligenceResearch projectEdge AI hardware

Why an AI chip sits idle, and what changes when it stops waiting on memory

A team of four students built a simulator for MobileNetV2, a network used for real-time camera perception in cars. In their baseline the chip spent 87% of its cycles waiting on memory. Keeping layer outputs on-chip made inference 2.1x faster at 2 MB of buffer, in simulation.

ConcentrationAI + Physical Intelligence
FormatPBL research paper
WhenApril–June 2026 · Team of 4
Project LeadResearcher in AI Hardware and Embedded Systems
InputMobileNetV2141 layers1 · Layer traceSimulator4 policiesthe only variable2 · Buffer policyOutput64 KB–8 MBper policy and buffer size3 · Cycles and DRAM traffic
Illustration of the team's method. The hardware model stays fixed; only the rule for what stays in on-chip memory changes.
87%Of cycles the chip waited on memory in the baseline
2.1xFewer cycles with activation reuse at a 2 MB buffer
12.3%Best gain from prefetching in a three-layer window
3.5%Slowdown at 2 MB, where prefetching got in the way
Abstract

Cars run camera networks on embedded chips under strict time and energy limits, and on those chips data movement, not arithmetic, sets the pace. This team built a trace-driven simulator that holds the hardware fixed and changes only the on-chip buffer policy, then compared four policies on MobileNetV2 across buffer sizes from 64 KB to 8 MB. Keeping activations on-chip cut simulated cycles 2.1x and memory traffic 2.7x at 2 MB, rising to 3.7x and 7x at 8 MB. Their own policy, which adds cautious prefetching of upcoming weights, helped by up to 12.3% in specific layer windows but slowed the network by 3.5% at 2 MB. In one 512 KB case, cycle-accurate DRAM tools confirmed a 1.52% DRAM-cycle and 0.88% DRAM-energy reduction. All results are from simulation.

Keywords · edge AI accelerators · memory scheduling · prefetching · MobileNetV2 · trace-driven simulation · DRAM energy

The team

Who did this work

4 students · 3 universities · 3 different majors
National Cheng Kung University
Engineering · Master's student
IndustryGoal: Build skills and portfolio
Concentration focus
AI + Physical Intelligence
University of Manchester
Computer Science · Recent graduate
ResearchGoal: Publish research
Concentration focus
AI + Physical Intelligence
Beijing Institute of Technology
AI / Machine Learning · Junior
ResearchGoal: Publish research
Concentration focus
AI + Physical Intelligence
National Cheng Kung University
Engineering · Master's student
ResearchGoal: Publish research
Concentration focus
AI + Physical Intelligence · AI + Health Tech

Students are not named. Year of study is as of fall 2026. Direction and goal come from each student's own program plan.

Background

1Why memory, not math, slows the chip

A car's cameras feed networks that detect objects and lanes on every frame, on a chip inside the vehicle. A missed deadline is a safety problem, and the power budget competes with range and heat [1].

MobileNetV2 was designed to need little arithmetic. But every layer still moves activations and weights between off-chip DRAM and a small on-chip buffer. If that buffer is managed poorly, the chip spends most of its time waiting. The team asked one question: with the hardware held fixed, how much does the buffer policy alone change speed and memory traffic?

Method

2Hold the hardware fixed, change one rule

  1. Built the simulator. A unified on-chip buffer, a 256 MAC-per-cycle compute array and a bandwidth-limited DRAM link, with sizes taken from configurations explored in SCALE-Sim [2].
  2. Fed it the real network. Layer-by-layer traces of all 141 MobileNetV2 layers [3].
  3. Compared four policies. Streaming everything from DRAM, pinning weights, keeping recently used tensors (LRU), and the team's priority policy, which adds cautious prefetching of the next layer's weights while reserving room for activations.
  4. Zoomed into three-layer windows. Swept buffer sizes from 64 KB to 8 MB to find the layer patterns where prefetching helps.
  5. Checked one case at DRAM level. Ran a 512 KB case through Ramulator and DRAMPower to confirm the traffic and energy change outside their own simulator.
Slide: edge accelerators manage their on-chip memory in software
Figure 1. The team's slide on why edge accelerators manage on-chip memory in software. From the final presentation.
Results

3Keeping activations on-chip mattered most; prefetching helped only with room to spare

64 KB2.9%128 KB9.9%192 KB9.9%256 KB12.3%512 KB1.7%1 MB3.8%2 MB4.2%0%15%
Figure 2. Best cycle reduction from the team's priority policy over LRU, per three-layer window, at each buffer size. Redrawn from the team's paper; all values from simulation.
PolicyBufferCycles vs. baselineDRAM traffic vs. baseline
Stream from DRAM (baseline)Any10.64M cycles, 87.1% waiting162.2 MB
Pin weightsAnyAbout the sameAbout the same
Keep recent tensors (LRU)2 MB2.1x fewer2.7x less
Keep recent tensors (LRU)8 MB3.7x fewer, 9.8% waiting7x less
Priority (team's policy)64 KB–1 MB0.1–0.5% better than LRUAbout the same as LRU
Priority (team's policy)2 MB3.5% worse than LRU—
Table 1. Full-network results from the team's paper. All figures are from simulation.
A result against their own idea. At 2 MB, the team's policy was slower than plain LRU (5.20M vs. 5.03M cycles). That is the size where large activations first fit on-chip, and a prefetched weight can push out an activation that costs more to reload. The team reported it as evidence for their main point: prefetching helps when there is spare room and hurts when the buffer is full.

In one 512 KB case the policy also cut DRAM reads by 2.23%. Cycle-accurate DRAM simulation (Ramulator) and an energy model (DRAMPower) confirmed a 1.52% drop in DRAM cycles and a 0.88% drop in DRAM energy for those windows.

From the team's paper and slides

4The paper and selected slides

Pages from the team's final paper and presentation. Student names are blurred.

First page of the team paper with author names blurred
Figure 3. First page of the team's paper: title and abstract.
Slide: MobileNetV2 trades computation for memory traffic
Figure 4. Slide: MobileNetV2 saves computation but still moves a lot of data.
Paper figure: total cycles and DRAM traffic by buffer size for four policies
Figure 5. From the paper: total cycles (left) and DRAM traffic (right) for the four policies across buffer sizes.
Slide: arithmetic intensity by layer
Figure 6. Slide: arithmetic intensity shows which layers are limited by memory.
Slide: best cycle reduction by buffer size
Figure 7. Slide: overall results, best cycle reduction at each buffer size.
Limitations

5What this does not show

  • All results come from simulation. Cycles are internally consistent units, not measured times on a real chip.
  • The simulator moves whole tensors and does not model tiling, separate on-chip memories or fused layers.
  • The DRAM energy check covers selected 512 KB windows, not the full network or the whole system.
  • MobileNetV2's shortcut connections are approximated as a straight chain of layers.
Skills and fit

6What this work shows the team can do

Hardware performance modelingTrace-driven simulationControlled experiment designMemory and cache policy designDesign space explorationCross-tool validation
Project Lead
PL

Researcher in AI Hardware and Embedded Systems

Did research at MIT on integrated photonics and AI-driven health sensing. 14+ journal papers and 5 patents. Builds wearable AI devices in industry.

The team designed the simulator, ran the experiments and wrote the paper. The Project Lead met with the team weekly and reviewed the research direction.

Good fit for applications in

Computer architecture · ML systems · Embedded AI

A research paper with a clear question and an honest negative result works well as a writing sample.

Project Lead review

7What the Project Lead observed

After the project, the Project Lead wrote an evaluation of each student and placed each one on a four-level scale. This section shows the highest level reached on the team. The notes below describe the team's project. Individual ratings are not published.

Highest level on this team

System-Level Insight

  1. 1Foundational Application
  2. 2Applied Reasoning
  3. 3Independent Problem Solving
  4. 4System-Level Insight

Level 4 of 4: anticipates broader implications, risks or downstream effects, and connects technical decisions to real-world impact.

Framing the problem
Grounded the research in a hardware bottleneck: lightweight convolutions cut computation, but not the latency and energy of moving activations and weights between DRAM and local memory.
Decisions and trade-offs
Refined the policy to keep LRU-style activation reuse while adding only conservative weight prefetching, and identified tiling as the natural next step for the accelerator design.
Evidence and testing
Organized a trace-driven method into one research story: comparisons with LRU, three-layer window analysis, and focused buffer cases at 192 KB and 512 KB.
Limits and risk
Connected scheduling choices to DRAM traffic, latency, energy and limited buffer capacity.

Where the Project Lead pushed further. State the validation boundary plainly: which findings come from the team's Python simulator and which were checked with Ramulator and DRAMPower.

Source: the Project Lead's written evaluations, June 2026.

References

References

  1. Lin, S.-C. et al. (2018). The architectural implications of autonomous driving: Constraints and acceleration. ASPLOS, 751–766.
  2. Raj, R. et al. (2025). SCALE-Sim v3: A modular cycle-accurate systolic accelerator simulator for end-to-end system analysis. ISPASS, 186–200.
  3. Sandler, M. et al. (2018). MobileNetV2: Inverted residuals and linear bottlenecks. CVPR.
AI+X team project (2026). A trace-driven study of priority-aware tensor prefetching for MobileNetV2. AI in Hardware, PBL research paper, June 2026. blendedlearn.org/outcome-mobilenetv2-memory-prefetching
Build work like this

Join the AI in Hardware program

This outcome came out of this project. Work with an AI hardware researcher on a real performance problem, and leave with research you can show.