HCRMap: Pressure-Aware Hot-Expert Residency Mapping for 3.5D MoE Chiplet Inference
Summary
HCRMap is a hot expert residency mapping framework designed to optimize Mixture-of-Experts (MoE) Large Language Model (LLM) inference on 3.5D multi-chiplet systems. It addresses the persistent expert hotness skew where a small set of experts receives most tokens, causing compute imbalance and amplifying pressure on communication, memory bandwidth, I/O, and execution queues. HCRMap dynamically determines which experts to promote, retain, demote, or evict based on expert hotness, weight loading cost, migration overhead, and runtime resource pressure. It then maps routed token groups to suitable resident replicas, jointly mitigating various bottlenecks. Experimental results demonstrate HCRMap reduces end-to-end latency by 43.6% and 43.0% over Hydra in prefill and decode stages, respectively; by 34.5% and 33.1% over MoEntwine; and by 46.7% and 46.0% over PIMoE.
Key takeaway
For machine learning engineers deploying MoE LLMs on 3.5D multi-chiplet systems, you should consider HCRMap's dynamic expert replica management to significantly reduce inference latency and alleviate communication, memory, and queue bottlenecks. This approach offers substantial performance gains over existing methods like Hydra, MoEntwine, and PIMoE, making it critical for optimizing resource utilization in high-performance MoE deployments.
Key insights
HCRMap dynamically manages hot expert replicas to mitigate resource bottlenecks in MoE LLM inference on 3.5D chiplets.
Principles
- Expert hotness skew amplifies resource pressure.
- Dynamic replica placement reduces bottlenecks.
- Optimizing expert residency improves MoE inference.
Method
HCRMap dynamically promotes, retains, demotes, or evicts experts based on hotness, loading cost, migration overhead, and runtime resource pressure, then maps token groups to replicas.
In practice
- Reduce MoE LLM inference latency.
- Optimize 3.5D multi-chiplet utilization.
- Manage expert replicas efficiently.
Topics
- Mixture-of-Experts
- Large Language Models
- 3.5D Chiplets
- Inference Optimization
- Expert Residency Mapping
- Resource Management
- Latency Reduction
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Hardware Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.