/pc-saddles

Code for NeurIPS 2024 paper "Only Strict Saddles in the Energy Landscape of Predictive Coding Networks?"

Primary LanguageJupyter NotebookMIT LicenseMIT

Only Strict Saddles in the Energy Landscape of Predictive Coding Networks?

Reproducibility

All the results and plots from the paper can be reproduced from the included notebooks. All notebooks except for the convergence experiments can be run in < 15min on a CPU. For the heavier convergence experiments, we recommend using a GPU.

  • Figure 1: Open in Colab
  • For the toy examples in Figure 2 (and the statistics in Figure 7): Open in Colab & Open in Colab
  • Figures 3-4 as well as 8-10: Open in Colab
  • Figure 5 as well as 11-12: Open in Colab
  • Figure 6: Open in Colab