NestOne cuts a Qwen3.5 decision model to 24 layers at nearly full accuracy

1 day ago

NestOne trains one Qwen3.5-4B decision model, the kind that routes requests, picks tools and flags unsafe prompts, to answer after 12, 16, 24 or 32 layers. Its 24-layer exit takes 34.7 ms instead of 46.5 ms and stays 0.6 points below the full model, and a hidden-state gate stops easy decisions earlier. Paper: https://arxiv.org/abs/2611.16321 Code: https://github.com/minhkks/NestOne

Ask about this presentation

Chapters

  1. 0:00NestOne
  2. 0:20Every decision pays for 32 layers
  3. 0:35The decision forms mid-depth
  4. 0:50Cutting the last layers
  5. 1:04Smaller models are weaker copies
  6. 1:19Nested training
  7. 1:34One checkpoint against four models
  8. 1:48The stop gate
  9. 2:01Savings depend on traffic
  10. 2:17Limits
  11. 2:35Where to start
Show transcript
  1. NestOne trains one Qwen3.5 decision model that can answer after 12, 16, 24 or all 32 layers. At 24 layers it saves about a quarter of the time per decision and stays within a point of the full model. Minh-Anh Nguyen built it at Nanyang Technological University.

    Slide text
    AI Papers

    NestOne cuts a Qwen3.5 decision model to 24 layers at nearly full accuracy

    Minh-Anh Nguyen · Nanyang Technological University

    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · submitted 12 Nov 2026
  2. Agents make many small choices: which tool to call, where to route a request, whether a prompt is unsafe. A System One model answers each in one pass, and every answer still runs all 32 layers.

    Slide text
    01 02 03 04 Where the decision is

    A decision model still runs all 32 layers to pick one option

    46.5 ms per decision for the full 32 layers of Qwen3.5-4B, at batch size one
    1 pass routing a request, choosing a tool, flagging an unsafe prompt, naming an intent
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · §1, §2, §5 setup
  3. Probe each layer for the right option, and accuracy climbs through the middle of the network, from layer 15 to layer 23. After layer 23 it goes flat. The last layers add no decision information a probe can find.

    Slide text
    01 02 03 04 Where the decision is

    The answer forms in layers 15 to 23, then stops changing

    Linear probe, logit lens and cumulative decision share by layer for Qwen3.5-4B and Intern-Decision-4B, with layers 15 to 23 shaded
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Figure 2, §3.1 · seven decision benchmarks, macro average
  4. Take a released decision model, Intern-Decision-4B, and delete layers 24 to 29 with no retraining. Accuracy doesn't move. Start the cut at layer 22 instead, and it loses almost 12 points.

    Slide text
    01 02 03 04 Where the decision is

    Cut layers 24 to 29 from Intern-Decision-4B and accuracy holds

    Uncut
    90.05
    Remove 24–29
    90.1
    Remove 24–31
    89.4
    Remove 22–31
    78.5

    Accuracy (%) without retraining · removing layers 15–23 from full32 costs 57.8 points

    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Table 1, §3.1
  5. Training a 12 or 16 layer model from scratch gives you a weaker version of the full one. Its own wins sit barely above the noise between training seeds, and it loses three to four decisions for every one it uniquely gets right.

    Slide text
    01 02 03 04 Where the decision is

    A smaller model trained on its own learns a weaker copy of the full one

    89–91% of what the 12- and 16-layer models get right, the full model also gets right
    3–4× as many decisions lost as uniquely solved
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Table 2, §3.2 · M16 and M12 trained separately on the same data
  6. So NestOne trains every depth together. The shorter models are prefixes of the full model. They share its last two layers and its decision head, and they learn from the full model's own answers as well as the labels.

    Slide text
    01 02 03 04 One checkpoint

    NestOne trains the 12-, 16- and 24-layer models inside the full one

    Nested training: exits after layers 9, 13, 21 and 29 share the last two layers and head; the 32-layer output teaches the shorter exits
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Figure 3a, §4.1–4.2
  7. One training run, about 80 minutes on a single H200, replaces four. The deep exits tie the separate models. The shallow ones come out ahead, and clearly so at 12 layers, by about three points.

    Slide text
    01 02 03 04 One checkpoint

    One checkpoint ties separate models at 24 and 32 layers and beats them below

    LayersSeparateNestOneChange
    3284.384.0−0.3
    2483.583.7+0.2
    1672.874.7+1.8
    1261.564.4+2.9

    Benchmark average (%), 3 seeds · only the 12-layer gain is significant · one 80-min run on one H200 against 94 min for four

    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Table 3, §5.1 · Qwen3.5-4B, seven benchmarks
  8. At each exit, a gate reads the hidden state and decides whether the answer is ready. Checking costs one dot product. Stopping at 24 layers takes about 35 milliseconds, against about 47 for the full model.

    Slide text
    01 02 03 04 Choosing depth

    A small gate at each exit decides whether to answer there

    Adaptive inference: gates after layers 9, 13 and 21 can skip to the shared head; 12, 16, 24 and 32 layers take 18.4, 23.8, 34.7 and 46.5 ms
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Figure 3b, §4.3, §5 setup · latency at batch size one
  9. How much you save depends on your traffic. On easy decisions, like news topics and jailbreak checks, the gate averages under 18 layers. Hard ones still run all 24, and accuracy on the four largest benchmarks holds.

    Slide text
    01 02 03 04 Choosing depth

    Easy traffic saves the most layers, and accuracy on the big benchmarks holds

    Left: benchmark average against layers run for separate, nested and gated models. Right: mean layers run falls from 20.7 to 17.5 as easy traffic rises from 50% to 100%
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Figure 1, Table 4, §5.2 · gate over exits 9, 13 and 21 · easy: AGNews, WildJailbreak
  10. Every test uses one model, Qwen3.5-4B, timed at batch size one. It scores about 84 percent, below Intern-Decision-4B at 90. And the gate's threshold shifts between data splits, so tune it on traffic like yours.

    Slide text
    01 02 03 04 Limits

    Tested on one 4B model, and the gate has to be tuned on your traffic

    One backbone Qwen3.5-4B only · larger models are future work
    Below the best 4B 84.3% retrained full model against 90.0% for Intern-Decision-4B, trained on more data
    Gate threshold shifted from 0.80 to 0.92 across data splits
    Small benchmarks three of them, 430 of 12,351 decisions, carry the 1.6 to 1.7 point average drop
    Latency batch size one · end-to-end serving untested · preprint, work in progress
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · §3, §5.2, §6
  11. If I served a fine-tuned decision model, I'd try the 24 layer exit first: a quarter less time, under a point lost, and no gate to tune. The code is on GitHub as NestOne.

    Slide text

    For a fine-tuned decision model, try the 24-layer exit first

    Paper arxiv.org/abs/2611.16321
    Code github.com/minhkks/NestOne
    License Apache-2.0
    Nguyen · From LLMs to Efficient System One Models via Adaptive Computation · arXiv 2611.16321v1 · Table 3, §5.1

What happened in AI today, in your inbox

One short email each evening. Read it and you’re caught up.

Built with ❤️ in New York