hku_diffusion_dllm / basecode /experiments /reproduction_matrix.md
Ouzhang's picture
Add files using upload-large-folder tool
13c5606 verified
|
Raw
History Blame Contribute Delete
2.29 kB
# Reproduction Matrix
This matrix is the execution checklist for source-code reproduction first, then
basecode reproduction with matched metrics.
| Paper / route | Source code path | Required assets | Source metrics to reproduce | Basecode hook | Status |
| --- | --- | --- | --- | --- | --- |
| LLaDA/Dream-style block diffusion | `reference/code/Nemotron-Labs-Diffusion`, `reference/code/mbd-lms` | `llada2-mini`, `dream-v0-base-7b`, GSM8K/MATH/HumanEval/MBPP | accuracy, NFE, latency, tokens per forward | `ConfidenceRemaskSampler`, `MultiBlockSampler` | planned |
| MBD-LMs | `reference/code/mbd-lms` | `llada2-mini`, math/code datasets | multi-block speed/quality tradeoff | `MultiBlockSampler` | planned |
| DMax | `reference/code/DMax` | `dmax-16b`, GSM8K/MATH/code evals | aggressive parallel decoding accuracy and speed | `DMaxSampler` | planned |
| TAD / trajectory self-distillation | `reference/code/TAD`, `reference/code/T3D` | trajectory datasets, distilled checkpoints | few-step quality retention | `DMaxSampler` with distilled adapter | planned |
| Fast-dLLM | `reference/code/Fast-dLLM` | discrete dLLM checkpoint | KV-cache speedup, parallel decode latency | adapter cache API | planned |
| DFlash / PRESTO | `reference/code/dflash`, Nemotron references | draft model + AR target | acceptance length, speedup, target quality | `SpeculativeSampler` | planned |
| dMoE | `reference/code/dMoE` | `dmoe-16b` | expert/block routing quality and speed | expert adapter extension | planned |
| ELF / continuous flows | `reference/code/ELF` | ELF checkpoints, OpenWebText/LM1B | perplexity, MAUVE/diversity, generation speed | separate continuous adapter | planned |
| FLM/FMLM one-step | `reference/code/flm` | manual checkpoints, OpenWebText/LM1B | unconditional generation PPL/time | separate continuous adapter | planned |
Run order:
1. Confirm remote assets with `python3 scripts/download_assets.py --dry-run`.
2. Download `core` tier only.
3. Run each source repository's smallest documented eval.
4. Normalize metrics into `runs/source_reproductions/<paper_id>/metrics.json`.
5. Implement/attach matching adapter in `basecode/blockdiff/adapters/`.
6. Run `basecode` evals into `runs/basecode/<method_id>/metrics.json`.
7. Compare source vs basecode metrics in one table.