PSO_python_demo

Demo script (Python) of particle swarm optimization (PSO) partly translated from SDMBIGDAT19 (MATLAB).

• Slide: https://slides.com/iphysresearch/pso
• Blog: Particle Swarm Optimization From Scratch Using Python
• Video: Bilibili

Mock Data Challenge

• Given
  • Training data containing only noise: TrainingData.mat
  • Noise is Gaussian, stationary
  • Data to be analysed: analysisData.mat
  • Signal: Quadratic chirp
  • Parameter search ranges:
    • 40 < a1 < 100
    • 1 < a2 < 50
    • 11 < a3 < 15
• Detection and Estimation
  • Detection: Is there a signal in the analysis data?
  • Estimation: If so, estimate its parameters
  • Use PSO to obtain generalized likelihood ratio test (GLRT) and maximum likelihood estimate (MLE)
  • Recommended PSO parameters:
    • Best of 8 runs
    • Termination at 2000 iterations

Pre-requirements

• Python 3.x
• Numpy
• Scipy
• tqdm

How to use it

Just run demo.py script or demo.ipynb.

$ python demo.py

Reference

• 2021 Gravitational Wave Data Analysis School in China (Soumya D. Mohanty)
  • https://github.com/mohanty-sd/GWSC
  • https://github.com/BNUGW/GWSC21

Licence

• MIT