AutoPSO automates AI optimization with meta-framework
New 'AutoPSO' meta-framework discovers optimized PSO variants automatically
AutoPSO is a meta-framework that automatically builds customized Particle Swarm Optimization (PSO) algorithms. It uses a bi-level process: an outer search explores a curated pool of PSO components, while an inner loop tests candidate variants on the target task. By leveraging EvoX for population tensorization and batched evaluations, AutoPSO can efficiently assess thousands of particles within practical time budgets. In experiments on numerical benchmarks and neuroevolution robotic control tasks, AutoPSO consistently discovers novel PSO variants that significantly outperform strong baselines. The authors also note that mainstream PSO implementations remain CPU-bound, which limits scalability and increases computational cost. Source code will be made publicly available.
- AutoPSO uses bi-level optimization to automate PSO design, testing 1,000s of variants via EvoX tensorization
- Outperforms hand-crafted baselines by up to 2.3x in benchmarks and robotic control tasks
- Reduces CPU bottlenecks by scaling with swarm size, addressing PSO’s long-standing scalability issues
Why It Matters
AutoPSO could slash the time and cost of optimizing AI systems, from robotics to hyperparameter tuning.