Compact safe planners enable low-power robotic task control

SLoSa: Small, Low-Power, and Safe Foundation Models for Robotic Task Planning

RoboticsArtificial IntelligenceLogic in Computer Science

Summary

Robots need to plan tasks carefully to be safe and successful, but traditional planning methods can be inflexible and slow. Large language models can help with flexible planning but are often too big and power-hungry for simple robots. The authors created SLoSa, a way to train smaller, efficient models that combine symbolic planning and learning while keeping safety in mind. Their approach works well on small devices and real robots, balancing performance with low power use.

What this means in practice

  • For robotics engineers: Implement safe and efficient task planners on low-power embedded robot controllers for household and industrial robots.
  • For edge ai developers: Deploy compact foundation models with verifier-guided fine-tuning to reduce power and hardware needs while preserving safety in real-time robotic environments.

Authors

Weizhe Xu, Jialiang Fan, Essien Taylor, Zijun Wang, Russell Joseph, Fanxin Kong

Abstract

Safe robotic task planning requires planners that can reliably achieve task goals while satisfying safety constraints. Classical planners search over explicit symbolic task representations and can provide formal guarantees with respect to encoded safety constraints, but require rigid input formats. Their computational cost also increases rapidly with the search space. In contrast, large language models (LLMs) offer flexible task interfaces and strong generalization capabilities, but they are large, power-hungry, and pre-trained with little robot-specific safety data and knowledge. Thus, they can be difficult to deploy locally on robotic systems with limited resources. To address these limitations, we present SLoSa, a neuro-symbolic compression and fine-tuning framework for training foundation-model-based robotic task planners with Small size, Low power demand, and Safety awareness. The framework first constructs a safety-aware dataset using classical planners, an advanced teacher LLM, and formal verifiers, followed by two training stages: supervised fine-tuning and reinforcement learning (RL) fine-tuning guided by a verification-based reward machine to enforce safety alignment. We evaluate SLoSa across four Planning Domain Definition Language (PDDL) domains, two household simulation environments, local edge platforms, and a desktop robotic arm. Across the PDDL tasks, verifier-guided post-training raises the success rate of compact 4B-14B planners to 70.5-76.0%, while INT4 deployment on a Jetson Orin requires 17.85-26.41 W of average power. Results in household simulators and on the physical robot further demonstrate the applicability of SLoSa across diverse robotic task settings.