arXiv:2503.20507cs.ARcs.DC2025-03被引 2

用多智能体强化学习同时优化存储系统数据放置与迁移,提升性能并降低开销。

Harmonia: Enhancing Data Placement and Migration in Hybrid Storage Systems via Multi-Agent Reinforcement Learning

  • 设计两个协同的强化学习智能体,分别负责数据放置和迁移决策。
  • 在4种异构设备上测试,平均性能比现有方法提升39.2%。
  • 低延迟(240纳秒)和小内存开销(206千字节),适合实际部署。

现代高性能计算环境依赖混合存储系统(HSS),融合多种具有不同延迟、带宽、耐久性和容量特性的存储设备,以满足数据密集型应用对性能、容量和成本的要求。HSS的性能高度依赖于两项关键数据管理策略:(1) 数据放置,决定将应用数据存于最合适的存储设备;(2) 数据迁移,动态地在设备间重新组织数据(如预取热数据、驱逐冷数据),以维持高性能。这两项策略紧密耦合,单独优化其一会导致整体性能不佳。然而,以往工作仅关注其中一项。本文提出Harmonia,一种基于多智能体强化学习的数据管理技术,通过两个轻量级自主智能体——数据放置智能体与数据迁移智能体——协同工作,在适应当前工作负载和系统配置的同时实现策略优化。我们在包含最多四个异构存储设备的真实系统上,对25个数据密集型工作负载进行了评估。在双设备性能/成本优化的HSS中,Harmonia平均优于最佳现有方法29.3%(44.8%)。在三(四)设备系统中,平均提升分别为38.9%(39.2%)。其性能优势伴随极低延迟(推理仅240纳秒)和小存储开销(两智能体共占206千字节DRAM)。

原文摘要 · Abstract (English)

Modern high-performance computing (HPC) environments rely on hybrid storage systems (HSS) that combine multiple storage devices with diverse latency, bandwidth, endurance, and capacity characteristics to meet the performance, capacity, and cost requirements of data-intensive applications. The performance of an HSS highly depends on two key data-management policies: (1) data placement, which determines the most suitable storage device to store application data, and (2) data migration, which dynamically reorganizes previously-stored data across storage devices (i.e., prefetching hot data and evicting cold data) to sustain high HSS performance. These policies are tightly interdependent, and thus, improving one without considering the other leads to suboptimal HSS performance. Unfortunately, prior works focus on optimizing only one of the policies. Our goal is to design a holistic data-management technique that optimizes both data-placement and data-migration policies to fully exploit the potential of an HSS. To this end, we propose Harmonia, a multi-agent reinforcement learning (RL)-based data-management technique. Harmonia employs two lightweight autonomous RL agents, a data-placement agent and a data-migration agent, that adapt their policies for the current workload and HSS configuration while coordinating with each other. We evaluate Harmonia on real HSS configurations with up to four heterogeneous storage devices and 25 data-intensive workloads. On a performance- (cost-) optimized HSS with two heterogeneous storage devices, Harmonia outperforms the best-performing prior approach by 29.3% (44.8%) on average. On an HSS with three (four) devices, Harmonia outperforms the best-performing prior work by 38.9% (39.2%) on average. Harmonia's performance benefits come with low latency (240 ns for inference) and storage (206 KiB in DRAM for both RL agents combined) overheads.

存储优化强化学习混合存储

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。