Efficient VLA inference: replacing the pi0.5 action expert with a compact Mamba modelExpert
MambaAct: Efficient Mamba Action Experts for Flow-Matching Vision-Language-Action Models
Swapping the 311.5M-parameter action module of Physical Intelligence's pi0.5 robot model for a 29.7M-parameter Mamba module cut end-to-end response time on an H100 from 228.6 ms to 107.6 ms (52.9% faster) and nudged LIBERO success up from 96.72% to 97.80%, in an MBZUAI and DTU preprint. On two real UR5 tasks the leaner model also handled unseen object positions better.