---
title: Neural Additive Experts (NAEs)
url: https://www.emergentmind.com/topics/neural-additive-experts-naes
type: topic
---

# Neural Additive Experts (NAEs)

Neural Additive Experts (NAEs) are a neural architecture that preserves an additive decomposition at the level of feature contributions while relaxing the strict additivity of generalized additive models through context-gated mixtures of per-feature experts. In the formulation introduced in "Neural Additive Experts: Context-Gated Experts for Controllable Model Additivity" [2602.10585], the prediction remains a sum of feature-wise terms, but each term is produced by multiple specialized experts whose mixture weights can depend on all input features. This construction is intended to retain feature-level attribution while capturing interaction structure that standard GAMs and NAMs cannot represent directly.

## 1. Definition and conceptual position

The motivating comparison is with the classical GAM form
\[
f(x) = \omega_0 + \sum_{i=1}^n f_i(x_i),
\]
which yields clear feature-level interpretability but cannot represent genuinely context-dependent effects. A standard extension is the GA\(^2\)M form
\[
f(x) = \omega_0 + \sum_{i=1}^n f_i(x_i) + \sum_{1\le i < j \le n} f_{ij}(x_i,x_j),
\]
which improves expressivity at the cost of explanation complexity growing to \(O(n^2)\) components. NAEs are designed to occupy the intermediate regime: they keep a top-level sum over features, but replace each single-feature function by a context-gated mixture of experts, thereby relaxing rigid additivity without abandoning a per-feature decomposition [2602.10585].

This design differs from a conventional mixture-of-experts architecture in a precise way. In NAEs, experts are organized per feature rather than as global subnetworks over the whole input, and the final prediction is still written as a sum of feature contributions. The resulting model is additive in its outputs \(o_i(x)\), but those outputs need not be additive in the raw inputs because the gating for feature \(i\) can depend on \(x_1,\dots,x_n\) [2602.10585].

A broader technical reading of the term has emerged around related additive decompositions. Several works interpret concept-level, feature-level, cluster-level, or neuron-level decompositions as NAE-like even when they do not use the term explicitly; examples include CAT’s concept encoders plus TaylorNet, CRISP-NAM’s feature-wise subnetworks for competing risks, NeurCAM’s additive clustering heads with selection gates, and MoNE’s decomposition of FFN experts into additive neuron experts [2406.17931] [2505.21360] [2408.13361] [2510.05781].

## 2. Architectural formulation

NAEs always predict through a feature-additive outer form:
\[
\hat{y}(x) = \omega_0 + \sum_{i=1}^n o_i(x).
\]
The difference from a NAM lies in how each \(o_i(x)\) is computed. First, each feature is encoded by a feature-specific encoder
\[
\mathcal{E}_i : \mathcal{X}_i \to \mathbb{R}^d, \quad z_i = \mathcal{E}_i(x_i).
\]
In the reported experiments, \(\mathcal{E}_i\) is a small MLP. For feature \(i\), the model instantiates \(K\) experts
\[
o_{ik} = g_{ik}(z_i) = g_{ik}\big(\mathcal{E}_i(x_i)\big), \quad g_{ik} : \mathbb{R}^d \to \mathbb{R},
\]
with \(g_{ik}\) typically implemented as a single linear layer [2602.10585].

The gating mechanism is the non-additive part. For feature \(j\), NAE computes expert scores
\[
\varphi_j = \mu_j + \sum_{i=1}^n \mathcal{A}_{ij}^\top \mathcal{E}_i(x_i),
\]
where \(\mu_j \in \mathbb{R}^K\) is a bias vector and \(\mathcal{A}_{ij} \in \mathbb{R}^{d \times K}\) is a routing matrix. After adding a sparsity-inducing mask \(M_j\), a softmax produces mixture weights
\[
r_{jk} = \frac{\exp\big(\varphi_j[k] + M_j[k]\big)}
{\sum_{l=1}^K \exp\big(\varphi_j[l] + M_j[l]\big)}.
\]
The contribution of feature \(j\) is then
\[
o_j(x) = \sum_{k=1}^K r_{jk}(x)\, g_{jk}\big(\mathcal{E}_j(x_j)\big),
\]
and the full predictor becomes
\[
\hat{y}(x) = \omega_0 + \sum_{j=1}^n \sum_{k=1}^K r_{jk}(x)\, g_{jk}\big(\mathcal{E}_j(x_j)\big).
\]
Thus, each expert remains feature-local in content, but expert selection is context-sensitive [2602.10585].

Two variants are described. **NAE-D** uses diagonal routing, setting \(\mathcal{A}_{ij}=0\) for \(i\ne j\), so each feature’s expert relevance depends only on its own encoder; this recovers a strictly additive regime. **NAE-E** alters the activation pattern so that activated experts are weighted evenly, producing a more discrete combinatorial gating behavior [2602.10585].

## 3. Additivity control, regularization, and explanation semantics

A central contribution of NAEs is explicit control over the degree of additivity. The paper defines the additivity metric
\[
\text{Additivity}
= \frac{1}{n}\sum_{i=1}^n
\frac{\operatorname{Var}\bigl(\mathbb{E}[o_i \mid x_i]\bigr) + \delta}
{\operatorname{Var}(o_i) + \delta},
\]
with \(\delta>0\) for numerical stability. If \(o_i\) is purely a function of \(x_i\), the ratio is \(1\); if \(o_i\) varies strongly with other features, the ratio falls below \(1\) [2602.10585].

Training minimizes the task loss plus an expert-variation penalty:
\[
\min_{\theta}\; \frac{1}{N}\sum_{t=1}^N \mathcal{L}\big(y^t, \hat{y}^t\big)
\;+\; \frac{\lambda}{n N K}\sum_{t=1}^N \sum_{i=1}^n \sum_{k=1}^K
\Bigg(o_{ik}^t - \frac{1}{K}\sum_{l=1}^K o_{il}^t\Bigg)^2.
\]
This second term penalizes the within-feature variance of expert outputs. When \(\lambda\) is large, all experts for a feature are pushed toward the same output, making the gate irrelevant and driving the model toward a GAM-like limit [2602.10585].

The interpretability mechanism remains feature-level. Because the predictor is always written as \(\hat{y}=\omega_0+\sum_i o_i(x)\), each feature has a scalar contribution on every example. In addition, the model defines intrinsic bounds
\[
\text{upper}(o_i|x_i)=\max_k g_{ik}\big(\mathcal{E}_i(x_i)\big),\qquad
\text{lower}(o_i|x_i)=\min_k g_{ik}\big(\mathcal{E}_i(x_i)\big),
\]
which provide a context-agnostic envelope of possible contributions for a given feature value. This differs from GA\(^2\)M-style interaction explanations, where a single feature’s effect is dispersed across many pairwise surfaces [2602.10585].

The empirical effect of \(\lambda\) is illustrated on the Housing dataset. With \(\lambda=0\), Additivity is approximately \(0.52\) and RMSE is approximately \(0.451\); with \(\lambda=0.1\), Additivity is approximately \(0.56\) and RMSE remains approximately \(0.451\); with \(\lambda=1\), Additivity is approximately \(0.64\) and RMSE rises to approximately \(0.458\); with \(\lambda=10\), Additivity is approximately \(0.90\) and RMSE is approximately \(0.515\); with \(\lambda=100\), Additivity reaches \(1.0\) and RMSE is approximately \(0.582\). The reported behavior is therefore a continuous trade-off between flexibility and strict additivity [2602.10585].

## 4. Expressivity and theoretical properties

The theoretical analysis places NAEs strictly beyond standard GAMs. First, any \(f \in GAM\) can be represented exactly by an NAE with \(K=1\): one sets \(r_{i1}(x)\equiv 1\) and chooses \(g_{i1}\circ \mathcal{E}_i=f_i\). This establishes the containment
\[
GAM \subset NAE(1) \subset NAE(K), \quad K\ge 1.
\]
Second, the paper proves approximate containment of GA\(^2\)M. The argument uses finite separable approximations
\[
f_{ij}(x_i,x_j) \approx \sum_{m=1}^{M_{ij}} u_{ijm}(x_i)v_{ijm}(x_j),
\]
followed by a two-expert construction that represents each separable term through context-dependent gating. In this sense, NAEs can approximate additive-plus-pairwise-interaction models to arbitrary precision [2602.10585].

A further theorem formalizes the additive limit induced by the variation penalty. Let
\[
\mathsf{Pen}(\theta)
= \frac{1}{n N K}
\sum_{t,i,k}\left(o_{ik}(x^t)-\frac{1}{K}\sum_{\ell} o_{i\ell}(x^t)\right)^2.
\]
As \(\lambda\to\infty\), \(\mathsf{Pen}(\theta)\to 0\), expert outputs within each feature collapse, the gate ceases to matter, and any limit point of minimizers is a GAM. The additivity metric is correspondingly nondecreasing in \(\lambda\) and converges to \(1\) [2602.10585].

This theoretical boundary has consequences for formal explanation. "Provably Explaining Neural Additive Models" shows that for strict NAMs one can obtain provably cardinally-minimal sufficient explanations with \(O(\log n)\) verification queries to the full model after preprocessing of univariate components. The same work also states that this machinery does not apply directly to additive-expert architectures with multivariate global gating, because the effective contribution of a feature is no longer a purely univariate function of that feature alone [2602.17530]. A plausible implication is that NAEs gain expressivity partly by moving outside the structural regime that currently supports the strongest formal minimality guarantees.

## 5. Empirical behavior

On synthetic data, NAEs are reported to match NAMs when the ground truth is purely additive and to exceed them when the target is multimodal or context-dependent. In a unimodal additive case, NAM and NAE both recover the correct shape for \(x_1\). In a multimodal case where the conditional relation between \(x_1\) and \(y\) flips sign according to a binary \(x_2\), NAM collapses to a near-linear shape while NAE reconstructs the oscillatory pattern through multiple experts and context gating. On this multimodal task, the reported additivity scores are \(0.597\), \(0.709\), and \(1.000\) for \(\lambda=0.1\), \(1\), and \(10\), respectively [2602.10585].

A second synthetic interaction study uses
\[
y = 2\sin(\pi x_1)\cos(\pi x_2)+0.5x_1^2+0.5x_2^2.
\]
For this target, NAE achieves RMSE approximately \(0.13\) whereas NAM achieves RMSE approximately \(1.01\), illustrating the effect of context-dependent expert selection on explicitly non-additive structure [2602.10585].

On real-world tabular datasets, the reported results compare NAE primarily against feature-explaining baselines such as NAM, EBM, and NODE-GAM.

| Dataset | Metric | Selected results |
|---|---|---|
| Housing | RMSE \(\downarrow\) | NAE 0.451; NAM 0.572; EBM 0.559; NODE-GAM 0.558 |
| MIMIC-II | AUC \(\uparrow\) | NAE 0.847; NAM 0.834; EBM 0.835; NODE-GAM 0.832 |
| MIMIC-III | AUC \(\uparrow\) | NAE 0.825; NAM 0.813; EBM 0.809; NODE-GAM 0.814 |
| Income | AUC \(\uparrow\) | NAE 0.927; NAM 0.910; EBM 0.927; NODE-GAM 0.927 |
| Credit | AUC \(\uparrow\) | NAE 0.982; NAM 0.977; EBM 0.974; NODE-GAM 0.981 |
| Year | MSE \(\downarrow\) | NAE 78.66; NAM 85.25; EBM 85.81; NODE-GAM 85.09 |

These results support the paper’s claim that NAEs substantially outperform standard GAMs and NAMs and are often competitive with interaction-explaining GA\(^2\)M variants and black-box baselines, while retaining feature-level explanation complexity of \(O(n)\) rather than \(O(n^2)\) [2602.10585].

The implementation used AdamW, cosine annealing, feature encoders with 3–4 MLP layers, hidden dimensions typically \(64\) or \(128\), normalization via LayerNorm or BatchNorm, and dropout in the range \([0.0,0.6]\). Main experiments used \(K=4\) with all experts active; ablations extended to \(K=16\) for NAE and \(K=128\) for NAE-E [2602.10585].

## 6. Relation to neighboring additive-expert models

The explicit NAE framework sits within a wider family of additive neural decompositions. CAT groups raw features into high-level concepts, encodes each group into a scalar concept value, and feeds the resulting concept vector into a Tucker-decomposed TaylorNet. A common interpretation is that each concept encoder acts as an expert over a feature group, while TaylorNet supplies additive and low-order interaction terms over concepts; unlike NAEs, CAT has no gating network and keeps all concept-level terms active [2406.17931].

CRISP-NAM extends neural additive models to competing-risks survival analysis by assigning a dedicated FeatureNet to each feature and linear cause-specific readouts to each feature representation. The cause-specific log-hazard remains additive over feature contributions, making it a multi-task additive-expert model with shared feature experts and task-specific heads, but again without context-dependent gating across features [2505.21360].

NeurCAM introduces interpretable clustering through additive models with selection gates. Its cluster logits are sums of univariate and pairwise shape functions, and entmax-based gates choose which features or interactions populate a fixed number of slots. This suggests an alternative NAE design in which sparsity is controlled by slot budgets rather than by mixtures of experts per feature [2408.13361].

AxNN organizes additive experts through ensembles of GAMnet and xNN subnetworks. GAMnet supplies per-feature experts, xNN supplies additive index experts, and the overall ensemble is decomposed into main effects and higher-order interaction components through thresholded projection patterns. In that sense, AxNN is an additive-expert system built from boosted or stacked subnetworks rather than from context-gated per-feature mixtures [2004.02353].

Monotonic Neural Additive Models specialize the per-feature additive regime to regulated credit scoring. Each feature has its own one-dimensional neural subnetwork, and derivative penalties enforce individual monotonicity and pairwise monotonicity relations. The resulting architecture shows how expert-factorized additive models make shape constraints computationally cheap, because monotonicity can be enforced on univariate grids rather than through global verification [2209.10070].

Hybrid deep additive neural networks replace some affine layers by additive basis-expansion layers and combine them with standard fully connected layers. Their analysis shows that additive/basis-function “expert layers” can retain universal approximation while improving parameter efficiency, suggesting a different route to additive experts in which specialization is implemented by basis blocks rather than by gating [2411.09175].

A distinct but relevant direction appears in Mixture of Neuron Experts. There, a GLU expert is decomposed into a sum of linear neuron-experts
\[
\mathbf{E}_i(\mathbf{x}) = \sum_{k=1}^{d_{\text{expert}}} \mathbf{G}_i[k]\;\mathbf{A}_k \mathbf{x},
\]
and only the top-\(K_N\) neurons are computed inside each selected MoE expert. This is an additive-expert view below the feature level: expert selection happens at the neuron granularity, not at the level of input features or concepts [2510.05781].

## 7. Limitations and open directions

The principal limitation reported for NAEs is computational. Full routing uses \(\mathcal{A}_{ij}\) for all feature pairs \((i,j)\), yielding \(O(n^2 d K)\) parameters in the gating matrices. This can become expensive on very high-dimensional tabular problems. NAE-D reduces this cost but also removes the mechanism by which cross-feature interactions are represented [2602.10585].

Interpretation also becomes less direct as the number of experts grows. In practice, the recommended interpretation is not to inspect every \(g_{ik}\) separately but to analyze aggregate feature curves and their lower/upper envelopes. The paper further notes sensitivity to the hyperparameters \(K\), \(C\), and \(\lambda\); larger \(K\) often requires stronger regularization to avoid overfitting, and no automatic selection rule is provided [2602.10585].

A further issue is visualization of higher-order interaction structure. NAEs can represent multi-way effects through gating, and the paper presents three-way interaction examples, but it also notes the absence of a standardized visualization toolkit for such behavior. Suggested future directions include sparsifying the routing matrix \(\mathcal{A}\), improving procedures for selecting the number of experts and the variation penalty, and developing better tools for multi-way interaction visualization [2602.10585].

From the perspective of formal interpretability, the broader literature suggests a structural trade-off. Strictly additive NAMs admit provable cardinally-minimal sufficient explanations under efficient verification routines, whereas additive-expert models with global context gating do not directly satisfy those assumptions. This suggests that NAEs exchange some of the formal tractability of pure additive models for a wider expressive envelope that includes pairwise and higher-order context effects [2602.17530].

Source: https://www.emergentmind.com/topics/neural-additive-experts-naes