Recent advancements have demonstrated the potential of reinforcement learning (RL) in enhancing cache management policies. However, existing approaches still face two critical limitations: (1) most methods treat prefetching and cache replacement as separate tasks, which restricts their adaptability to diverse and dynamic workloads; and (2) the high computational and storage overheads of typical RL techniques hinder their deployment in real-world hardware systems. To overcome these challenges, we propose LightCacheRL, a lightweight and practical RL-based cache management framework that unifies prefetching and replacement decisions within a unified formulation. Specifically, LightCacheRL models the cache management problem as a lightweight online Multi-Armed Bandit (MAB) process, enabling efficient and adaptive policy learning with minimal runtime overhead. The reward function in LightCacheRL integrates both instruction-per-cycle (IPC) and system-level bandwidth feedback, providing a hardware-aware optimization objective. We conduct comprehensive evaluations through simulation and hardware synthesis across single-core and multi-core configurations. Experimental results show that LightCacheRL achieves 11.5% average IPC improvement over LRU on a 16-core system, and outperforms state-of-the-art policies, including Hawkeye (4.35%), Glider (3.75%), and Mockingjay (2.1%), with negligible hardware cost.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

LightCacheRL: A Lightweight Reinforcement Learning Framework for Unified Cache Management

  • Kunming Zhang,
  • Zhihua Fan,
  • Yingchun Fu,
  • Yanhuan Liu,
  • Lexin Wang,
  • Yuqun Liu,
  • Haibin Wu,
  • Wenming Li

摘要

Recent advancements have demonstrated the potential of reinforcement learning (RL) in enhancing cache management policies. However, existing approaches still face two critical limitations: (1) most methods treat prefetching and cache replacement as separate tasks, which restricts their adaptability to diverse and dynamic workloads; and (2) the high computational and storage overheads of typical RL techniques hinder their deployment in real-world hardware systems. To overcome these challenges, we propose LightCacheRL, a lightweight and practical RL-based cache management framework that unifies prefetching and replacement decisions within a unified formulation. Specifically, LightCacheRL models the cache management problem as a lightweight online Multi-Armed Bandit (MAB) process, enabling efficient and adaptive policy learning with minimal runtime overhead. The reward function in LightCacheRL integrates both instruction-per-cycle (IPC) and system-level bandwidth feedback, providing a hardware-aware optimization objective. We conduct comprehensive evaluations through simulation and hardware synthesis across single-core and multi-core configurations. Experimental results show that LightCacheRL achieves 11.5% average IPC improvement over LRU on a 16-core system, and outperforms state-of-the-art policies, including Hawkeye (4.35%), Glider (3.75%), and Mockingjay (2.1%), with negligible hardware cost.