Skip to content
PraveenAShuklaPublic

About

Rank-6 DFlash K=15 solution for the AdaptFM Efficient Qwen Challenge at ICML 2026

Resources

Contributing

Stars

5 stars

Watchers

0 watching

Forks

Latest commit

 

History

8 Commits

Folders and files

Repository files navigation

DFlash K=15 for Qwen3.5-4B on a single A10G

Efficient LLM Challenge: sixth place ICML 2026 AdaptFM workshop Docker pulls Licence: MIT

Team AFM-qd39x2e6, sixth place in the Efficient LLM Challenge, AdaptFM workshop at ICML 2026. Around 176 teams entered; 41 landed a valid submission.

5.366× mean speedup, all three quality gates passed.

Prompt Baseline Ours Gate Required Ours
64 tokens 2,582 ms 345 ms MMLU-Pro 0.621 0.661
2,048 tokens 5,441 ms 881 ms IFEval 0.814 0.845
8,192 tokens 6,576 ms 2,684 ms GPQA-Diamond 0.630 0.667

A 4-bit AWQ Qwen3.5-4B target, the public z-lab/Qwen3.5-4B-DFlash drafter at a fixed K=15, and vLLM 0.20.1 with one dtype patch. No new training.

submission/ the configuration that ranked
docs/method.md why K=15, the dtype patch, how to rebuild
docs/submission-history.md what blocked the fast configurations
docs/what-we-tried.md every build, and what wasted our time
docs/future-work.md open questions

The short version

Speculative decoding on this model corrupts the recurrent state when drafts are rejected, so generation runs away and blows the 60-second per-sample limit on the quality evaluation. Our fastest entries were fast and wrong. The one that passed did so by serving the two evaluations from separate backends; it reached second overall and was removed as benchmark routing. Five later submissions never started at all.

The entry that ranked is the slow one that booted.

The winning entry used the same recipe we ended on, a 4-bit target with DFlash at K=15. The gap between sixth and first looks like quantization detail and per-step serving overhead rather than the choice of algorithm.

What transfers

Most of what cost us time was not specific to the challenge.

  • Rejected drafts can corrupt a recurrent state. Most of Qwen3.5-4B's layers are Gated DeltaNet, and in vLLM a rejected speculative draft left their state corrupted: generation ran away, fluent, long and wrong. Tracked upstream as vllm#39273 and vllm#35288. (more)
  • Build for the card you run on. A build carrying sm_80 and sm_89 kernels loads on an A10G (sm_86), then fails at the first kernel launch. It cost us more time than any algorithmic decision; the fix is TORCH_CUDA_ARCH_LIST=8.6. (more)
  • Test the artifact you upload, not the image you built it from. Flattening with docker export | docker import got us under the 20 GB limit and broke docker run <image> serve; five submissions never answered the health check. (more)
  • Name images after their configuration. Two of our tags named for DFlash contained n-gram configurations. Record the digest when you build. (more)

Reproducing

Most of our images are public on Docker Hub; what-we-tried.md shows how to read any one's configuration without downloading it.

docker pull 11noel11/adaptfm-submission:dflash-k15

That build is DFlash K=15 from before the chat-template change. Latency and quality figures here are the official results.

Dependencies and licence

No model weights are stored here; Qwen3.5-4B, the AWQ quantization and the DFlash drafter each carry their own terms (DEPENDENCIES.md). MIT (LICENSE). Contributions welcome (CONTRIBUTING.md).

Thanks to the AdaptFM organizers and the Amazon Edge AI team, and to the teams behind Qwen3.5, DFlash, vLLM, AWQ and Marlin.

Citation

Praveen A. Shukla and Noel Thomas, Mohamed bin Zayed University of Artificial Intelligence.

@misc{shukla2026adaptfm,
  author = {Shukla, Praveen A. and Thomas, Noel},
  title  = {{DFlash K=15 for Qwen3.5-4B on a single A10G}},
  year   = {2026},
  note   = {Sixth place, Efficient LLM Challenge, AdaptFM Workshop, ICML 2026},
  url    = {https://github.com/PraveenAShukla/adaptfm}
}

About

Rank-6 DFlash K=15 solution for the AdaptFM Efficient Qwen Challenge at ICML 2026

Resources

Contributing

Stars

5 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages