HyperAIHyperAI

Command Palette

Search for a command to run...

Jeva : des modèles System One fondés sur les premiers principes

Qi Li Xinchao Wang

Résumé

Les modèles System One permettent une prise de décision directe et à faible latence. Pourtant, la pratique récente consistant à prédire une réponse à partir d’un texte sérialisé ne respecte pas, en elle-même, la structure d’un problème de décision. Nous soutenons que les conceptions actuelles sous-estiment quatre exigences structurelles importantes. (i) sensible à l’absence de réponse : les modèles devraient distinguer l’absence de réponse de l’incertitude entre les candidats ; (ii) invariant par permutation : réordonner les alternatives ne devrait pas modifier la décision ; (iii) extensible en nombre d’options : l’augmentation des ensembles de candidats ne devrait pas exiger de redéfinir l’espace de sortie ; (iv) réutilisable en cache : le calcul indépendant de la requête portant sur des alternatives récurrentes devrait être réutilisable entre les requêtes. Nous présentons Jeva, une architecture nativement décisionnelle qui répond à ces exigences conjointement. Un encodeur partagé représente indépendamment la requête et chaque option, tandis qu’une pile décisionnelle compacte affine uniquement l’état de la requête face à une mémoire d’options immuable. L’agrégation symétrique produit des probabilités équivariantes par permutation ; une fonction de score partagée prend en charge des ensembles de candidats de taille variable. Une estimation distincte de la couverture distingue l’absence de réponse de la concurrence entre candidats, permettant des sorties nulles explicites. Les représentations d’options indépendantes de la requête et les projections de couches favorisent la réutilisation entre des ensembles nouvellement constitués. Ces mécanismes unifient les sorties Choice, Noul et Score. Dans les évaluations, Jeva combine une forte performance de décision typée avec une détection quasi parfaite de l’absence de réponse et aucune inversion de décision sous toutes les permutations testées. Il passe à l’échelle de milliers de candidats et réduit considérablement la latence grâce à la réutilisation entre requêtes sans changer les décisions.

One-sentence Summary

Researchers at the National University of Singapore introduce Jeva, a decision-native architecture that jointly enforces null-awareness, permutation invariance, option scalability, and cacheability through a shared encoder, immutable option memory, symmetric aggregation, and coverage estimation, thereby achieving strong typed-decision performance, near-perfect null detection, no permutation-induced decision flips, and reduced latency via cross-request reuse.

Key Contributions

  • Jeva is a decision-native architecture that jointly addresses null awareness, permutation invariance, option scalability, and cacheability by independently encoding the request and each option, refining a request state against immutable option memory, and applying symmetric aggregation and shared scoring.
  • The architecture unifies Choice, Noul, and Score outputs, with a separate coverage estimate distinguishing answer absence from candidate competition to enable explicit null outcomes.
  • With a Qwen3.5-4B backbone and one decision block, Jeva achieves strong typed-decision performance, near-perfect null detection, and no decision flips under tested permutations; it scales to 2,048 candidates and provides more than 2x median latency speedup for full-set scoring and more than 20x for cross-request reuse without changing decisions.

Introduction

Many language-model requests require compact typed decisions such as choosing a tool, identifying a document, or assigning a score. These decisions need candidate order symmetry, support for variable candidate counts, explicit null awareness, and reusable candidate representations to reduce latency. Prior open-source system one models commonly encode requests and candidates jointly as text, which couples candidate representation to request-specific computation, makes predictions sensitive to candidate order, restricts candidate counts through fixed output spaces, and does not clearly separate answer absence from uncertainty among alternatives. The authors introduce Jeva, a decision-native architecture that independently encodes the request and each option with a shared language encoder, then refines only the request state by reading an immutable option memory. Symmetric aggregation, shared candidate scoring, and a separate coverage gate provide permutation-equivariant probabilities, variable-size candidate support, explicit null detection, and reusable option computation.

Method

Independent Encoding and Collective Decisions

The authors leverage a shared language backbone to independently encode the request and each option, separating reusable semantic encoding from set-dependent judgment. A compact decision module refines only the request state against immutable option memory, enabling computation reuse while retaining set-dependent scoring.

As shown in the figure below:

A request comprises state xxx, question ι\iotaι, decision type ttt, and candidates O=(o1,...,oK)\mathcal{O} = (o_1, ..., o_K)O=(o1​,...,oK​). Inputs are formatted with two text templates: Tx(t,ι,x)T_x(t, \iota, x)Tx​(t,ι,x) combines the decision type, question, and state, while To(t,oi)T_o(t, o_i)To​(t,oi​) contains only the decision type and candidate text. Let EθE_\thetaEθ​ be the language backbone followed by last-non-padding-token pooling. Bias-free projections yield:

z0=PxEθ(Tx(t,ι,x)),ui=nrm(PoEθ(To(t,oi))),\begin{array}{l} z_{0} = P_{x} E_{\theta}(T_{x}(t, \iota, x)), \\ u_{i} = \mathrm{nrm}(P_{o} E_{\theta}(T_{o}(t, o_{i}))), \end{array}z0​=Px​Eθ​(Tx​(t,ι,x)),ui​=nrm(Po​Eθ​(To​(t,oi​))),​

where nrm is stabilized ℓ2\ell_2ℓ2​ normalization and z0,ui∈Rdz_0, u_i \in \mathbb{R}^dz0​,ui​∈Rd. The rows of U=[u1⊤;…;uK⊤]U = [u_1^\top; \ldots; u_K^\top]U=[u1⊤​;…;uK⊤​] stay fixed within a forward pass. Jeva conditions the scoring state on the candidate set, enabling preference reversals through compact per-request computation while keeping option representations reusable.

Candidate-Conditioned State Refinement

Each of LLL blocks reads UUU through HHH heads. For head hhh, queries are derived from the state, while keys and values are derived from the options. The candidate-weighted message is computed via attention:

aℓih=exp⁡((qℓh)⊤kℓih/τℓ)∑jexp⁡((qℓh)⊤kℓjh/τℓ),mℓ=WℓOConcat⁡h=1H(∑iaℓihvℓih).\begin{array}{l} a_{\ell i}^{h} = \frac{\exp((q_{\ell}^{h})^\top k_{\ell i}^{h} / \tau_{\ell})}{\sum_{j} \exp((q_{\ell}^{h})^\top k_{\ell j}^{h} / \tau_{\ell})}, \\ m_{\ell} = W_{\ell}^{O} \operatorname{Concat}_{h=1}^{H} \left(\sum_{i} a_{\ell i}^{h} v_{\ell i}^{h}\right). \end{array}aℓih​=∑j​exp((qℓh​)⊤kℓjh​/τℓ​)exp((qℓh​)⊤kℓih​/τℓ​)​,mℓ​=WℓO​Concath=1H​(∑i​aℓih​vℓih​).​

These weights yield an entropy-regularized allocation of candidate evidence for state refinement. The state undergoes two residual updates:

z~ℓ=zℓ+αℓmℓ,zℓ+1=z~ℓ+L−1/2Fℓ(RMSNorm(z~ℓ)),\begin{array}{c} \widetilde{z}_{\ell} = z_{\ell} + \alpha_{\ell} m_{\ell}, \\ z_{\ell+1} = \widetilde{z}_{\ell} + L^{-1/2} F_{\ell}(\mathrm{RMSNorm}(\widetilde{z}_{\ell})), \end{array}zℓ​=zℓ​+αℓ​mℓ​,zℓ+1​=zℓ​+L−1/2Fℓ​(RMSNorm(zℓ​)),​

where FℓF_\ellFℓ​ is a bias-free SwiGLU network and αℓ=τℓσ(γℓ)/L\alpha_\ell = \tau_\ell \sigma(\gamma_\ell) / \sqrt{L}αℓ​=τℓ​σ(γℓ​)/L​. Only zℓz_\ellzℓ​ evolves; keys and values depend solely on the unchanged options. This design allows a single block to uniquely select different options from the same set depending on the surrounding candidates, a property impossible for independent scorers.

Refer to the figure below:

Competition, Coverage, and Typed Outputs

Jeva computes final scores si=zL⊤uis_i = z_L^\top u_isi​=zL⊤​ui​ and conditional candidate probabilities q=softmax(s/τdec)q = \mathrm{softmax}(s / \tau_{\mathrm{dec}})q=softmax(s/τdec​). To form a full distribution ppp over candidates and null (⊥\bot⊥), the model separately estimates the probability ρ\rhoρ that at least one acceptable candidate is present:

ρ=σ(max⁡isi−bcovτcov),p⊥=1−ρ,pi=ρqi.\begin{array}{l} \rho = \sigma\left(\frac{\max_{i} s_{i} - b_{\mathrm{cov}}}{\tau_{\mathrm{cov}}}\right), \\ p_{\perp} = 1 - \rho, \qquad p_{i} = \rho q_{i}. \end{array}ρ=σ(τcov​maxi​si​−bcov​​),p⊥​=1−ρ,pi​=ρqi​.​

Here, bcovb_{\mathrm{cov}}bcov​ is a learned coverage threshold and τcov>0\tau_{\mathrm{cov}} > 0τcov​>0 is a learned temperature. This implements P(Y=i)=P(Y≠⊥)P(Y=i∣Y≠⊥)P(Y=i) = P(Y \neq \bot) P(Y=i \mid Y \neq \bot)P(Y=i)=P(Y=⊥)P(Y=i∣Y=⊥). Adding options below the existing maximum leaves ρ\rhoρ unchanged, distinguishing coverage estimation from candidate-score normalization.

Symmetry, Reuse, and Scaling

An atomic option memory stores both its semantic vector and projected views:

Mt(oi)=(ui,{kℓih,vℓih}ℓ,h).\mathcal{M}_{t}(o_{i}) = \left(u_{i}, \{k_{\ell i}^{h}, v_{\ell i}^{h}\}_{\ell, h}\right).Mt​(oi​)=(ui​,{kℓih​,vℓih​}ℓ,h​).

Requests gather these entries in their current candidate order. The set cache stores assembled candidate tuples, while the hybrid cache additionally reuses individual options across newly assembled sets. State encoding and decision refinement still execute for every request.

As shown in the figure below:

Permuting candidates permutes candidate probabilities but leaves zLz_LzL​, null probability, and typed outputs unchanged. Replacing option computations by identical cached entries also leaves all outputs unchanged. The decision cost is linear in KKK, and a bank of MMM options stores O(M(L+1)d)O(M(L+1)d)O(M(L+1)d) values.

Direct Supervised Learning

For normalized open-set targets (y⊥,y1,…,yK)(y_\perp, y_1, \ldots, y_K)(y⊥​,y1​,…,yK​), let cy=∑iyic_y = \sum_i y_icy​=∑i​yi​ and g=logit(ρ)g = \mathrm{logit}(\rho)g=logit(ρ). The authors minimize:

ℓ=BCEWithLogits(g,cy)−∑iyilog⁡qi.\ell = \text{BCEWithLogits}(g, c_y) - \sum_{i} y_i \log q_i.ℓ=BCEWithLogits(g,cy​)−i∑​yi​logqi​.

This primary loss is joint cross-entropy on ppp: all-null examples supervise availability without imposing a spurious candidate ranking. The model uses a Qwen3.5-4B backbone with pooled width 2560, decision width d=768d=768d=768, L=1L=1L=1, H=8H=8H=8, and f=3072f=3072f=3072. Training employs rank-64 LoRA with both projections and the decision head, using a curriculum spanning typed decisions, intent selection, verification, reading and reasoning, and large-candidate retrieval. LoRA is merged for serving, and memories are prepared for the resulting fixed model.

Experiment

The experiments assess decision accuracy, null detection, candidate scalability, batching efficiency, and cache reuse across Typed Decisions, BANKING77, MASSIVE, and Mind2Web slices. Jeva demonstrates strong full-coverage decision quality, near-perfect order invariance, and effective detection of unsupported requests. It also scales to very large candidate sets with one-pass scoring and length-aware batching, while exact-set and hybrid caching preserve decisions and substantially reduce latency. Overall, the results validate Jeva’s robustness, coverage, and efficiency advantages over baselines.

With a fixed model, type, and encoding policy, an initially empty option bank starts with no reusable candidate work, then reuses prepared options as new requests overlap with prior sets. Whole-set lookups consistently miss, while per-option reuse grows across overlapping sets and state encoding plus decisions remain fresh. Reuse rises from no hits on the first set to partial hits on a partially overlapping set and full hits on a fully covered set. Only previously prepared options are reused; new options are prepared on demand without eviction. Whole-set cache lookups miss for every new request set, while state encoding and decision outputs remain fresh.

Jeva records the highest first-argmax accuracy on seven of eight full-coverage decision slices, leading BANKING77 and MASSIVE at both candidate counts and all Mind2Web splits. The exception is typed decisions, where a competing model ranks higher. Smaller models trail by wide margins on BANKING77, MASSIVE, and Mind2Web. Jeva leads seven of eight full-coverage accuracy slices, including BANKING77 and MASSIVE at both candidate counts and all Mind2Web splits. Typed decisions are the only full-coverage slice where Jeva is not the top model. Several smaller models show large accuracy drops on BANKING77, MASSIVE, and Mind2Web.

Jeva shows the strongest typed fidelity and order stability among the evaluated models. It achieves the lowest Noul and Score error metrics against soft targets and records zero decision flips across candidate order changes, while other models show higher error and non-zero flip rates. The remaining baselines vary substantially, with smaller variants and Re-Flex showing the largest errors. Jeva records the lowest Noul Brier and MAE against soft probability targets, along with the lowest Score MAE and a Score RMSE close to InternDecision-4B. Jeva has zero choice flips across original, reversed, and seeded random orders, with mean total variation near zero and identical original and all-order accuracy. Jev-1.13.0, InternDecision-4B, and JevK5 show order sensitivity, with flip rates between 17 and 26 percent and reduced all-order accuracy. Among the remaining baselines, InternDecision-0.8B has the highest flip rate, while Re-Flex-4B has the largest Noul and Score errors.

Jeva and JevK5 maintain full valid coverage across all candidate-count cohorts, while other approaches quickly become unsupported as the candidate set grows. Jeva also covers all requests with higher aggregate accuracy than recursive JevK5, and length-aware batching sharply reduces p50 latency at larger candidate counts without changing decisions. Jeva and JevK5 are the only methods with full valid coverage across every tested candidate-count cohort. KEV-BF16 supports the smallest cohorts but drops to zero coverage beyond 128 candidates, while InternDecision variants cover only a subset of requests at 32 candidates and none beyond. ReFlex and SemiF show no valid coverage in any tested cohort. Jeva attains higher aggregate accuracy than recursive JevK5 while scoring complete candidate sets in one decision-head pass. Length-aware batching preserves accuracy and order while cutting p50 latency dramatically at large candidate counts, with roughly an eightfold reduction at the largest cohort. In candidate-set expansion tests, Jeva remains correct at every size on more queries than JevK5 and CLM, and its all-size accuracy exceeds JevK5 by a notable margin.

The experiments evaluate option-bank reuse, decision accuracy, typed fidelity and order stability, and scaling behavior across candidate counts. A reuse test shows whole-set cache lookups always miss, while per-option reuse increases from no hits on the first set to full hits on covered sets, with only previously prepared options reused and decisions remaining fresh. In benchmark and fidelity tests, Jeva leads seven of eight full-coverage accuracy slices, records zero decision flips under order changes, and shows the lowest soft-target errors, while smaller baselines are less accurate or more order-sensitive. Coverage and latency experiments find Jeva and JevK5 are the only methods with full valid coverage across all candidate-count cohorts, with Jeva more accurate than recursive JevK5 and length-aware batching reducing latency without changing decisions.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp