PO-CKAN Review Experiments

This repository provides the implementation for the PO-CKAN review-stage manuscript experiments.

Contents

• experiments/burgers/: low-viscosity Burgers' equation experiment and data-generation scripts.
• experiments/eikonal/: Eikonal equation experiment.
• experiments/fpde/: FPDE experiment.
• experiments/diffusion_reaction/: diffusion-reaction experiment.
• shared/: shared PO-CKAN layers.
• matlab/: MATLAB data-generation helpers for Burgers' equation.

The repository includes source code only. Datasets, generated result tables, figures, logs, notebooks, and training checkpoints are managed outside the source release.

Environment

Install the Python dependencies listed in requirements.txt. GPU execution depends on the CUDA, JAX, and PyTorch configuration of the target environment.

Example Commands

python experiments/burgers/run_experiment.py --model po_ckan --nu 0.001
python experiments/eikonal/run_experiment.py
python experiments/diffusion_reaction/run_experiment.py
python experiments/fpde/run_experiment.py

The low-viscosity Burgers' scripts read data/Burger0.001.mat. Prepare this dataset with the included data-generation utilities, or provide an equivalent file at that path before running the Burgers' training and plotting scripts.