cs.ARMar 16, 2026

Co-Design of Memory-Storage Systems for Workload Awareness with Interpretable Models

Authors: Jay Sarkar, Vamsi Pavan Rayaprolu, Abhijeet Bhalerao

Organizations: Micron Technology San Jose, California, U.S.A.

Abstract

Solid-state storage architectures based on NAND or emerging memory devices (SSD), are fundamentally architected and optimized for both reliability and performance. Achieving these simultaneous goals requires co-design of memory components with firmware-architected Error Management (EM) algorithms for density- and performance-scaled memory technologies. We describe a Machine Learning (ML) for systems methodology and modeling for co-designing the EM subsystem together with the natural variance inherent to scaled silicon process of memory components underlying SSD technology. The modeling analyzes NAND memory components and EM algorithms interacting with comprehensive suite of synthetic (stress-focused and JEDEC) and emulation (YCSB and similar) workloads across Flash Translation abstraction layers, by leveraging a statistically interpretable and intuitively explainable ML algorithm. The generalizable co-design framework evaluates several thousand datacenter SSDs spanning multiple generations of memory and storage technology. Consequently, the modeling framework enables continuous, holistic, data-driven design towards generational architectural advancements. We additionally demonstrate that the framework enables Representation Learning of the EM-workload domain for enhancement of the architectural design-space across broad spectrum of workloads.

Explore similar work

CardsList
  1. Machine Learning-Driven Intelligent Memory System Design: From On-Chip Caches to Storage

    Mar 15, 2026Rahul Bera, Rakesh Nadig, Onur MutluAdaptive Learning

  2. Joint Hardware-Workload Co-Optimization for In-Memory Computing Accelerators

    Mar 4, 2026Olga Krestinskaya, Mohammed E. Fouda, Ahmed Eltawil +1Compute-in-MemoryHardware-Software Co-Design

  3. Energy Efficient Software Hardware CoDesign for Machine Learning: From TinyML to Large Language Models

    Mar 24, 2026Mohammad Saleh Vahdatpour, Yanqing ZhangPower-Efficient InferenceEnergy-Efficient ML