---
title: Psi-Decision Trees in Convex Analysis
url: https://www.emergentmind.com/topics/psi-decision-trees
type: topic
---

# Psi-Decision Trees in Convex Analysis

Searching arXiv for the cited papers and closely related work to ground the article.
Psi-Decision Trees, in the Bregman-divergence formulation, are decision trees whose node impurity and splitting criteria are induced by a differentiable, strictly convex potential $\psi$ via its associated Bregman divergence. They generalize the CART paradigm by replacing ad hoc impurity choices with a unified construction grounded in convex analysis and information geometry, and they subsume squared error, entropy, Gini, Poisson deviance, and Itakura–Saito as special cases [2606.13984]. The notation $\psi$ is also used in other decision-tree literatures to denote a bounded path-complexity measure for deterministic and nondeterministic trees [2312.01116] and a user-specified requirement for black-box policy synthesis [2409.03260]; a related decision-theoretic method, PSICA, uses trees to summarize probability-of-best treatment vectors in randomized trials with categorical treatments [1811.09065].

## 1. Formal definition in the Bregman framework

Let $\psi$ be a differentiable strictly convex function defined on a convex set $C \subset \mathbb{R}^d$. The associated Bregman divergence is

$$
D_{\psi}(x, y) = \psi(x) - \psi(y) - \nabla \psi(y)^{\top}(x-y).
$$

It measures how much $\psi$ deviates from its first-order tangent approximation at $y$ when evaluated at $x$. It satisfies $D_\psi \ge 0$ and equals $0$ iff $x=y$ [2606.13984].

For a dataset $S=\{z_i\}_{i\in S}$ in $C$, node impurity is defined as

$$
I_{\psi}(S) = \min_{\theta \in C} \frac{1}{|S|}\sum_{i\in S} D_{\psi}(z_i,\theta).
$$

The minimizer $\theta^*$ is the Bregman centroid of $S$, characterized by

$$
\nabla \psi(\theta^*) = \frac{1}{|S|}\sum_{i\in S}\nabla \psi(z_i).
$$

This is dual averaging: one averages in the dual coordinates induced by $\nabla \psi$ and then maps back through $(\nabla \psi)^{-1}$.

For a candidate split $S \to (S_L,S_R)$, the impurity gain is

$$
\Delta_{\psi} = I_{\psi}(S) - \frac{|S_L|}{|S|}I_{\psi}(S_L) - \frac{|S_R|}{|S|}I_{\psi}(S_R).
$$

Splits are chosen to maximize $\Delta_\psi$. Because $I_\psi$ is an expected convex Jensen gap, $\Delta_\psi \ge 0$.

A technical distinction arises between $\psi$ and $\phi=\psi^*$ in exponential-family modeling. In natural-parameter space, $\psi$ often denotes the log-partition. In mean space, the canonical prediction loss is typically generated by $\phi=\psi^*$. When the response lives in mean space, one uses $D_\phi$ and replaces $\nabla \psi$ with $\nabla \phi$ where appropriate. This is not a separate algorithmic idea; it is a coordinate choice within the same convex-analytic framework.

## 2. Recovery of classical CART criteria

The unifying claim of the framework is that many standard impurity measures are instances of particular convex generators. CART’s variance, entropy, and Gini criteria therefore appear not as unrelated design choices but as special cases derived from a common divergence construction [2606.13984].

| Setting | Generator | Induced impurity or divergence |
|---|---|---|
| Regression | $\psi(x)=\frac{1}{2}\|x\|^2$ | $D_\psi(x,y)=\frac{1}{2}\|x-y\|^2$; impurity is mean squared error/within-node variance |
| Multiclass classification | $\psi(p)=\sum_{j=1}^K p_j\log p_j$ | $D_\psi(p,q)=KL(p\|q)$; impurity simplifies to entropy $H(\hat p)$ |
| Gini-like classification | $\psi(p)=\frac{1}{2}\|p\|^2$ on the simplex | impurity becomes $\frac{1}{2}(1-\sum_j \hat p_j^2)$, proportional to Gini |
| Count data | $\phi(\mu)=\mu\log \mu-\mu$ | $D_\phi(y,\mu)=y\log(y/\mu)-y+\mu$; impurity is average Poisson deviance |
| Positive heteroscedastic data | $\phi(\mu)=-\log \mu-1$ | $D_\phi(y,\mu)=y/\mu-\log(y/\mu)-1$; Itakura–Saito divergence |

For multiclass classification, taking $\psi(p)=\sum_j p_j\log p_j$ over the simplex yields $D_\psi(p,q)=KL(p\|q)$. With one-hot labels, the centroid is the empirical class-proportion vector $\hat p$, and node impurity reduces to $H(\hat p)=-\sum_j \hat p_j\log \hat p_j$. Entropy reduction is thus a Bregman impurity reduction. If instead one uses a quadratic potential on the simplex, the same construction produces a quantity proportional to the Gini index.

For counts and positive responses, the framework moves naturally to mean-space generators. The Poisson choice $\phi(\mu)=\mu\log\mu-\mu$ induces the Poisson deviance, while $\phi(\mu)=-\log \mu -1$ induces Itakura–Saito, which is scale-invariant and is therefore suited to multiplicative or heteroscedastic positive responses.

The exponential-family interpretation is central. For a regular exponential family with density
$$
p_{\psi,\theta}(x)=\exp\{\langle \theta,T(x)\rangle-\psi(\theta)\}b(x),
$$
Banerjee’s representation gives
$$
p_{\psi,\theta}(x)\propto \exp(-D_{\phi}(x,\mu(\theta))).
$$
Optimizing node impurity with $D_\phi$ then corresponds to likelihood optimization, while $D_\psi$ in dual coordinates aligns with KL between family members. This suggests that the choice of $\psi$ is not merely computational; it encodes a statistical model and an underlying geometry.

## 3. Algorithmic construction and computational structure

The generic tree-construction procedure mirrors CART but replaces the impurity calculation and node representative with their Bregman analogues [2606.13984]. At a node containing sample set $S$, one computes the centroid $\theta^*$ from
$$
\nabla \psi(\theta^*)=\frac{1}{|S|}\sum_{i\in S}\nabla \psi(z_i),
$$
evaluates
$$
I_\psi(S)=\frac{1}{|S|}\sum_{i\in S}D_\psi(z_i,\theta^*),
$$
and stops if standard criteria hold: depth at least $d_{\max}$, node size below $n_{\min}$, impurity below $\epsilon$, or no split with gain at least $\gamma_{\min}$. Otherwise, one scans candidate feature-threshold splits and selects the split maximizing $\Delta_\psi$.

Efficient threshold scanning follows from sufficient statistics. The centroid depends on
$$
G=\sum_{i\in S}\nabla \psi(z_i),
\qquad
\theta^*=(\nabla \psi)^{-1}(G/|S|).
$$
During a left-to-right scan along a sorted feature, one updates $G_L(\tau)$ and $G_R(\tau)$ incrementally as points move across the threshold. Child centroids are recomputed as $(\nabla \psi)^{-1}(G_L/|S_L|)$ and $(\nabla \psi)^{-1}(G_R/|S_R|)$, while impurities are updated using streaming sums of $\psi(z_i)$ and inner products through
$$
D_{\psi}(z_i,\theta)=\psi(z_i)-\psi(\theta)-\langle \nabla \psi(\theta), z_i-\theta\rangle.
$$
This yields $O(n)$ scanning per feature after an $O(n\log n)$ sort, comparable to CART.

Prediction at leaves depends on the coordinate system. In $\psi$-space one predicts the representative $\theta^*$ or maps it to the mean parameter $\mu^*=\nabla \psi(\theta^*)$. In $\phi$-space, the leaf prediction is directly $\mu^*$. For multiclass classification this is the leaf class-probability vector; for regression it is the mean; for Poisson models it is the rate; for Gamma or Exponential settings it is the positive mean parameter.

Pruning extends in the expected way. Cost-complexity pruning is formulated with the $\psi$-induced impurity:
$$
R(T)=\sum_{t\ \mathrm{leaf}}\frac{|t|}{n}I_\psi(t),
\qquad
R_\alpha(T)=R(T)+\alpha |\mathrm{leaves}|.
$$
The penalty parameter $\alpha$ is selected by cross-validation, optionally using the $1$-SE rule, under the same $\psi$-induced loss.

## 4. Geometry, bounds, and consistency

The gain of a split admits an exact centroid decomposition. If $\theta^*$, $\theta_L^*$, and $\theta_R^*$ are the centroids of $S$, $S_L$, and $S_R$, then

$$
\Delta_{\psi}
=
\frac{|S_L|}{|S|}D_{\psi}(\theta_L^*,\theta^*)
+
\frac{|S_R|}{|S|}D_{\psi}(\theta_R^*,\theta^*).
$$

Equivalently, if a random variable $M$ takes values $\theta_L^*$ and $\theta_R^*$ with probabilities $|S_L|/|S|$ and $|S_R|/|S|$, then
$$
\Delta_\psi = E[\psi(M)]-\psi(E[M]).
$$
The gain is therefore a Jensen gap and is nonnegative [2606.13984].

Curvature properties of $\psi$ translate directly into gain bounds. If $\psi$ is $\mu$-strongly convex,
$$
D_{\psi}(x,y)\ge \frac{\mu}{2}\|x-y\|^2,
$$
then
$$
\Delta_{\psi}\ge
\frac{\mu}{2}\frac{|S_L||S_R|}{|S|^2}\|\theta_L^*-\theta_R^*\|^2.
$$
Splits that separate child centroids in Euclidean norm therefore have guaranteed nontrivial gain. If $\psi$ has $L$-Lipschitz gradient,
$$
D_{\psi}(x,y)\le \frac{L}{2}\|x-y\|^2,
$$
and one obtains the upper bounds
$$
\Delta_{\psi}
\le
\frac{L}{2}\left[
\frac{|S_L|}{|S|}\|\theta_L^*-\theta^*\|^2+
\frac{|S_R|}{|S|}\|\theta_R^*-\theta^*\|^2
\right]
\le
\frac{L}{2}\frac{|S_L||S_R|}{|S|^2}\|\theta_L^*-\theta_R^*\|^2.
$$

Consistency follows the classical partition-regularity route. Under shrinking cell diameters, growing sample sizes per leaf, and finite second moment of the response, the Bregman risk
$$
\mathcal{R}_{\psi}(f)=E[D_{\psi}(Z,f(X))]
$$
is minimized by
$$
f^*(x)=E[Z\mid X=x].
$$
If $\psi$ is $L$-smooth, then
$$
\mathcal{R}_{\psi}(\hat f_n)-\mathcal{R}_{\psi}(f^*)
\le
\frac{L}{2}E[(\hat f_n(X)-f^*(X))^2]\to 0.
$$
This places Psi-Decision Trees under the same partition regularity used for CART, provided $\psi$ is smooth.

The information-geometric interpretation is equally explicit. Natural and mean coordinates satisfy $\mu=\nabla \psi(\theta)$ and $\theta=\nabla \phi(\mu)$. Centroids are averages in dual space, while the Bregman Pythagorean theorem explains why projections onto convex sets minimize expected divergence. A plausible implication is that the framework is best understood not as a catalogue of impurity formulas but as a recursive-partitioning method defined on dual affine geometries.

## 5. Other meanings of $\psi$ in decision-tree research

A distinct line of work uses $\psi$ not as a loss generator but as a bounded complexity measure on paths in deterministic and nondeterministic decision trees for many-valued decision tables [2312.01116]. In that setting, $\psi$ satisfies positivity, commutativity, monotonicity, and boundedness from above, with the additional bounded-complexity condition $\psi(\alpha)\ge |\alpha|$. For a tree $\Gamma$, complexity is
$$
\psi(\Gamma)=\max\{\psi(\pi(\tau)):\tau\text{ is a complete path}\},
$$
and one studies the minimum deterministic and nondeterministic complexities, $\psi^d(T)$ and $\psi^a(T)$, as well as the worst-case comparison function $H^\infty_{\psi,A}(n)$. The main structural result is a dichotomy: if $H^\infty_{\psi,A}$ is everywhere defined, then it is either bounded above by a constant or satisfies $H^\infty_{\psi,A}(n)\ge n$ for infinitely many $n$. The same paper also proves that for any nondecreasing $\varphi$ with $\varphi(n)\ge n$ and $\varphi(0)=0$, one can construct a closed class $A$ and a bounded complexity measure $\psi$ such that
$$
\varphi(n)\le H^\infty_{\psi,A}(n)\le \varphi(n)+n.
$$
Here, “$\psi$-decision tree” refers to a cost model on attribute queries rather than to Bregman impurity.

A separate synthesis literature treats $\psi$ as a specification or cost functional for control. Decision-tree policies are synthesized for deterministic black-box systems under a finite predicate discretization and bounded tree size, with optimality defined with respect to a trace-based objective such as steps-to-goal [2409.03260]. The search enumerates finite tree hypotheses and prunes them through trace equivalence and branch-and-bound. Its key device is a distance-based pruning rule: after evaluating a predicate $[x_j\ge v]$ at a node, one records the minimum value of $x_j$ among states that took the true branch and skips all later thresholds $v'\le d_i$, because under determinism those trees are trace-equivalent on the evaluated initial states. Under deterministic dynamics, finite predicate discretization, bounded tree size or depth, and finite-horizon evaluation, the method is complete for the finite hypothesis class and returns the smallest optimal tree among ties. The reported runtime reduction from trace-based pruning is $4$–$20\times$, with examples including CartPole $3.13\,\mathrm{s}\to 0.22\,\mathrm{s}$ for $|S_0|=1$ and MountainCar $4.72\,\mathrm{s}\to 0.21\,\mathrm{s}$ for $|S_0|=1$.

PSICA is related but different again. It addresses randomized trials with more than two categorical treatments by fitting treatment-wise predictive models, estimating for each covariate vector the probability that each treatment is best, and then growing a decision tree on those probability vectors [1811.09065]. For a subgroup $S$, the leaf reports
$$
p(S)=(p_1(S),\ldots,p_K(S)),
$$
where $p_j(S)$ is the probability that treatment $j$ is best, and the leaf label is the set of possible best treatments after $\alpha$-based exclusion of low-probability options. Splitting is based on a loss tied to expected misassignment cost, and an optional significance mask uses a chi-square test to suppress spurious splits. In simulations, PSICA methods typically achieved near-perfect accuracy, about $0.97$–$1.00$, while uncertainty and suspect-split behavior depended on whether probability estimation used bootstrap or infinitesimal-jackknife uncertainty propagation. The leaf-level ability to say “I don’t know” marks a decision-theoretic extension of tree outputs beyond single-label prediction.

## 6. Practical use, limitations, and extensions

Within the Bregman framework, the operational recommendation is to choose $\psi$ to match the response type and noise structure [2606.13984]. Squared error is appropriate for Gaussian-like homoscedastic regression, negative entropy for categorical or multiclass responses, quadratic potentials on the simplex for Gini-like criteria, Poisson divergence for counts, and Itakura–Saito for positive heteroscedastic data with multiplicative or scale noise. The same source notes that gains in generalization can occur when $\psi$ matches the data-generating process, whereas performance is indistinguishable from CART when both optimize the same criterion.

The framework has domain and regularity constraints. Responses must lie in the domain of $\psi$; on the boundary of the probability simplex, gradients may be undefined, so smoothing is needed. For probabilities, Laplace smoothing avoids $p_j=0$ when computing logarithms. For positive responses, clipping at $\epsilon>0$ avoids $\log(0)$ and division by zero under Itakura–Saito. Numerical stabilization may require log-sum-exp for KL calculations and stabilized gradients more generally.

Model mismatch is a statistical limitation rather than a formal failure mode. Using squared error under multiplicative noise, for example, reduces statistical efficiency. The text explicitly states that one should choose $\psi$ aligned with the data’s exponential-family model or error structure. A related misconception is that every convex loss fits the framework: absolute deviation is not a Bregman divergence. Extensions to non-differentiable objectives are possible through generalized Bregman divergences with subgradients or by replacing the loss with other convex risks such as pinball loss, but the paper notes that one then loses closed-form centroids.

The framework extends naturally to ensembles. Psi-Decision Trees can be inserted into random forests and boosting by replacing impurity or loss with the $\psi$-induced version. The same source states that diversity or ambiguity decompositions extend to Bregman divergences, supporting ensemble analysis under the $\psi$ framework. This suggests a broad research program in which recursive partitioning, convex duality, and model-aware loss design are treated as a single object rather than as separate design decisions.

Across the alternate meanings of $\psi$, the main source of confusion is terminological. In Bregman-based trees, $\psi$ is a convex potential; in complexity theory, it is a bounded complexity measure; in policy synthesis, it is a specification or cost functional; and in PSICA the emphasis is on probability-of-best treatment vectors rather than on a literal $\psi$ generator. The commonality is structural rather than definitional: each framework uses trees as interpretable recursive partitions while elevating a mathematically explicit objective—divergence, path cost, trace specification, or decision-theoretic loss—to the central organizing role.

Source: https://www.emergentmind.com/topics/psi-decision-trees