Lightweight lite/ world-model variant: −80% params, quality preserved (proxy benchmark) - #62
Draft
coderXcode wants to merge 9 commits into
Conversation
A compact action-conditioned world model that runs on MacBook MPS within 28GB, used as the frozen baseline for benchmarked lightweight-variant experiments. Includes a deterministic raycaster ground-truth world, a conditional U-Net predictor, trainer, and an evaluator reporting fidelity (SSIM/PSNR), action-following, latency, and peak RAM. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
venv mode builds an empty virtualenv and network is disabled, so torch cannot be installed there; call the system framework python (torch+MPS) directly instead. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Wrap metrics as {schema_version, primary_metric, secondary_metrics,...}
so the benchmark result validates against MetricResult.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
…r models Smaller backbones need more steps than the baseline to reach the same action-following; a 250-step screening budget under-trained them and tripped the hard-constraint check. 600 gives a fair filter. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Reduced-budget screening under-trained smaller models and false-rejected near-winners on the hard-constraint check. Full-benchmark judging removes the bias; the model is small enough that a full run per experiment is cheap. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Wall-clock latency does not change when params are halved at this model scale (overhead-bound on MPS at 64x64). params is the metric that reflects the lightweight objective; latency kept as a <=9ms no-regression constraint. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Hypothesis: hyp-003 — Backbone depth/width reduction trades minimal quality for large RAM and latency gains Experiment: exp-015 — exp-013 + base_ch 24 + 2x training: On exp-013, cut width 32->24 and raise train_steps 900->1800: smaller model, more training to hold quality. Contract: 39fd7519fd9e47ccbb2867618d359652 v5 Baseline: e12ece4 (params=2095731.0) Validated: params 2095731.0 -> mean 4.241e+05 (n=3, stdev=0) Pre-ship: params=424091.0; constraints: ssim >= 0.7: pass, action_following >= 0.85: pass, latency_ms_per_frame <= 9.0: pass, peak_ram_mb <= 1500.0: pass Tests: no test command configured in the contract; no new tests authored (Claude-assisted test authoring arrives in Phase 1E) Changed: lite/config.yaml Generated by ResearchForge 0.1.0 from experiment records.
coderXcode
force-pushed
the
researchforge/backbone-depth-width-reduction-trades-mi
branch
from
July 22, 2026 06:43
9023fc8 to
0df7703
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Lightweight
lite/world-model variant — validated, −80% parametersThis draft adds a compact, MacBook-runnable (Apple MPS) action-conditioned
world model under
lite/, plus a frozen ResearchForge benchmark, and tunes itinto a lightweight variant.
Important scope note: the real
lingbot-worldmodel is Wan2.2 i2v-A14B(dual 14B experts + UMT5-XXL, 8-GPU reference) and cannot run losslessly in
28 GB. All numbers here are proxy-model measurements on a deterministic
raycaster world used to make the quality/size trade-off measurable on a
Mac — they are not the real A14B model's numbers.
Result (validated: 3/3 repeated full-benchmark runs, all constraints held)
What the change is
Three stacked, config-only levers (each independently benchmarked):
base_ch48→24Full evidence chain (baseline, every experiment incl. rejected ones, and the
validation spread) is in
.researchforge/reports/engineering-report.md.🤖 Generated with Claude Code