DiSPo: Diffusion-SSM based Policy Learning for Coarse-to-Fine Action Discretization
<|reference_start|>DiSPo: Diffusion-SSM based Policy Learning for Coarse-to-Fine Action Discretization: We aim to solve the problem of generating coarse-to-fine skills learning from demonstrations (LfD). To scale precision, traditional LfD approaches often rely on extensive fine-grained demonstrations with external interpolations or dynamics models with limited generalization capabilities. For memory-efficient learning and convenient granularity change, we propose a novel diffusion-SSM based policy (DiSPo) that learns from diverse coarse skills and produces varying control scales of actions by leveraging a state-space model, Mamba. Our evaluations show the adoption of Mamba and the proposed step-scaling method enables DiSPo to outperform in five coarse-to-fine benchmark tests while DiSPo shows decent performance in typical fine-grained motion learning and reproduction. We finally demonstrate the scalability of actions with simulation and real-world manipulation tasks.<|reference_end|>
@article{oh2024dispo:,
    title={DiSPo: Diffusion-SSM based Policy Learning for Coarse-to-Fine Action
  Discretization},
    author={Nayoung Oh, Moonkyeong Jung and Daehyung Park},
    journal={arXiv preprint arXiv:2409.14719},
    year={2024},
    archivePrefix={arXiv},
    eprint={2409.14719},
    primaryClass={cs.RO}
}