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
Show transcriptHide transcript
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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
Up next
2:33MME-VLA robot memory counts a pause as work done and stops early
@ai-papers
3 days ago
2:16GPT-6 Astra fixed a small Qwen’s math by editing 1,023 of its weights
@ai-papers
5 days ago
2:42LoopVQ cuts looped models’ KV cache to half a bit and lifts Ouro’s throughput up to 6×
@ai-papers
1 week ago
2:35CUA-Master-27B is an open computer-use agent that trails only Claude Opus 5 and GPT-5.6-Sol on Agents' Last Exam
@ai-papers
1 week ago
2:37CaPtain trains Qwen3.5-4B to control a robot in code and beats several frontier-model agents
@ai-papers
1 week ago
2:45Trajectory2Gym data lifts a 4B Qwen3 agent past the stock 30B model on Tau2
@ai-papers
2 weeks ago
2:46“Wait” in reasoning models comes from training data, and RL almost never invents it
@ai-papers
3 weeks ago
2:08Looking Ahead makes Wan video models more physically plausible, with generation unchanged
@ai-papers
3 weeks ago
2:30OpenVLHarness gives vision models tools and a memory, and seven models each gain about 10 points
@ai-papers
3 weeks ago
2:413DCodeVerse makes the same model write better 3D code than Claude Code, Codex or Gemini CLI
@ai-papers
4 weeks ago
2:53Agent-Reflex trains an 8B agent to recover when its MCP tools fail, and it gets derailed half as often
@ai-papers
1 month ago
2:45EmbodiedCLBench: one practice task gives AI agents most of their gain, and more practice doesn’t add to it
@ai-papers
1 month ago















