File size: 3,467 Bytes
3d03e2b 2fb73c2 4531155 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | ---
license: other
tags:
- diffusion
- control
- controller-synthesis
---
# DiffusionControl checkpoints
Trained models for the DiffusionControl project (conditional diffusion models that generate
certified feedback controllers). Code, protocols and results live in the GitHub repo
`matCercola18/matte_server` under `DiffusionControl/`, branch `icml-experiments`; each family has
a research doc in `DiffusionControl/docs/`.
Files keep their repo paths (`models/...`), so code that loads `models/<name>.pth` works after
`huggingface_hub.snapshot_download`. Only final checkpoints are included (no `Last_*`, no
intermediate step checkpoints).
| family | files | doc |
|---|---|---|
| Furuta (QUBE-Servo 2), headline 1D U-Net generator | `models/v4/unet_a*_s96000.pth`, `unet_s_s40000.pth`, `unet_sxd_s40000.pth` | `furuta_generated_trajectory_v4.md` |
| Furuta regression twins and MLP ablations | `models/v4/reg_*`, `models/v4/mlp_*` | same |
| Furuta earlier trajectory models (v1-v3, clip-bug fix) | `Best_furuta_traj_cond_*` | `furuta_generated_trajectory*.md` |
| Furuta energy-shaping models (superseded) | `Best_furuta_cond`, `_full_cond`, `_handoff_cond` | `diffusion_complete_nonlinear_controller.md` |
| Pendulum | `Best_pendulum_*` (multi-solution `multi_*`, physics-informed `pg_*`/`abl_dl_*`, ablations `abl_*`) | `pendulum_multi_solution.md`, `physics_informed_generation.md`, `ablations.md`, `ood_shift.md` |
| SISO Lur'e | `Best_nonlinear_cond_popov_*`, `siso_pg_c*`, `costcond_siso`, `cert_perfspan_L`, `regressor_siso` | RESULTS.md, `physics_informed_generation.md` |
| MIMO 2x2 | `Best_mimo_cond_2x2*` | RESULTS.md |
| CVAE baselines | `Best_cvae_*` | `ablations.md` |
| Witsenhausen | `*witsenhausen_cond*` | `witsenhausen.md` |
Note: `Best_nonlinear_cond_popov_L_sector/_closedloop/_distill` are documented negative results.
## Round 2 (2026-10-03): `models/round2/`, `data/round2/`
Files here do not keep repo paths; point the loaders at them explicitly.
| family | checkpoints | data | doc |
|---|---|---|---|
| Furuta, friction-conditioned (mp, Lp, Dr, Dp, Coulomb Cr, Cp) | `round2/furuta_v5/v5_s{0,1,2}_s96000.pth` | `data/round2/furuta/furuta_nmpc_bank_v5merged.npz`, `furuta_traj_dataset_v5.pt` | `furuta_friction_hardware_ready.md` |
| Furuta, voltage-headroom (Vref) conditioned | `round2/furuta_f2/f2_s*_s8000.pth` | `furuta_bank_vref_*.npz`, `furuta_traj_dataset_f2.pt` | `furuta_robustness.md` |
| Furuta, unified hardware model (friction + Vref, 7-d) | `round2/furuta_f3/f3_s{0,1,2}_s24000.pth` | `furuta_bank_f3.npz`, `furuta_traj_dataset_f3.pt` | `furuta_robustness.md` |
| Unstable coupled MIMO, full-state (`mlp_*`, `asinh_*`) and output-feedback BMI (`sof*`, `c3*`) | `round2/mimo_unstable/` (`*_diff_*` diffusion, `*_reg_*` regression twin) | `data/round2/mimo_unstable/` | `mimo_unstable_family.md` |
| Output-feedback scaling m=2/3/4, banks 1k/3k/8k | `round2/mimo_unstable/d2m{m}n{N}_*` | `data/round2/mimo_sof_scaling/` (banks, held-out sets, datasets, evaluated candidates) | `mimo_unstable_family.md` §D2 |
| Output-feedback scaling m=2..5, final full recipe (96k x 64, 3 seeds), banks 8k (+16k for m=4) | `round2/mimo_sof_scaling_full/d3f{m}{n}_{diff,reg}_s*_s96000.pth` | `data/round2/mimo_sof_scaling/` (`d3f_m*.json` results) | `mimo_unstable_family.md` §D3 |
| Furuta online re-synthesis on the unified F3 model (episode logs) | — | `data/round2/furuta_resynth_b2/` | `furuta_online_resynthesis.md` §B2 |
|