Visual TL;DR. MoE LLMs on CIM leads to Hardware Noise. Hardware Noise leads to Routing Suboptimality. Routing Suboptimality causes Performance Degradation. ROMER Framework enables Restores Load Balance. ROMER Framework enables Stabilizes Routing. Restores Load Balance shows Broad Generalizability. Stabilizes Routing shows Broad Generalizability.
- MoE LLMs on CIM: sparse expert activation promising for memory bandwidth-intensive compute-in-memory architectures
- Hardware Noise: analog CIM hardware imperfections perturb stored weights, disrupting expert load balance
- Routing Suboptimality: standard routing decisions become consistently suboptimal in the presence of hardware noise
- Performance Degradation: critical disruption in load balance and suboptimal routing directly impacting model performance
- ROMER Framework: a novel calibration framework for noisy MoE deployments on analog CIM systems
- Restores Load Balance: significantly improves accuracy by restoring expert load balance and stabilizing routing
- Stabilizes Routing: significantly improves accuracy by restoring expert load balance and stabilizing routing
- Broad Generalizability: demonstrates broad generalizability across various MoE models and hardware noise levels
Visual TL;DR
