|
| 1 | +# Pixelized gradient-sampler experiment — can MultiStartAdam/ADABelief/Lion work for pix? |
| 2 | + |
| 3 | +Type: experiment |
| 4 | +Target: autolens_workspace_developer |
| 5 | +Repos: |
| 6 | +- @autolens_workspace_developer |
| 7 | +Difficulty: large |
| 8 | +Autonomy: supervised |
| 9 | +Priority: normal |
| 10 | +Status: issued |
| 11 | + |
| 12 | +Research question: do the newly-promoted multi-start gradient MAP optimizers |
| 13 | +(af.MultiStartAdam / MultiStartADABelief / MultiStartLion, Fit#1369) work on a |
| 14 | +PIXELIZED source reconstruction, not just the MGE likelihood the benchmark used? |
| 15 | + |
| 16 | +Setup (searches_minimal/, extending the MGE benchmark harness): |
| 17 | +- Model = SLaM SOURCE_PIX[1] style: lens MGE linear light with FIXED non-linear |
| 18 | + geometry; lens mass (Isothermal + ExternalShear) FREE; source = |
| 19 | + RectangularSplineAdaptImage (differentiable spline mesh) + adaptive |
| 20 | + regularization (al.reg.Adapt); regularization coefficient FREE. Free non-linear |
| 21 | + params ~= mass + shear + reg (~7-D). Adapt image bootstrapped from a quick |
| 22 | + RectangularAdaptDensity+Constant fit (no adapt image needed), mirroring SLaM. |
| 23 | +- FD feasibility gate FIRST (probe_grad_pix.py): reverse-mode jax.grad of the |
| 24 | + spline-pixelized log-evidence, FD-cross-checked. If FAIL_FD_MISMATCH, that IS |
| 25 | + the answer — stop, report, no A100 burn. |
| 26 | +- Samplers: af.MultiStartAdam/ADABelief/Lion + af.Nautilus baseline. |
| 27 | +- Runs: local CPU smoke, then A100 on RAL (euclid_jump pipeline). |
| 28 | +- Deliverable: findings doc (do gradient MAP optimizers recover the mass basin |
| 29 | + with a pixelized source vs Nautilus?). |
| 30 | + |
| 31 | +Decisions (human, 2026-07-14): SplineAdaptImage + adaptive reg; mass+reg free / |
| 32 | +light fixed. Checkpoint after the FD probe before samplers/A100. |
| 33 | + |
| 34 | +<!-- research spun off the multi-start gradient promotion; grad harness = searches_minimal --> |
0 commit comments