Narrow Decision Models: A System-1 Architecture for Calibrated, Typed Decisions
Organizations: Softplan
Abstract
System One models, introduced with TypeSafe's Jev, answer typed questions about a state with calibrated probabilities instead of generated text: choosing among options described in natural language (Choice), placing the state on an ordered scale (Score), or judging a yes/no statement (Noul). Jev is general-purpose: one set of weights, served as an API and adapted to a domain only through the request. We study the other end of the design space. A Narrow Decision Model (NDM) offers the same interface but is trained for the decisions of one organization, is small enough to run on its own infrastructure, and has an open architecture whose properties can be stated and tested. The state can be any text or structured record about which decisions are made; our running case is the routing of court communications. The architecture serializes a request as a prefix tree (the state, then each question, then each option) and runs a bidirectional encoder under a block attention mask with logical positions, so that each answer is, by construction, invariant to the order of its options and independent of the other questions asked. Routing through an organization chart is a composition of Choice decisions with an explicit "none of these" option. We describe training with proper scoring rules through a curriculum, calibration and abstention, decisions composed by rubrics, and a protocol that accepts or rejects the mechanism before held-out data is opened. A minimal validation on public and fictitious data shows that the invariances hold to rounding in a trained model, that a small instance routes on organization charts it has never seen well above lexical retrieval, and that, at equal compute, the block architecture reaches a lower loss than a pairwise cross-encoder. A first instance trained on real legal documents agrees with the LLM classifier in production on 91% of held-out documents.
Figures & tables
| NDM -Block | NDM -Pair | Decoder scoring | |
|---|---|---|---|
| Forward passes per request | a | ||
| Times the state is encoded | a | ||
| Options compared with each other | yes (set head) | no | no |
| Option-order invariance | by construction | by construction | no |
| Question isolation | by construction | by construction | prompt-dependent |
| Trained for calibration | yes | yes | no |
| float32 | float64 | ||||
| test | sdpa | eager | sdpa | eager | negative control |
| P1: permute the options | 2.6e-6 | 4.7e-6 | 2.3e-9 | 1.6e-7 | sequential positions: 3.7 |
| P2: question alone vs. batched | 3.8e-6 | 3.5e-6 | 6.2e-9 | 3.2e-7 | no mask: 14.3 |
| P2: state with and without questions | 0 | 0 | 0 | 0 | no mask: 2.0 |
| P2: change a sibling question | 0 | 0 | 0 | 0 | no mask: 3.9 |
| P2: order of the questions | 2.4e-6 | 2.7e-6 | 1.9e-8 | 1.3e-7 | sequential positions: 6.2 |
| dataset (labels) | ||||||
|---|---|---|---|---|---|---|
| MASSIVE intents | 0.956 | 0.924 | 0.878 | 0.811 | 0.740 | — |
| Banking77 | 0.953 | 0.913 | 0.860 | 0.801 | — | 0.701 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Component | Value |
|---|---|
| Backbone | mmBERT-base: 22 layers (8 global, 14 local, window logical positions), ; frozen embeddings (196.6 M), trained body (110.3 M) |
| Markers | 9 used and 7 reserved vocabulary entries; offsets , initialized to the mean embedding of an equivalent phrase |
| Question and option vectors | on plus a type embedding; |
| Set head | 2 pre-norm layers, width 384, 6 heads, feed-forward 1,536; output projections initialized at zero |
| Relative scorer | (1,152) linear (384) GELU one linear reading per type |
| Adequacy | (2,304) linear (384) GELU linear |
LLM Decision-Making
LLM decision-making studies how language models choose, act, seek information, and allocate resources under goals, uncertainty, constraints, and social pressures.
See all 229 papers