November 04, 2025
We study whether latent geometry can be treated as an explicit component of representation quality in data-scarce learning. For an encoder \(\phi\), we define a geometry-aware population score \(Q_{\beta,\gamma}(\phi)=I(\phi(X);Y)-\beta\mathcal{C}(\phi)-\gamma d_{\mathrm{int}}(\phi)\), where \(I(\phi(X);Y)\) measures task-relevant information, \(\mathcal{C}(\phi)\) measures encoder-induced curvature, and \(d_{\mathrm{int}}(\phi)\) measures the intrinsic dimension of the latent representation. This extends bottleneck-style representation learning by making latent geometry part of the learning criterion rather than only a post hoc diagnostic.
Under explicit smooth-manifold, loss-transfer, and estimator-concentration assumptions, we derive non-asymptotic low-label generalization bounds in which latent intrinsic dimension and latent covering complexity enter directly. We also characterize the associated information–geometry frontier and prove consistency of the empirical surrogate used to estimate the population score. These results identify a mechanism by which encoder geometry affects low-label learning through latent covering numbers, loss-class entropy, and uniform deviation.
We instantiate the theory as the Variational Geometric Information Bottleneck, V-GIB, which augments variational bottleneck training with stochastic curvature and intrinsic-dimension penalties. Experiments on low-label real benchmarks
compare V-GIB with ERM, VIB, and ablated variants across label fractions from \(1\%\) to \(20\%\). The results show that geometry-aware training can improve predictive
performance and reduce latent geometric complexity in several data-scarce regimes, especially on FashionMNIST and CIFAR-10, while also revealing cases where ablations or simpler baselines are competitive. The evidence supports a precise conclusion, i.e.,
latent geometry is a measurable and controllable axis of representation quality under label scarcity, not a universal guarantee of dominance by one fixed regularizer.
Keywords: representation learning; data scarcity; information bottleneck; latent geometry; intrinsic dimension; curvature regularization; generalization bounds; variational geometric information bottleneck; low-label learning; geometric complexity
MSC Classification: 68T05; 68T07; 68Q32; 94A17; 62H30; 54E35.
Modern supervised learning is often constrained by label scarcity. This has motivated few-shot learning, meta-learning, self-supervision, and representation learning, all of which study how useful representations can be learned when labels are limited [1]–[6].
A standard answer is information. In the information bottleneck view, a good representation preserves target-relevant information while controlling complexity through compression [7], [8]. This view is influential, but it leaves open a structural issue. That is, two representations may retain similar task-relevant information while inducing very different latent geometries. In low-label settings, such geometric differences can change the effective complexity of the downstream learning problem and therefore affect generalization.
A separate line of work shows that geometry matters in learning. Manifold methods exploit low-dimensional structure in high-dimensional data [9], [10]. Learning theory under manifold assumptions shows that statistical complexity can depend on intrinsic rather than ambient dimension [11], [12]. Geometric deep learning further emphasizes structured latent spaces and geometry-aware inductive bias [13], [14]. What remains less developed is a representation-learning analysis in which geometry is treated not only as an assumption or diagnostic, but as an explicit component of representation quality.
This paper studies this geometric axis. We ask whether latent geometry can be formalized as a measurable part of representation quality under data scarcity, whether it enters low-label generalization bounds explicitly, whether it can be estimated in practice, and whether a practical training objective can improve the observed information–geometry trade-off.
We answer this question under explicit smoothness, transfer, and estimator assumptions. We define a geometry-aware population score combining task-relevant information with penalties on latent curvature and intrinsic dimension. Under these assumptions, geometry affects low-label generalization through the chain \[\text{encoder geometry} \;\Longrightarrow\; \text{latent covering numbers} \;\Longrightarrow\; \text{loss-class entropy} \;\Longrightarrow\; \text{generalization}.\] This yields non-asymptotic bounds in which latent intrinsic dimension and latent covering complexity enter explicitly. We also prove consistency of the empirical surrogate under concentration assumptions for the information, curvature, and intrinsic-dimension estimators.
These results motivate the Variational Geometric Information Bottleneck (V-GIB). V-GIB augments variational bottleneck training with tractable stochastic proxies for curvature and intrinsic dimension. The
method is not proposed as uniformly dominant over all alternatives. Its role is to operationalize the theory and test whether the geometric quantities identified by the analysis track low-label behavior across datasets.
The paper makes four contributions.
We identify geometry as an explicit axis of representation quality under data scarcity and formalize it through a population score combining information, curvature, and intrinsic dimension.
We derive non-asymptotic low-label generalization bounds depending explicitly on latent intrinsic dimension and latent covering complexity, and characterize the associated information–geometry frontier.
We prove consistency of the empirical surrogate optimized by V-GIB under explicit concentration assumptions for the information, curvature, and intrinsic-dimension estimators.
We propose V-GIB and evaluate it on real low-label benchmarks to test whether geometric quantities track low-label behavior and whether geometry-aware training can improve the empirical frontier.
The claim is narrower than a general theory of representation learning and sharper than a generic regularization claim. We do not claim that geometry-aware training is uniformly best on every dataset or at every label fraction. We claim that, under the stated assumptions, geometry is a statistically meaningful variable of representation quality, and that accounting for it can improve the observed information–geometry trade-off in low-label regimes.
The closest information-theoretic starting point is the Information Bottleneck, which studies representations that preserve target-relevant information while discarding irrelevant variation [7]. Variational relaxations made this principle practical for deep models [8]. In these formulations, complexity is controlled mainly through compression.
Geometry enters learning through a different route. Manifold learning and manifold regularization exploit low-dimensional structure in high-dimensional observations [9], [10]. Statistical learning under manifold assumptions shows that covering numbers and sample complexity can scale with intrinsic dimension rather than ambient dimension [11], [12]. Geometric deep learning extends these ideas to structured domains and geometry-aware architectures [13], [14].
Our setting combines these perspectives but is not identical to either. Unlike standard bottleneck formulations, we treat latent geometry as part of representation quality itself. Unlike manifold regularization, we do not only assume geometry in the input data and regularize along it. We study how encoder-induced geometry enters low-label generalization through latent covering complexity and loss-class entropy.
The paper is also related to interpretable and structured representations, although the objective is different. Many interpretability methods analyze trained models after the fact through attribution, concept activation, or neuron-level inspection [15]–[17]. Here, geometric structure is placed directly in the representation-learning criterion and studied before post hoc interpretation begins.
This section formalizes one claim, precisely, in low-label learning, predictive fit alone does not determine representation quality. Geometry provides an additional axis because it affects effective complexity and hence generalization. We define a geometry-aware population score, state the assumptions under which geometry enters low-label bounds, and identify the mechanism by which encoder geometry affects generalization. Full proofs for this section are given in Appendix 4.
All statements are conditional on the assumptions below. No claim is made for arbitrary data distributions or arbitrary network classes.
Let \(X \subset \mathbb{R}^D\) be the input space and \(Y\) the label space. A representation is an encoder \[\phi \colon X \to \mathbb{R}^m,\] combined with a predictor \[g \colon \mathbb{R}^m \to \mathcal{Y},\] so that predictions have the form \(g(\phi(x))\).
We use the standard representation-learning factorization, i.e., the encoder determines the latent geometry, and the predictor uses that geometry for the downstream task. This is compatible with information bottleneck learning, manifold-based learning, and geometric representation learning [7]–[10], [13], [14], [18]. The difference here is that geometry is not treated as a post hoc descriptor. In the low-label regime, latent geometry affects covering numbers, metric entropy, and uniform deviation, so it becomes a measurable variable of representation quality.
We use two geometric quantities. The first is the curvature proxy \[\mathcal{C}(\phi) = \mathbb{E}_{x \sim P_X}\bigl[\|\nabla^2 \phi(x)\|_F^2\bigr],\] which measures local variation of the encoder Jacobian. The second is the intrinsic dimension \(d_{\mathrm{int}}(\phi)\), the effective dimension of the representation \(\phi(X)\). In practice, it can be estimated by local PCA or nearest-neighbor methods [19]; in the theory it is treated as a population quantity.
Definition 1 (Geometric complexity). For an encoder \(\phi\), define \[G_{\gamma}(\phi) = \mathcal{C}(\phi) + \gamma\, d_{\mathrm{int}}(\phi), \qquad \gamma \ge 0.\]
Definition 2 (Geometry-aware representation score). For an encoder \(\phi\), define \[Q_{\beta,\gamma}(\phi) = I\bigl(\phi(X);Y\bigr) - \beta\,\mathcal{C}(\phi) - \gamma\, d_{\mathrm{int}}(\phi), \qquad \beta,\gamma \ge 0.\]
The first term measures retained task-relevant information [7], [20]. The remaining terms penalize latent irregularity and latent effective dimension. Thus \(Q_{\beta,\gamma}\) treats geometry as part of representation quality itself, not as an after-training summary.
The analysis uses three assumptions.
Assumption 1 (Representation regularity). The support of \(P_X\) is contained in a compact smooth manifold \(\mathcal{M} \subset \mathbb{R}^D\). For each feasible encoder \(\phi \in \Phi\), the image \(\phi(\mathcal{M})\) is contained in a compact \(d_\phi\)-dimensional \(C^2\) submanifold \(\mathcal{N}_\phi \subset \mathbb{R}^m\) with reach \(\tau_\phi>0\), finite volume \[V_\phi=\operatorname{Vol}_{d_\phi}(\mathcal{N}_\phi)<\infty,\] and sectional curvature bounded by \[|\kappa_\phi|\le \kappa_{\phi,\max}.\] Moreover, \(\phi\) is \(L_\phi\)-Lipschitz on \(\mathcal{M}\): \[\|\phi(x)-\phi(x')\|_2 \le L_\phi \|x-x'\|_2 \qquad \text{for all } x,x'\in\mathcal{M}.\] We set \[d_{\mathrm{int}}(\phi):=d_\phi.\]
This assumption specifies the geometric regime in which covering arguments are meaningful. Reach, curvature, and volume control the latent manifold, while Lipschitz regularity controls the encoder-induced distortion [11], [21], [22].
Assumption 2 (Loss-class transfer). For each encoder \(\phi\), let \[\mathcal{F}_{\phi} = \{(x,y)\mapsto \ell(g(\phi(x)),y): g\in\mathcal{G}\},\] where \(\ell\) is bounded by \(B\). There exist constants \(A_0,A_1>0\), independent of \(N\), such that for every empirical measure \(P_N\) and every \(\eta\in(0,1]\), \[\log \mathcal{N}\!\bigl(\eta,\mathcal{F}_{\phi},L_2(P_N)\bigr) \le A_0 \left[ d_{\mathrm{int}}(\phi)\log\!\Bigl(\frac{A_1}{\eta}\Bigr) + \log \mathcal{N}\!\bigl(\eta/A_1,\phi(\mathcal{M}),\|\cdot\|_2\bigr) \right].\]
This is the structural bridge between latent geometry and statistical complexity. It is also where predictor-class complexity enters the analysis. The results are conditional on this transfer bound and do not claim it for arbitrary predictor classes.
Assumption 3 (Estimator control). There exist estimators \(\widehat I_N(\phi)\), \(\widehat{\mathcal{C}}_{N,K}(\phi)\), and \(\widehat d_N(\phi)\) such that, uniformly over \(\phi\in\Phi\), \[\sup_{\phi \in \Phi}\bigl|\widehat I_N(\phi)-I(\phi(X);Y)\bigr| = O_p\!\bigl(r_I(N)\bigr),\] \[\sup_{\phi \in \Phi}\bigl|\widehat d_N(\phi)-d_{\mathrm{int}}(\phi)\bigr| = O_p\!\bigl(r_d(N)\bigr),\] and the curvature estimator satisfies the concentration condition stated in Appendix 5.2.
This assumption isolates the estimation burden. It does not require a specific mutual-information estimator or intrinsic-dimension estimator. It only requires the uniform control used later in the surrogate consistency result.
The theory uses the following chain: \[\text{encoder geometry} \;\Longrightarrow\; \text{latent covering numbers} \;\Longrightarrow\; \text{loss-class entropy} \;\Longrightarrow\; \text{generalization}.\] Assumption 1 controls the covering numbers of the latent image \(\phi(\mathcal{M})\). Assumption 2 transfers this latent covering control to the entropy of the induced loss class. Empirical-process bounds then convert the entropy bound into a uniform deviation bound.
Lemma 1 (Latent covering bound). Under Assumption 1, for each feasible encoder \(\phi \in \Phi\), there exists a constant \[C_\phi = C\bigl(d_{\mathrm{int}}(\phi),\tau_\phi,\kappa_{\phi,\max},V_\phi\bigr)>0\] such that, for all sufficiently small \(\varepsilon>0\), \[\mathcal{N}\!\bigl(\varepsilon,\phi(\mathcal{M}),\|\cdot\|_2\bigr) \le C_\phi \,\varepsilon^{-d_{\mathrm{int}}(\phi)}.\]
Lemma 1 gives the first part of the mechanism. Latent intrinsic dimension and the covering constant \(C_\phi\) control the size of the representation space seen by the downstream predictor. The proof is given in Appendix 4.1.
Theorem 1 (Geometry-sensitive uniform deviation). Under Assumptions 1 and 2, there exist constants \(C_1,C_2>0\), independent of \(N\) and \(\phi\), such that for any \(\delta\in(0,1)\), with probability at least \(1-\delta\), \[\sup_{f\in\mathcal{F}_{\phi}} \bigl(R(f)-\widehat R_N(f)\bigr) \le C_1 \sqrt{ \frac{ d_{\mathrm{int}}(\phi)\log(C_2 N) + \log\!\bigl(1+C_\phi\bigr) + \log(2/\delta) }{N} },\] where \(C_\phi\) is the latent covering constant from Lemma 1.
Theorem 1 is the main mechanism result. In this conditional regime, low-label generalization depends explicitly on latent intrinsic dimension and latent covering complexity, not only on predictive fit. Its proof is given in Appendix 4.2.
The theorem also gives a direct sample-complexity reading. At fixed task information, representations with smaller latent intrinsic dimension and smaller covering constant require fewer labels to reach the same uniform-deviation level. The formal corollary and proof are given in Appendix 4.3.
The previous results show how geometry enters generalization. The next result shows how it enters optimization.
Theorem 2 (Information–geometry frontier). Fix \[\Phi(R_0) = \{\phi \in \Phi : \mathcal{R}(\phi)\le R_0\},\] where \(\mathcal{R}(\phi)\) is a population prediction-risk functional. For \(\beta\ge 0\), define \[\phi_{\beta} \in \arg\max_{\phi\in\Phi(R_0)} \left\{ I\bigl(\phi(X);Y\bigr)-\beta G_{\gamma}(\phi) \right\}.\] Assume uniqueness of \(\phi_\beta\), differentiability of \(\beta\mapsto \phi_\beta\), and differentiability of \(\phi\mapsto I(\phi(X);Y)\) and \(\phi\mapsto G_\gamma(\phi)\). Then, whenever \(dG_\gamma(\phi_\beta)/d\beta\neq 0\), \[\frac{d\, I(\phi_{\beta}(X);Y)}{d\, G_{\gamma}(\phi_{\beta})} = \beta.\] If \(dG_\gamma(\phi_\beta)/d\beta<0\), then increasing the geometry penalty decreases geometric complexity along the selected frontier.
The proof is given in Appendix 5.1. Theorem 2 is the optimization counterpart of Theorem 1. Geometry does not only enter the generalization bound; it parameterizes a trade-off curve between retained task information and geometric complexity. The slope identity says that \(\beta\) is the marginal information cost of increasing the allowed geometric complexity along the selected frontier.
The population score \(Q_{\beta,\gamma}\) is not optimized directly. The practical method optimizes the empirical surrogate \[\widehat Q_{N,K}(\phi) = \widehat I_N(\phi) - \beta \widehat{\mathcal{C}}_{N,K}(\phi) - \gamma \widehat d_N(\phi).\] Thus the optimizer targets a sample-based surrogate for \(Q_{\beta,\gamma}\), not the population score itself.
The fixed-encoder concentration statement for the stochastic curvature estimator is given in Appendix 5.2. The uniform surrogate result below assumes uniform control of the three empirical components over the feasible encoder class.
Theorem 3 (Uniform surrogate consistency). Assume Assumption 3 and suppose, in addition, that the curvature estimator satisfies the uniform control \[\sup_{\phi\in\Phi} \bigl| \widehat{\mathcal{C}}_{N,K}(\phi)-\mathcal{C}(\phi) \bigr| = O_p\!\bigl((NK)^{-1/2}\bigr).\] Then \[\sup_{\phi\in\Phi} \bigl| \widehat Q_{N,K}(\phi)-Q_{\beta,\gamma}(\phi) \bigr| = O_p\!\left( r_I(N) + (NK)^{-1/2} + r_d(N) \right).\] If, in addition, \(\Phi\) is a compact metric space, \(Q_{\beta,\gamma}\) is continuous on \(\Phi\), and \(Q_{\beta,\gamma}\) has a unique maximizer \(\phi^\star\), then any empirical maximizer \[\widehat \phi_{N,K} \in \arg\max_{\phi\in\Phi} \widehat Q_{N,K}(\phi)\] satisfies \[Q_{\beta,\gamma}(\widehat \phi_{N,K}) \to Q_{\beta,\gamma}(\phi^\star), \qquad \widehat \phi_{N,K} \to \phi^\star\] in probability.
The proof is given in Appendix 5.3. Theorem 3 links the optimizer to the population theory. It does not claim that the empirical objective is exact. It states that, under explicit concentration assumptions, the empirical objective is a consistent surrogate for the geometry-aware population score.
Under the stated assumptions, geometry can be formalized as part of representation quality, enters low-label generalization through latent covering complexity, defines an information–geometry frontier, and can be targeted by a consistent empirical surrogate. The experiments test the corresponding empirical question, i.e., whether these geometric quantities track low-label behavior across datasets and whether geometry-aware training improves the observed trade-off.
The V-GIB family generally improves over ERM across the tested label fractions. The aim is not to show that V-GIB is uniformly dominant. The aim is to test the specific prediction of the theory, basically, when
labels are scarce, latent geometry should act as a measurable axis of representation quality, and geometry-aware training can improve the information–geometry trade-off in some regimes.
The validation is organized around three questions. First, does V-GIB improve low-label predictive performance relative to ERM and VIB? Second, do the geometric diagnostics expose differences that are not visible
from predictive fit alone? Third, do the ablations show that curvature and intrinsic dimension contribute differently across datasets?
We evaluate on image and tabular benchmarks. The primary comparative results use Breast Cancer, FashionMNIST, and CIFAR-10, for which complete predictive summaries are available. We also include CovType as a supporting optimization-side check because the available CovType summaries report training loss rather than full validation and test metrics. We therefore use CovType only as supporting optimization evidence, not as primary generalization evidence.
All methods are trained on stratified subsets of the available training data with label fractions \[0.01,\qquad 0.05,\qquad 0.10,\qquad 0.20.\] Results are reported as mean \(\pm\) standard deviation over seeds \(13,29,47\).
We compare five methods. The first is deterministic empirical-risk minimization, denoted ERM. The second is the variational information bottleneck baseline, denoted VIB. The third is the full proposed method, denoted
V-GIB. The remaining two are ablations, V-GIB-no-curv and V-GIB-no-dim, which remove the curvature and intrinsic-dimension penalties, respectively.
All methods use the same latent-classifier backbone. The common training objective is \[\mathcal{L}
=
\mathcal{L}_{\mathrm{CE}}
+
\beta_{\mathrm{KL}}\mathcal{L}_{\mathrm{KL}}
+
\beta_{\mathrm{curv}}\mathcal{L}_{\mathrm{curv}}
+
\gamma_{\mathrm{dim}}\mathcal{L}_{\mathrm{dim}}.\] For ERM, all regularization weights are zero and the encoder is deterministic. For VIB, only the KL term is active. For V-GIB, the KL, curvature, and dimension
terms are active. The two ablations remove one geometric term at a time.
The curvature diagnostic is a stochastic Jacobian-norm proxy measuring the sensitivity of the latent mean \(\mu(x)\) to the input. The intrinsic-dimension diagnostic is the participation ratio of the latent covariance, reported as a fraction of the latent dimension. All geometry diagnostics are computed from \(\mu(x)\), not from sampled latent variables. Checkpoints are selected using validation performance and test metrics are evaluated only after checkpoint selection. Full architecture, preprocessing, hyperparameter, diagnostic, and checkpoint-selection details are given in Appendix 5.5.
The Breast Cancer benchmark gives a small-data tabular stress test. The \(1\%\) case is deliberately severe, because it contains only four labeled training examples. In that regime, ERM and VIB
achieve the highest mean test accuracy, both \(0.8333\), while the full V-GIB model reaches \(0.8099\), as reported in Table 1 and Figure 1. We therefore do not treat the \(1\%\) case as evidence of dominance.
The clearer low-label signal appears from \(5\%\) onward. At \(5\%\), V-GIB improves test accuracy over ERM from \(0.8158\) to
\(0.8977\), and improves macro-F1 from \(0.7960\) to \(0.8849\); see Table 1 and Figure 1 (b). At \(10\%\), V-GIB reaches \(0.9094\) test accuracy, matching VIB and improving over ERM by \(0.0556\), as shown in Table 1 and Figure 1 (a). At \(20\%\), V-GIB reaches \(0.9298\) test accuracy and \(0.9263\) macro-F1, while V-GIB-no-curv gives the best result in the geometric family; the corresponding predictive trends are shown in Figure 1.
The conclusion on this dataset is therefore precise. Geometry-aware variants improve low-label performance once the labeled set is large enough to stabilize training, but the full V-GIB objective is not uniformly best over all ablations, as
seen in Table 1. This is consistent with the theory, which predicts a geometry-information trade-off rather than universal dominance of a single regularization setting.
| Labels | ERM | VIB | V-GIB | Best | Best variant | \(\Delta\) ERM | \(\Delta\) VIB |
|---|---|---|---|---|---|---|---|
| 0.01 | \(0.8333\!\pm\!0.0780\) | \(0.8333\!\pm\!0.0780\) | \(0.8099\!\pm\!0.0883\) | \(0.8246\!\pm\!0.0575\) | No-curv | \(-0.0234\) | \(-0.0234\) |
| 0.05 | \(0.8158\!\pm\!0.1412\) | \(0.9006\!\pm\!0.0681\) | \(0.8977\!\pm\!0.0355\) | \(0.9006\!\pm\!0.0396\) | No-curv | \(+0.0819\) | \(-0.0029\) |
| 0.10 | \(0.8538\!\pm\!0.1009\) | \(0.9094\!\pm\!0.0704\) | \(0.9094\!\pm\!0.0571\) | \(0.9094\!\pm\!0.0571\) | V-GIB | \(+0.0556\) | \(+0.0000\) |
| 0.20 | \(0.8918\!\pm\!0.1229\) | \(0.9269\!\pm\!0.0571\) | \(0.9298\!\pm\!0.0575\) | \(0.9474\!\pm\!0.0304\) | No-curv | \(+0.0380\) | \(+0.0029\) |
2pt
Figure 1: Predictive performance on Breast Cancer under label scarcity. The \(1\%\) case contains only four labeled examples and is unstable. From \(5\%\) onward, the
V-GIB family improves test accuracy and macro-F1 relative to ERM, while AUROC is already high for most methods.. a — Test accuracy., b — Macro-F1., c — AUROC.
FashionMNIST gives a clearer image-domain test of the geometric hypothesis. Across all four label fractions, the full V-GIB model improves mean test accuracy over ERM, as shown in Table 2 and Figure 2. The gains are \(+0.0199\), \(+0.0480\), \(+0.0151\), and \(+0.0500\) at \(1\%\), \(5\%\), \(10\%\), and \(20\%\) labels, respectively.
The strongest mean predictive performance comes either from the full model or from V-GIB-no-dim. At \(1\%\), \(5\%\), and \(10\%\) labels,
V-GIB-no-dim gives the best mean test accuracy, with \(0.6978\), \(0.7894\), and \(0.8415\) (Table 2; Figure 2 (a)). At \(20\%\), the full V-GIB model is best, reaching \(0.8692\)
test accuracy, \(0.8679\) macro-F1, and \(0.9890\) AUROC (Table 2; Figure 2 (b)).
The geometry diagnostics show why this is not only a predictive-performance story. The full V-GIB model keeps test curvature substantially below ERM, as seen in Figure 2 (c). The
curvature proxy for V-GIB is \(39.8209\), \(34.6334\), \(37.7687\), and \(30.2081\) across the four label
fractions, compared with \(79.9008\), \(114.4583\), \(182.2521\), and \(142.1634\) for ERM. Thus, on
FashionMNIST, the geometric regularizer improves or preserves predictive performance while producing a smoother latent representation; the corresponding utility behavior is shown in Figure 2 (d).
| Method | \(1\%\) labels | \(5\%\) labels | \(10\%\) labels | \(20\%\) labels |
|---|---|---|---|---|
| ERM | \(0.6729 \pm 0.0599\) | \(0.7409 \pm 0.0115\) | \(0.8261 \pm 0.0030\) | \(0.8192 \pm 0.0221\) |
| VIB | \(0.6734 \pm 0.0107\) | \(0.7831 \pm 0.0374\) | \(0.8341 \pm 0.0105\) | \(0.8343 \pm 0.0148\) |
| V-GIB | \(0.6928 \pm 0.0100\) | \(0.7889 \pm 0.0271\) | \(0.8412 \pm 0.0098\) | \(\mathbf{0.8692 \pm 0.0098}\) |
| V-GIB-no-curv | \(0.6822 \pm 0.0262\) | \(0.7811 \pm 0.0005\) | \(0.8142 \pm 0.0111\) | \(0.8492 \pm 0.0203\) |
| V-GIB-no-dim | \(\mathbf{0.6978 \pm 0.0117}\) | \(\mathbf{0.7894 \pm 0.0258}\) | \(\mathbf{0.8415 \pm 0.0069}\) | \(0.8679 \pm 0.0081\) |
Figure 2: FashionMNIST validation. The geometric methods outperform ERM across the tested label fractions. The full V-GIB model becomes strongest at \(20\%\) labels. The main
structural distinction is curvature. The full model and the no-dimension ablation remain at substantially lower curvature than ERM, VIB, and the no-curvature ablation while retaining the best predictive performance.. a — Test
accuracy., b — Test AUROC., c — Test curvature proxy., d — Test utility proxy.
CIFAR-10 is the hardest primary benchmark in the validation suite. It is therefore the most important check that the geometric objective remains useful beyond small tabular data and simple image structure. The full V-GIB model improves over
both ERM and VIB in test accuracy at every label fraction, as shown in Table 3 and Figure 3.
At \(1\%\) labels, V-GIB improves test accuracy from \(0.1323\) for ERM and \(0.1947\) for VIB to \(0.3405\) (Table 3; Figure 3 (a)). The corresponding macro-F1 improves to \(0.3277\), compared with \(0.0578\) for ERM and \(0.1380\) for VIB (Figure 3 (b)). At \(10\%\) and \(20\%\), the full V-GIB model is the strongest method among the compared variants. At \(20\%\), it reaches \(0.5929\) test accuracy and \(0.5926\) macro-F1, improving over ERM by \(0.0660\) accuracy and \(0.0770\) macro-F1, and over VIB by \(0.0398\) accuracy and \(0.0532\) macro-F1.
The ablations show a useful transition. Removing the curvature term gives the best result only at \(1\%\) and \(5\%\), while the full V-GIB model is best at \(10\%\) and \(20\%\) (Table 3). This suggests that the curvature term becomes more useful once the labeled set is large enough for geometric
regularization to shape the representation without overwhelming the predictive signal. The AUROC results in Figure 3 (c) provide the corresponding ranking-quality check.
| Labels | ERM | VIB | V-GIB | Best | Best variant | \(\Delta\) ERM | \(\Delta\) VIB |
|---|---|---|---|---|---|---|---|
| 0.01 | \(0.1323\!\pm\!0.0189\) | \(0.1947\!\pm\!0.1040\) | \(0.3405\!\pm\!0.0126\) | \(0.3409\!\pm\!0.0044\) | No-curv | \(+0.2082\) | \(+0.1458\) |
| 0.05 | \(0.4560\!\pm\!0.0147\) | \(0.4614\!\pm\!0.0064\) | \(0.4625\!\pm\!0.0225\) | \(0.4639\!\pm\!0.0344\) | No-curv | \(+0.0065\) | \(+0.0011\) |
| 0.10 | \(0.5123\!\pm\!0.0192\) | \(0.4683\!\pm\!0.0483\) | \(0.5303\!\pm\!0.0069\) | \(0.5303\!\pm\!0.0069\) | V-GIB | \(+0.0181\) | \(+0.0620\) |
| 0.20 | \(0.5269\!\pm\!0.0308\) | \(0.5531\!\pm\!0.0383\) | \(0.5929\!\pm\!0.0375\) | \(0.5929\!\pm\!0.0375\) | V-GIB | \(+0.0660\) | \(+0.0398\) |
2pt
Figure 3: Predictive performance on CIFAR-10 under label scarcity. The full V-GIB model improves over ERM and VIB in test accuracy at all tested label fractions, with the largest gains at \(1\%\), \(10\%\), and \(20\%\).. a — Test accuracy., b — Macro-F1., c — AUROC.
The available CovType postprocessed summaries report training loss rather than full validation or test metrics. We therefore treat CovType as supporting optimization evidence rather than primary generalization evidence.
Across all methods, the mean training loss decreases as the label fraction increases from \(0.01\) to \(0.10\), indicating stable optimization under increased supervision (Table 4). ERM attains the lowest training loss at every reported fraction, which is expected because it has the least constrained objective. VIB remains below full V-GIB, while
V-GIB-no-curv improves the optimization profile relative to full V-GIB. This pattern is useful rather than negative. It shows that the geometric terms impose a real optimization cost, so the predictive gains observed on
FashionMNIST and CIFAR-10 are not a trivial consequence of easier training.
| Fraction | ERM | VIB | V-GIB | No-curv | No-dim |
|---|---|---|---|---|---|
| 0.01 | \(1.1761\!\pm\!0.0114\) | \(1.2486\!\pm\!0.0103\) | \(1.3032\!\pm\!0.00996\) | \(1.2528\!\pm\!0.0082\) | \(1.3002\!\pm\!0.01006\) |
| 0.05 | \(0.788960\!\pm\!0.001583\) | \(0.864654\!\pm\!0.002360\) | \(0.911863\!\pm\!0.003630\) | \(0.867844\!\pm\!0.003656\) | \(0.908960\!\pm\!0.003831\) |
| 0.10 | \(0.705979\!\pm\!0.002311\) | \(0.778726\!\pm\!0.003290\) | \(0.826828\!\pm\!0.000763\) | \(0.781763\!\pm\!0.001505\) | \(0.824233\!\pm\!0.000736\) |
2pt
The experiments support the theoretical claim in a specific and limited sense. They do not show that the full V-GIB objective is always best. They show that geometry is a consequential representation variable under label scarcity.
On Breast Cancer, geometry-aware variants improve over ERM once the labeled sample is not extremely small. On FashionMNIST, the V-GIB family improves predictive performance while substantially reducing curvature relative to
ERM. On CIFAR-10, the full V-GIB model gives the strongest primary evidence, improving over both ERM and VIB at all tested label fractions. The ablations show that curvature and intrinsic dimension are
not interchangeable. Removing the dimension term can help in very low-label FashionMNIST, while the full model becomes stronger at higher low-label budgets and on CIFAR-10. Removing curvature can ease optimization, as seen in CovType and some low-label
regimes, but it also weakens the structural control that is central to the proposed objective.
These results support the paper’s main claim. Latent geometry is not only a post hoc diagnostic. Under label scarcity, it can be measured, regularized, and used to move the empirical information–geometry frontier. The evidence is strongest on
FashionMNIST and CIFAR-10, where V-GIB improves predictive performance while controlling geometric complexity. The appropriate conclusion is therefore not that V-GIB universally dominates existing methods, but that it is a serious
geometry-aware alternative to standard ERM and variational bottleneck training in data-scarce regimes.
The author received no financial support for the research, authorship, or publication of this article.
The author declares no competing interests.
Not applicable. This study uses publicly available benchmark datasets and does not involve human participants, animal subjects, or newly collected personal data.
Not applicable.
The datasets used in this study are publicly available benchmark datasets. The primary experiments use the Breast Cancer Wisconsin Diagnostic dataset, FashionMNIST, CIFAR-10, and CovType. The processed experimental summaries used to generate the reported tables and figures are available through the project repository.
Not applicable.
The code used to run the experiments and generate the reported tables and figures is publicly available at https://github.com/karjxenval/V-GIB.
R.K. conceived the study, developed the theoretical framework, proved the main results, designed and performed the experiments, analyzed and interpreted the results, and wrote the manuscript.
This appendix gives the proofs for the geometry-sensitive results in Sec. 2. The statements are conditional on Assumptions 1 and 2.
Proof of Lemma 1. By Assumption 1, for each feasible encoder \(\phi\in\Phi\), the image \(\phi(\mathcal{M})\) is contained in a compact \(d_{\mathrm{int}}(\phi)\)-dimensional \(C^2\) submanifold \(\mathcal{N}_\phi\subset\mathbb{R}^m\). This submanifold has reach \(\tau_\phi>0\), sectional curvature bounded by \(\kappa_{\phi,\max}\), and finite \(d_{\mathrm{int}}(\phi)\)-dimensional volume \[V_\phi=\operatorname{Vol}_{d_{\mathrm{int}}(\phi)}(\mathcal{N}_\phi)<\infty.\]
For compact \(C^2\) submanifolds with positive reach, bounded curvature, and finite volume, standard manifold covering estimates imply that there is a constant \[C_\phi = C\bigl(d_{\mathrm{int}}(\phi),\tau_\phi,\kappa_{\phi,\max},V_\phi\bigr)>0\] such that, for all sufficiently small \(\varepsilon>0\), \[\mathcal{N}\!\bigl(\varepsilon,\mathcal{N}_\phi,\|\cdot\|_2\bigr) \le C_\phi \varepsilon^{-d_{\mathrm{int}}(\phi)}.\] The exponent is the intrinsic dimension of the latent manifold, while the constant absorbs the effects of reach, curvature, and volume.
Since \(\phi(\mathcal{M})\subset \mathcal{N}_\phi\), any \(\varepsilon\)-cover of \(\mathcal{N}_\phi\) is also an \(\varepsilon\)-cover of \(\phi(\mathcal{M})\). Therefore \[\mathcal{N}\!\bigl(\varepsilon,\phi(\mathcal{M}),\|\cdot\|_2\bigr) \le \mathcal{N}\!\bigl(\varepsilon,\mathcal{N}_\phi,\|\cdot\|_2\bigr) \le C_\phi \varepsilon^{-d_{\mathrm{int}}(\phi)}\] for all sufficiently small \(\varepsilon>0\). This proves the claim. ◻
Proof of Theorem 1. We use a standard empirical-process argument. For a bounded loss class \(\mathcal{F}_\phi\), Dudley’s entropy integral and standard Rademacher-complexity bounds give \[\widehat{\mathfrak R}_N(\mathcal{F}_{\phi}) \le \frac{12}{\sqrt N} \int_0^B \sqrt{ \log \mathcal{N}\!\bigl(\eta,\mathcal{F}_{\phi},L_2(P_N)\bigr) }\, d\eta\] up to universal constants [23], [24].
By Assumption 2, for every empirical measure \(P_N\) and every \(\eta\in(0,1]\), \[\log \mathcal{N}\!\bigl(\eta,\mathcal{F}_{\phi},L_2(P_N)\bigr) \le A_0 \left[ d_{\mathrm{int}}(\phi)\log\!\Bigl(\frac{A_1}{\eta}\Bigr) + \log \mathcal{N}\!\bigl(\eta/A_1,\phi(\mathcal{M}),\|\cdot\|_2\bigr) \right].\] Lemma 1 gives, for sufficiently small \(\eta>0\), \[\mathcal{N}\!\bigl(\eta/A_1,\phi(\mathcal{M}),\|\cdot\|_2\bigr) \le C_\phi \left(\frac{A_1}{\eta}\right)^{d_{\mathrm{int}}(\phi)}.\] Taking logarithms, \[\log \mathcal{N}\!\bigl(\eta/A_1,\phi(\mathcal{M}),\|\cdot\|_2\bigr) \le \log C_\phi + d_{\mathrm{int}}(\phi)\log\!\Bigl(\frac{A_1}{\eta}\Bigr).\] Substituting this into the loss-class entropy bound yields \[\log \mathcal{N}\!\bigl(\eta,\mathcal{F}_{\phi},L_2(P_N)\bigr) \le A_0 \left[ 2d_{\mathrm{int}}(\phi)\log\!\Bigl(\frac{A_1}{\eta}\Bigr) + \log C_\phi \right].\] Using \(\log(1+C_\phi)\) instead of \(\log C_\phi\) absorbs the case \(C_\phi<1\) and keeps the bound nonnegative.
Therefore, \[\widehat{\mathfrak R}_N(\mathcal{F}_{\phi}) \le \frac{12\sqrt{A_0}}{\sqrt N} \int_0^B \sqrt{ 2d_{\mathrm{int}}(\phi)\log\!\Bigl(\frac{A_1}{\eta}\Bigr) + \log(1+C_\phi) }\,d\eta .\] The entropy integral is finite because the loss is bounded on \([0,B]\). Standard bounds for logarithmic entropy integrals imply that there exist constants \(C_1',C_2'>0\), depending only on \(A_0,A_1,B\) and universal constants, such that \[\widehat{\mathfrak R}_N(\mathcal{F}_{\phi}) \le C_1' \sqrt{ \frac{ d_{\mathrm{int}}(\phi)\log(C_2' N) + \log(1+C_\phi) }{N} }.\] The factor \(\log(C_2'N)\) is the usual finite-sample upper envelope obtained when the entropy integral is truncated at the empirical resolution scale.
Finally, for bounded losses, a standard concentration inequality gives, with probability at least \(1-\delta\), \[\sup_{f\in\mathcal{F}_\phi} \bigl(R(f)-\widehat R_N(f)\bigr) \le 2\widehat{\mathfrak R}_N(\mathcal{F}_\phi) + B\sqrt{\frac{\log(2/\delta)}{2N}}.\] Substituting the Rademacher bound and absorbing constants into \(C_1,C_2\) gives \[\sup_{f\in\mathcal{F}_{\phi}} \bigl(R(f)-\widehat R_N(f)\bigr) \le C_1 \sqrt{ \frac{ d_{\mathrm{int}}(\phi)\log(C_2 N) + \log(1+C_\phi) + \log(2/\delta) }{N} }.\] This proves the theorem. ◻
Corollary 1 (Sample-complexity consequence). Under the assumptions of Theorem 1, it is sufficient to take \[N \gtrsim \frac{ d_{\mathrm{int}}(\phi)\log\!\bigl(C/\varepsilon\bigr) + \log(1+C_\phi) + \log(2/\delta) }{\varepsilon^2}\] to guarantee \[\sup_{f\in\mathcal{F}_{\phi}} \bigl(R(f)-\widehat R_N(f)\bigr) \le \varepsilon\] with probability at least \(1-\delta\).
Proof. Theorem 1 gives \[\sup_{f\in\mathcal{F}_{\phi}} \bigl(R(f)-\widehat R_N(f)\bigr) \le C_1 \sqrt{ \frac{ d_{\mathrm{int}}(\phi)\log(C_2 N) + \log(1+C_\phi) + \log(2/\delta) }{N} }.\] To make the right-hand side at most \(\varepsilon\), it is enough that \[C_1^2 \frac{ d_{\mathrm{int}}(\phi)\log(C_2 N) + \log(1+C_\phi) + \log(2/\delta) }{N} \le \varepsilon^2.\] Solving this inequality up to absolute constants and replacing the logarithmic dependence on \(N\) by the standard sufficient envelope \(\log(C/\varepsilon)\) yields \[N \gtrsim \frac{ d_{\mathrm{int}}(\phi)\log(C/\varepsilon) + \log(1+C_\phi) + \log(2/\delta) }{\varepsilon^2}.\] This proves the stated sufficient condition. ◻
This appendix gives the proofs for Theorems 2 and 3, together with the curvature concentration statement used in the surrogate analysis.
Proof of Theorem 2. For fixed \(\gamma\), define \[G_\gamma(\phi) = \mathcal{C}(\phi)+\gamma d_{\mathrm{int}}(\phi).\] For each \(\beta\ge 0\), the selected representation \(\phi_\beta\) solves \[\phi_{\beta} \in \arg\max_{\phi\in\Phi(R_0)} \left\{ I(\phi(X);Y)-\beta G_\gamma(\phi) \right\}.\] By assumption, this maximizer is unique, the path \(\beta\mapsto\phi_\beta\) is differentiable, and the maps \[\phi\mapsto I(\phi(X);Y), \qquad \phi\mapsto G_\gamma(\phi)\] are differentiable along this path.
Define the value function \[J(\beta) = I(\phi_\beta(X);Y)-\beta G_\gamma(\phi_\beta).\] By the envelope theorem, differentiating the optimized value with respect to the scalar parameter \(\beta\) gives \[J'(\beta) = -\,G_\gamma(\phi_\beta).\] The dependence of \(\phi_\beta\) on \(\beta\) does not contribute an additional first-order term because \(\phi_\beta\) is the optimizer of the objective at that value of \(\beta\).
Now differentiate the same expression directly along the differentiable optimizer path: \[J'(\beta) = \frac{d}{d\beta} I(\phi_\beta(X);Y) - G_\gamma(\phi_\beta) - \beta\frac{d}{d\beta}G_\gamma(\phi_\beta).\] Equating the two expressions for \(J'(\beta)\) gives \[-\,G_\gamma(\phi_\beta) = \frac{d}{d\beta} I(\phi_\beta(X);Y) - G_\gamma(\phi_\beta) - \beta\frac{d}{d\beta}G_\gamma(\phi_\beta).\] Canceling the common term \(-G_\gamma(\phi_\beta)\) yields \[\frac{d}{d\beta} I(\phi_\beta(X);Y) = \beta\frac{d}{d\beta}G_\gamma(\phi_\beta).\] Whenever \[\frac{d}{d\beta}G_\gamma(\phi_\beta)\neq 0,\] we may divide both sides by this derivative to obtain \[\frac{d\, I(\phi_\beta(X);Y)}{d\,G_\gamma(\phi_\beta)} = \beta.\] This proves the slope identity.
If \[\frac{d}{d\beta}G_\gamma(\phi_\beta)<0,\] then increasing \(\beta\) decreases the selected geometric complexity. Since \[\frac{d}{d\beta} I(\phi_\beta(X);Y) = \beta\frac{d}{d\beta}G_\gamma(\phi_\beta),\] and \(\beta\ge0\), the retained information is nonincreasing as \(\beta\) increases, and strictly decreasing whenever \(\beta>0\) and \(dG_\gamma(\phi_\beta)/d\beta<0\). Thus the selected solutions trace a trade-off frontier between retained information and geometric complexity. This proves the theorem. ◻
The following result is a fixed-encoder concentration statement for the stochastic curvature estimator. The uniform surrogate consistency theorem in the main text assumes the corresponding uniform control over \(\Phi\). Such uniform control can be obtained under additional complexity conditions on the encoder class, or imposed directly as in Theorem 3.
Theorem 4 (Fixed-encoder curvature concentration). Let \(X_1,\dots,X_N\) be i.i.d. samples from \(P_X\). For each \(i\), let \(V_{i,1},\dots,V_{i,K}\) be independent probe vectors, independent of the samples. Assume the probes are normalized so that, for every \(x\), \[\mathbb{E}_V\bigl[\|(\nabla^2\phi(x))V\|_2^2\bigr] = \|\nabla^2\phi(x)\|_F^2.\] Define \[h_\phi(X;V) = \|(\nabla^2\phi(X))V\|_2^2\] and \[\widehat{\mathcal{C}}_{N,K}(\phi) = \frac{1}{NK} \sum_{i=1}^N \sum_{k=1}^K h_\phi(X_i;V_{i,k}).\] Assume that, for fixed \(\phi\), the centered random variables \[h_\phi(X;V)-\mathbb{E}[h_\phi(X;V)]\] are sub-Gaussian with proxy variance \(\sigma_H^2\). Then, for any \(t>0\), \[\Pr\!\left( \bigl| \widehat{\mathcal{C}}_{N,K}(\phi) - \mathcal{C}(\phi) \bigr| \ge t \right) \le 2\exp\!\left( -\frac{NKt^2}{2\sigma_H^2} \right).\] Equivalently, with probability at least \(1-\delta\), \[\bigl| \widehat{\mathcal{C}}_{N,K}(\phi) - \mathcal{C}(\phi) \bigr| \le \sigma_H \sqrt{ \frac{2\log(2/\delta)}{NK} }.\]
Proof. By definition, \[\mathcal{C}(\phi) = \mathbb{E}_X\bigl[\|\nabla^2\phi(X)\|_F^2\bigr].\] Using the probe normalization, \[\mathbb{E}_V\bigl[h_\phi(X;V)\mid X\bigr] = \mathbb{E}_V\bigl[\|(\nabla^2\phi(X))V\|_2^2\mid X\bigr] = \|\nabla^2\phi(X)\|_F^2.\] Taking expectation over \(X\) gives \[\mathbb{E}_{X,V}[h_\phi(X;V)] = \mathbb{E}_X\bigl[\|\nabla^2\phi(X)\|_F^2\bigr] = \mathcal{C}(\phi).\] Therefore \[\mathbb{E}[\widehat{\mathcal{C}}_{N,K}(\phi)] = \mathcal{C}(\phi).\]
The variables \[h_\phi(X_i;V_{i,k}), \qquad i=1,\ldots,N,\quad k=1,\ldots,K,\] are independent by the independence of the samples and probes. By assumption, their centered versions are sub-Gaussian with proxy variance \(\sigma_H^2\). The average of \(NK\) independent centered sub-Gaussian variables is sub-Gaussian with proxy variance \(\sigma_H^2/(NK)\). Hence \[\Pr\!\left( \left| \widehat{\mathcal{C}}_{N,K}(\phi) - \mathbb{E}[\widehat{\mathcal{C}}_{N,K}(\phi)] \right| \ge t \right) \le 2\exp\!\left( -\frac{NKt^2}{2\sigma_H^2} \right).\] Since \[\mathbb{E}[\widehat{\mathcal{C}}_{N,K}(\phi)] = \mathcal{C}(\phi),\] the first claim follows. Setting the right-hand side equal to \(\delta\) and solving for \(t\) gives \[t = \sigma_H \sqrt{ \frac{2\log(2/\delta)}{NK} },\] which proves the high-probability form. ◻
Proof of Theorem 3. Recall the population score \[Q_{\beta,\gamma}(\phi) = I(\phi(X);Y) - \beta\mathcal{C}(\phi) - \gamma d_{\mathrm{int}}(\phi),\] and the empirical surrogate \[\widehat Q_{N,K}(\phi) = \widehat I_N(\phi) - \beta\widehat{\mathcal{C}}_{N,K}(\phi) - \gamma\widehat d_N(\phi).\] For any fixed \(\phi\in\Phi\), the triangle inequality gives \[\bigl| \widehat Q_{N,K}(\phi)-Q_{\beta,\gamma}(\phi) \bigr| \le \bigl| \widehat I_N(\phi)-I(\phi(X);Y) \bigr| + \beta \bigl| \widehat{\mathcal{C}}_{N,K}(\phi)-\mathcal{C}(\phi) \bigr| + \gamma \bigl| \widehat d_N(\phi)-d_{\mathrm{int}}(\phi) \bigr|.\] Taking suprema over \(\phi\in\Phi\) yields \[\sup_{\phi\in\Phi} \bigl| \widehat Q_{N,K}(\phi)-Q_{\beta,\gamma}(\phi) \bigr| \le \sup_{\phi\in\Phi} \bigl| \widehat I_N(\phi)-I(\phi(X);Y) \bigr| + \beta \sup_{\phi\in\Phi} \bigl| \widehat{\mathcal{C}}_{N,K}(\phi)-\mathcal{C}(\phi) \bigr| + \gamma \sup_{\phi\in\Phi} \bigl| \widehat d_N(\phi)-d_{\mathrm{int}}(\phi) \bigr|.\] By Assumption 3, \[\sup_{\phi\in\Phi} \bigl| \widehat I_N(\phi)-I(\phi(X);Y) \bigr| = O_p(r_I(N)),\] and \[\sup_{\phi\in\Phi} \bigl| \widehat d_N(\phi)-d_{\mathrm{int}}(\phi) \bigr| = O_p(r_d(N)).\] By the additional uniform curvature-control condition in Theorem 3, \[\sup_{\phi\in\Phi} \bigl| \widehat{\mathcal{C}}_{N,K}(\phi)-\mathcal{C}(\phi) \bigr| = O_p((NK)^{-1/2}).\] Combining the three bounds gives \[\sup_{\phi\in\Phi} \bigl| \widehat Q_{N,K}(\phi)-Q_{\beta,\gamma}(\phi) \bigr| = O_p\!\left( r_I(N) + (NK)^{-1/2} + r_d(N) \right).\] This proves uniform convergence of the empirical surrogate to the population score at the stated rate.
It remains to prove the argmax claim. Let \[\Delta_{N,K} = \sup_{\phi\in\Phi} \bigl| \widehat Q_{N,K}(\phi)-Q_{\beta,\gamma}(\phi) \bigr|.\] The first part of the theorem shows that \(\Delta_{N,K}\to0\) in probability.
Let \(\widehat\phi_{N,K}\in\arg\max_{\phi\in\Phi}\widehat Q_{N,K}(\phi)\). Since \(\phi^\star\) maximizes \(Q_{\beta,\gamma}\), and \(\widehat\phi_{N,K}\) maximizes \(\widehat Q_{N,K}\), \[\widehat Q_{N,K}(\widehat\phi_{N,K}) \ge \widehat Q_{N,K}(\phi^\star).\] Then \[Q_{\beta,\gamma}(\phi^\star) - Q_{\beta,\gamma}(\widehat\phi_{N,K}) \le \bigl| Q_{\beta,\gamma}(\phi^\star)-\widehat Q_{N,K}(\phi^\star) \bigr| + \bigl| \widehat Q_{N,K}(\widehat\phi_{N,K}) - Q_{\beta,\gamma}(\widehat\phi_{N,K}) \bigr| \le 2\Delta_{N,K}.\] Since \(\Delta_{N,K}\to0\) in probability, \[Q_{\beta,\gamma}(\widehat\phi_{N,K}) \to Q_{\beta,\gamma}(\phi^\star)\] in probability.
Now assume that \(\Phi\) is a compact metric space, \(Q_{\beta,\gamma}\) is continuous, and \(\phi^\star\) is its unique maximizer. Let \(U\) be any open neighborhood of \(\phi^\star\). Since \(\Phi\setminus U\) is compact and does not contain the unique maximizer, continuity and uniqueness imply that there exists \(\eta_U>0\) such that \[Q_{\beta,\gamma}(\phi^\star) - \sup_{\phi\in\Phi\setminus U}Q_{\beta,\gamma}(\phi) \ge \eta_U.\] If \(\widehat\phi_{N,K}\notin U\), then \[Q_{\beta,\gamma}(\phi^\star) - Q_{\beta,\gamma}(\widehat\phi_{N,K}) \ge \eta_U.\] But we have already shown that this difference is at most \(2\Delta_{N,K}\). Therefore \[\Pr(\widehat\phi_{N,K}\notin U) \le \Pr(2\Delta_{N,K}\ge \eta_U) \to 0.\] Since this holds for every open neighborhood \(U\) of \(\phi^\star\), we conclude that \[\widehat\phi_{N,K}\to \phi^\star\] in probability. This completes the proof. ◻
The curvature proxy is computed from the sensitivity of the latent mean \(\mu(x)\) to the input. For a random probe vector \(r\), the default stochastic Jacobian proxy computes \[\left\|\nabla_x \langle \mu(x),r\rangle\right\|_2^2\] and averages over the batch and probes. A Hessian-based proxy is implemented, but the default experiments use the Jacobian proxy for stability and computational cost.
The intrinsic-dimension proxy is the participation ratio of the latent covariance. If \(\lambda_1,\ldots,\lambda_{d_z}\) are the eigenvalues of the empirical latent covariance, then \[d_{\mathrm{PR}} = \frac{\left(\sum_j \lambda_j\right)^2}{\sum_j \lambda_j^2}, \qquad \text{dimension ratio} = \frac{d_{\mathrm{PR}}}{d_z}.\] All geometry diagnostics are computed from \(\mu(x)\), not sampled latent variables. For computational stability, geometry penalties are evaluated on a capped sub-batch of \(32\) examples per training step.
All models are trained with AdamW using learning rate \(10^{-3}\), weight decay \(10^{-4}\), batch size \(128\), and gradient clipping at norm \(5.0\). Models are trained for at most \(25\) epochs with early stopping on validation performance using patience \(5\) and minimum improvement \(10^{-4}\).
The default regularization weights are \[\beta_{\mathrm{KL}}=10^{-3}, \qquad \beta_{\mathrm{curv}}=10^{-3}, \qquad \gamma_{\mathrm{dim}}=10^{-2}.\] The KL, curvature, and dimension penalties are linearly warmed up during the first \(5\) epochs.
Checkpoints are selected by predictive validation performance rather than by the geometry-based utility score. In automatic mode, binary tasks use validation AUROC and multiclass tasks use validation macro-F1. Ties are broken by utility proxy, then lower curvature proxy, then lower dimension ratio. Final validation and test metrics are computed from the selected checkpoint.
Predictive metrics include accuracy, balanced accuracy, macro-F1, AUROC, and cross-entropy. Binary AUROC uses the positive-class probability, while multiclass AUROC uses macro one-versus-rest scoring.
The information proxy is \[\widehat I_{\mathrm{proxy}} = \widehat H(Y_{\mathrm{train}}) - \widehat{\mathcal{L}}_{\mathrm{CE}},\] where \(\widehat H(Y_{\mathrm{train}})\) is the empirical label entropy of the labeled training subset and \(\widehat{\mathcal{L}}_{\mathrm{CE}}\) is the validation or test cross-entropy.
The utility proxy is \[\widehat U = \widehat I_{\mathrm{proxy}} - \beta_{\mathrm{KL}}\widehat{\mathcal{L}}_{\mathrm{KL}} - \beta_{\mathrm{curv}}\widehat{\mathcal{L}}_{\mathrm{curv}} - \gamma_{\mathrm{dim}}\widehat{\mathcal{L}}_{\mathrm{dim}}.\] Interpretive efficiency is \[\widehat E = \frac{\widehat U}{N_{\mathrm{lab}}},\] where \(N_{\mathrm{lab}}\) is the number of labeled training examples used in the run.
For every dataset, method, label fraction, and seed, the implementation saves the training history, selected checkpoint, configuration, validation and test metrics, confusion matrix, and classification report. The runner also writes aggregate files
containing individual runs, mean and standard-deviation summaries, and pairwise comparisons against ERM and VIB. These summaries are used to produce the final tables and figures.
This appendix provides the extended experimental summaries supporting Section 3. For each dataset, we report the additional predictive metrics, geometry diagnostics, utility proxies, scatter diagnostics, and pairwise comparisons that were omitted from the main text for space. Unless otherwise stated, all values are computed on the test split and reported as mean \(\pm\) standard deviation over seeds \(13,29,47\). The geometric diagnostics are used to interpret the information–geometry tradeoff and should not be read as direct substitutes for predictive performance.
The Breast Cancer benchmark is a small-data tabular setting. The \(1\%\) split contains only four labeled training examples, so the corresponding results are unstable and are included mainly for completeness. The more
informative comparisons are from \(5\%\) labels onward, where the geometry-aware variants improve substantially over ERM in accuracy and macro-F1. The complete predictive and geometric summary is reported in
Table 5, while the pairwise differences for the full V-GIB model are reported in Table 6.
Figure 4 reports the Breast Cancer geometry and utility diagnostics. The curvature proxy, dimension ratio, and utility proxy are shown separately in Figures 4 (a), 4 (b), and 4 (c). These diagnostics help interpret how the geometric objective changes the learned representation across label fractions.
Figure 4: Geometry and utility diagnostics on the Breast Cancer benchmark. The diagnostics help interpret the information–geometry tradeoff, but they do not reduce predictive performance to a single geometric statistic.. a — Curvature proxy., b — Dimension ratio., c — Utility proxy.
Figure 5 gives the corresponding scatter diagnostics. The three panels compare accuracy with curvature, dimension ratio, and utility proxy, respectively; see Figures 5 (a), 5 (b), and 5 (c). These plots show that the predictive gains are not explained by a single monotone geometric statistic across all label fractions.
Figure 5: Diagnostic scatter plots for Breast Cancer. The predictive gains are not explained by a single monotone geometric statistic across all label fractions, which is consistent with the paper’s claim that geometry is a controllable representation axis rather than a universal scalar predictor of accuracy.. a — Accuracy vs. curvature., b — Accuracy vs. dimension ratio., c — Accuracy vs. utility proxy.
Table 5 reports the full Breast Cancer test summary across all label fractions and methods. It includes accuracy, macro-F1, AUROC, curvature, dimension ratio, and utility, so that the predictive and geometric behavior can be read together.
| Fraction | Method | Accuracy | Macro-F1 | AUROC | Curvature | Dim. ratio | Utility |
|---|---|---|---|---|---|---|---|
| 0.01 | ERM | \(0.8333\!\pm\!0.0780\) | \(0.8058\!\pm\!0.0965\) | \(0.9199\!\pm\!0.0501\) | \(3.8257\!\pm\!2.6345\) | \(0.1277\!\pm\!0.0317\) | \(0.1215\!\pm\!0.1791\) |
| 0.01 | VIB | \(0.8333\!\pm\!0.0780\) | \(0.8020\!\pm\!0.0995\) | \(0.9005\!\pm\!0.0622\) | \(3.6907\!\pm\!2.9739\) | \(0.1311\!\pm\!0.0465\) | \(0.0679\!\pm\!0.1707\) |
| 0.01 | V-GIB | \(0.8099\!\pm\!0.0883\) | \(0.7812\!\pm\!0.1074\) | \(0.8628\!\pm\!0.1088\) | \(8.9016\!\pm\!9.7932\) | \(0.1700\!\pm\!0.0813\) | \(0.0381\!\pm\!0.1901\) |
| 0.01 | V-GIB-no-curv | \(0.8246\!\pm\!0.0575\) | \(0.8017\!\pm\!0.0694\) | \(0.8865\!\pm\!0.0675\) | \(7.7434\!\pm\!9.8594\) | \(0.1674\!\pm\!0.0771\) | \(0.0388\!\pm\!0.1956\) |
| 0.01 | V-GIB-no-dim | \(0.8070\!\pm\!0.0890\) | \(0.7771\!\pm\!0.1085\) | \(0.8625\!\pm\!0.1090\) | \(8.9447\!\pm\!9.8185\) | \(0.1701\!\pm\!0.0818\) | \(0.0394\!\pm\!0.1895\) |
| 0.05 | ERM | \(0.8158\!\pm\!0.1412\) | \(0.7960\!\pm\!0.1467\) | \(0.9835\!\pm\!0.0198\) | \(0.4390\!\pm\!0.3030\) | \(0.1551\!\pm\!0.0597\) | \(0.0350\!\pm\!0.0733\) |
| 0.05 | VIB | \(0.9006\!\pm\!0.0681\) | \(0.8831\!\pm\!0.0890\) | \(0.9896\!\pm\!0.0114\) | \(0.6833\!\pm\!0.4925\) | \(0.1147\!\pm\!0.0365\) | \(0.0662\!\pm\!0.0777\) |
| 0.05 | V-GIB | \(0.8977\!\pm\!0.0355\) | \(0.8849\!\pm\!0.0480\) | \(0.9848\!\pm\!0.0158\) | \(8.3622\!\pm\!13.2866\) | \(0.1149\!\pm\!0.0462\) | \(0.1683\!\pm\!0.2300\) |
| 0.05 | V-GIB-no-curv | \(0.9006\!\pm\!0.0396\) | \(0.8880\!\pm\!0.0523\) | \(0.9856\!\pm\!0.0164\) | \(1.9240\!\pm\!2.1502\) | \(0.1181\!\pm\!0.0435\) | \(0.1123\!\pm\!0.1367\) |
| 0.05 | V-GIB-no-dim | \(0.8977\!\pm\!0.0355\) | \(0.8849\!\pm\!0.0480\) | \(0.9846\!\pm\!0.0156\) | \(8.2780\!\pm\!13.1368\) | \(0.1154\!\pm\!0.0465\) | \(0.1693\!\pm\!0.2295\) |
| 0.10 | ERM | \(0.8538\!\pm\!0.1009\) | \(0.8426\!\pm\!0.1012\) | \(0.9889\!\pm\!0.0113\) | \(4.1718\!\pm\!6.7965\) | \(0.1238\!\pm\!0.0572\) | \(0.2000\!\pm\!0.3188\) |
| 0.10 | VIB | \(0.9094\!\pm\!0.0704\) | \(0.9033\!\pm\!0.0718\) | \(0.9901\!\pm\!0.0113\) | \(0.7589\!\pm\!0.6344\) | \(0.1191\!\pm\!0.0681\) | \(0.1050\!\pm\!0.1128\) |
| 0.10 | V-GIB | \(0.9094\!\pm\!0.0571\) | \(0.8970\!\pm\!0.0716\) | \(0.9893\!\pm\!0.0106\) | \(0.6934\!\pm\!0.4202\) | \(0.1297\!\pm\!0.0561\) | \(0.0511\!\pm\!0.0561\) |
| 0.10 | V-GIB-no-curv | \(0.9094\!\pm\!0.0571\) | \(0.8970\!\pm\!0.0716\) | \(0.9893\!\pm\!0.0106\) | \(0.6919\!\pm\!0.4136\) | \(0.1303\!\pm\!0.0563\) | \(0.0512\!\pm\!0.0559\) |
| 0.10 | V-GIB-no-dim | \(0.9094\!\pm\!0.0571\) | \(0.8971\!\pm\!0.0718\) | \(0.9899\!\pm\!0.0109\) | \(4.8845\!\pm\!7.6547\) | \(0.1222\!\pm\!0.0656\) | \(0.1777\!\pm\!0.2647\) |
| 0.20 | ERM | \(0.8918\!\pm\!0.1229\) | \(0.8886\!\pm\!0.1214\) | \(0.9874\!\pm\!0.0159\) | \(1.6132\!\pm\!2.1743\) | \(0.1139\!\pm\!0.0593\) | \(0.1967\!\pm\!0.2730\) |
| 0.20 | VIB | \(0.9269\!\pm\!0.0571\) | \(0.9164\!\pm\!0.0688\) | \(0.9910\!\pm\!0.0061\) | \(0.9594\!\pm\!0.9239\) | \(0.0916\!\pm\!0.0232\) | \(0.1336\!\pm\!0.1497\) |
| 0.20 | V-GIB | \(0.9298\!\pm\!0.0575\) | \(0.9263\!\pm\!0.0581\) | \(0.9856\!\pm\!0.0156\) | \(2.4191\!\pm\!2.8805\) | \(0.1167\!\pm\!0.0582\) | \(0.1629\!\pm\!0.2026\) |
| 0.20 | V-GIB-no-curv | \(0.9474\!\pm\!0.0304\) | \(0.9423\!\pm\!0.0336\) | \(0.9909\!\pm\!0.0066\) | \(2.6412\!\pm\!2.9100\) | \(0.0932\!\pm\!0.0196\) | \(0.1814\!\pm\!0.1828\) |
| 0.20 | V-GIB-no-dim | \(0.9211\!\pm\!0.0464\) | \(0.9171\!\pm\!0.0460\) | \(0.9856\!\pm\!0.0156\) | \(2.1410\!\pm\!2.4047\) | \(0.1175\!\pm\!0.0579\) | \(0.1365\!\pm\!0.1580\) |
3pt
Table 6 reports the pairwise differences between the full V-GIB model and the two main baselines, ERM and VIB. Positive entries indicate that
V-GIB is higher than the comparator on the corresponding metric.
| \(\Delta\) vs. ERM | \(\Delta\) vs. VIB | |||||||
|---|---|---|---|---|---|---|---|---|
| 2-5 (lr)6-9 Fraction | Accuracy | Macro-F1 | AUROC | Utility | Accuracy | Macro-F1 | AUROC | Utility |
| 0.01 | \(-0.0234\) | \(-0.0246\) | \(-0.0571\) | \(-0.0834\) | \(-0.0234\) | \(-0.0208\) | \(-0.0377\) | \(-0.0298\) |
| 0.05 | \(+0.0819\) | \(+0.0890\) | \(+0.0013\) | \(+0.1332\) | \(-0.0029\) | \(+0.0018\) | \(-0.0049\) | \(+0.1020\) |
| 0.10 | \(+0.0556\) | \(+0.0544\) | \(+0.0004\) | \(-0.1489\) | \(+0.0000\) | \(-0.0064\) | \(-0.0008\) | \(-0.0540\) |
| 0.20 | \(+0.0380\) | \(+0.0376\) | \(-0.0019\) | \(-0.0338\) | \(+0.0029\) | \(+0.0099\) | \(-0.0054\) | \(+0.0293\) |
FashionMNIST provides a clear geometry-sensitive image benchmark. The additional diagnostics show that the main improvement is not only predictive. The full V-GIB model and the no-dimension ablation keep curvature substantially lower than
ERM, VIB, and the no-curvature ablation, while remaining among the best predictive methods. The additional FashionMNIST diagnostics are shown in Figure 6, with macro-F1 in Figure 6 (a), dimension ratio in Figure 6 (b), and scatter diagnostics in Figures 6 (c), 6 (d), and 6 (e). The predictive metrics are reported in Table 7, and the geometric and
utility diagnostics are reported in Table 8.
Figure 6: Additional FashionMNIST diagnostics. Macro-F1 tracks the accuracy curves closely. The strongest method separation is associated with curvature control rather than tighter dimension compression. The full V-GIB model and
the no-dimension ablation occupy the low-curvature, high-accuracy region, while the no-curvature ablation moves toward the high-curvature regime despite remaining competitive in accuracy.. a — Test macro-F1., b — Test dimension ratio., c — Accuracy vs.
curvature., d — Accuracy vs. dimension ratio., e — Accuracy vs. utility proxy.
Table 7 reports the FashionMNIST predictive metrics.
| Label fraction | Method | Test accuracy | Test macro-F1 | Test AUROC |
|---|---|---|---|---|
| 1% | ERM | 0.6729 \(\pm\) 0.0599 | 0.6637 \(\pm\) 0.0659 | 0.9468 \(\pm\) 0.0134 |
| VIB | 0.6734 \(\pm\) 0.0107 | 0.6649 \(\pm\) 0.0044 | 0.9479 \(\pm\) 0.0043 | |
| V-GIB | 0.6928 \(\pm\) 0.0100 | 0.6878 \(\pm\) 0.0114 | 0.9502 \(\pm\) 0.0016 | |
| V-GIB-no-curv | 0.6822 \(\pm\) 0.0262 | 0.6704 \(\pm\) 0.0321 | 0.9490 \(\pm\) 0.0036 | |
| V-GIB-no-dim | 0.6978 \(\pm\) 0.0117 | 0.6934 \(\pm\) 0.0138 | 0.9506 \(\pm\) 0.0018 | |
| 5% | ERM | 0.7409 \(\pm\) 0.0115 | 0.7346 \(\pm\) 0.0151 | 0.9647 \(\pm\) 0.0021 |
| VIB | 0.7831 \(\pm\) 0.0374 | 0.7765 \(\pm\) 0.0476 | 0.9725 \(\pm\) 0.0072 | |
| V-GIB | 0.7889 \(\pm\) 0.0271 | 0.7858 \(\pm\) 0.0248 | 0.9739 \(\pm\) 0.0062 | |
| V-GIB-no-curv | 0.7811 \(\pm\) 0.0005 | 0.7739 \(\pm\) 0.0027 | 0.9730 \(\pm\) 0.0010 | |
| V-GIB-no-dim | 0.7894 \(\pm\) 0.0258 | 0.7862 \(\pm\) 0.0230 | 0.9740 \(\pm\) 0.0061 | |
| 10% | ERM | 0.8261 \(\pm\) 0.0030 | 0.8267 \(\pm\) 0.0050 | 0.9827 \(\pm\) 0.0003 |
| VIB | 0.8341 \(\pm\) 0.0105 | 0.8327 \(\pm\) 0.0094 | 0.9835 \(\pm\) 0.0014 | |
| V-GIB | 0.8412 \(\pm\) 0.0098 | 0.8411 \(\pm\) 0.0117 | 0.9840 \(\pm\) 0.0015 | |
| V-GIB-no-curv | 0.8142 \(\pm\) 0.0111 | 0.8150 \(\pm\) 0.0127 | 0.9794 \(\pm\) 0.0019 | |
| V-GIB-no-dim | 0.8415 \(\pm\) 0.0069 | 0.8415 \(\pm\) 0.0090 | 0.9842 \(\pm\) 0.0013 | |
| 20% | ERM | 0.8192 \(\pm\) 0.0221 | 0.8213 \(\pm\) 0.0182 | 0.9813 \(\pm\) 0.0032 |
| VIB | 0.8343 \(\pm\) 0.0148 | 0.8317 \(\pm\) 0.0120 | 0.9840 \(\pm\) 0.0034 | |
| V-GIB | 0.8692 \(\pm\) 0.0098 | 0.8679 \(\pm\) 0.0116 | 0.9890 \(\pm\) 0.0005 | |
| V-GIB-no-curv | 0.8492 \(\pm\) 0.0203 | 0.8477 \(\pm\) 0.0215 | 0.9851 \(\pm\) 0.0038 | |
| V-GIB-no-dim | 0.8679 \(\pm\) 0.0081 | 0.8671 \(\pm\) 0.0095 | 0.9891 \(\pm\) 0.0006 |
Table 8 reports the FashionMNIST geometric and utility diagnostics.
| Label fraction | Method | Test curvature proxy | Test dimension ratio | Test utility proxy | Best epoch |
|---|---|---|---|---|---|
| 1% | ERM | \(79.9008\!\pm\!42.0998\) | \(0.0329\!\pm\!0.0050\) | \(1.3773\!\pm\!0.1934\) | \(14.3333\!\pm\!4.6188\) |
| VIB | \(96.2153\!\pm\!19.0023\) | \(0.0351\!\pm\!0.0052\) | \(1.3383\!\pm\!0.0590\) | \(17.6667\!\pm\!2.0817\) | |
| V-GIB | \(39.8209\!\pm\!9.3773\) | \(0.0362\!\pm\!0.0022\) | \(1.3490\!\pm\!0.0193\) | \(19.0000\!\pm\!1.0000\) | |
| V-GIB-no-curv | \(105.0087\!\pm\!20.9835\) | \(0.0374\!\pm\!0.0049\) | \(1.3629\!\pm\!0.0796\) | \(19.3333\!\pm\!0.5774\) | |
| V-GIB-no-dim | \(38.9628\!\pm\!7.2261\) | \(0.0372\!\pm\!0.0033\) | \(1.3637\!\pm\!0.0250\) | \(19.3333\!\pm\!0.5774\) | |
| 5% | ERM | \(114.4583\!\pm\!32.8709\) | \(0.0358\!\pm\!0.0045\) | \(1.5746\!\pm\!0.0378\) | \(9.0000\!\pm\!2.6458\) |
| VIB | \(170.6544\!\pm\!48.0709\) | \(0.0478\!\pm\!0.0101\) | \(1.6552\!\pm\!0.0813\) | \(14.0000\!\pm\!5.0000\) | |
| V-GIB | \(34.6334\!\pm\!2.1547\) | \(0.0464\!\pm\!0.0031\) | \(1.6531\!\pm\!0.0663\) | \(15.3333\!\pm\!4.7258\) | |
| V-GIB-no-curv | \(154.3777\!\pm\!9.0674\) | \(0.0464\!\pm\!0.0037\) | \(1.6608\!\pm\!0.0018\) | \(12.3333\!\pm\!1.5275\) | |
| V-GIB-no-dim | \(34.7127\!\pm\!2.1710\) | \(0.0465\!\pm\!0.0029\) | \(1.6530\!\pm\!0.0653\) | \(15.3333\!\pm\!4.7258\) | |
| 10% | ERM | \(182.2521\!\pm\!35.7148\) | \(0.0458\!\pm\!0.0030\) | \(1.7895\!\pm\!0.0487\) | \(13.6667\!\pm\!3.7859\) |
| VIB | \(210.9575\!\pm\!2.7980\) | \(0.0563\!\pm\!0.0039\) | \(1.7904\!\pm\!0.0299\) | \(14.3333\!\pm\!3.5119\) | |
| V-GIB | \(37.7687\!\pm\!7.3237\) | \(0.0505\!\pm\!0.0057\) | \(1.7655\!\pm\!0.0279\) | \(17.3333\!\pm\!1.5275\) | |
| V-GIB-no-curv | \(159.6611\!\pm\!18.2292\) | \(0.0548\!\pm\!0.0055\) | \(1.7331\!\pm\!0.0223\) | \(9.3333\!\pm\!2.5166\) | |
| V-GIB-no-dim | \(37.5780\!\pm\!6.5192\) | \(0.0512\!\pm\!0.0063\) | \(1.7690\!\pm\!0.0218\) | \(17.3333\!\pm\!1.5275\) | |
| 20% | ERM | \(142.1634\!\pm\!17.7174\) | \(0.0430\!\pm\!0.0061\) | \(1.7912\!\pm\!0.0547\) | \(5.0000\!\pm\!1.0000\) |
| VIB | \(197.2881\!\pm\!24.6936\) | \(0.0608\!\pm\!0.0071\) | \(1.7783\!\pm\!0.0525\) | \(8.6667\!\pm\!3.5119\) | |
| V-GIB | \(30.2081\!\pm\!3.6243\) | \(0.0493\!\pm\!0.0013\) | \(1.8486\!\pm\!0.0167\) | \(14.3333\!\pm\!1.5275\) | |
| V-GIB-no-curv | \(189.2930\!\pm\!60.6088\) | \(0.0531\!\pm\!0.0041\) | \(1.8234\!\pm\!0.0593\) | \(8.6667\!\pm\!3.2146\) | |
| V-GIB-no-dim | \(30.2305\!\pm\!3.0083\) | \(0.0498\!\pm\!0.0020\) | \(1.8493\!\pm\!0.0144\) | \(14.3333\!\pm\!1.5275\) |
4pt
CIFAR-10 is the most challenging primary benchmark in this validation suite. in the validation suite. The additional results show that the full V-GIB model improves predictive performance while keeping curvature substantially lower than
ERM and VIB from \(5\%\) labels onward. This supports the combined information–geometry interpretation rather than an explanation based on prediction alone. The geometry and utility diagnostics are
reported in Figure 7, with the individual curvature, dimension-ratio, and utility panels shown in Figures 7 (a), 7 (b),
and 7 (c). The scatter diagnostics are reported in Figure 8, with the individual panels shown in Figures 8 (a), 8 (b), and 8 (c). The full numerical summary is given in Table 9, and the pairwise differences for the
full V-GIB model are given in Table 10.
Figure 7: Geometry and utility diagnostics on CIFAR-10. The full V-GIB model keeps curvature much lower than ERM and VIB from \(5\%\) labels onward while maintaining
stronger predictive performance at \(10\%\) and \(20\%\).. a — Curvature proxy., b — Dimension ratio., c — Utility proxy.
Figure 8: Diagnostic scatter plots for CIFAR-10. The utility proxy tracks accuracy more clearly than curvature or dimension alone, supporting the use of the combined information–geometry score rather than a single geometric statistic.. a — Accuracy vs. curvature., b — Accuracy vs. dimension ratio., c — Accuracy vs. utility proxy.
Table 9 reports the full CIFAR-10 test summary across all label fractions and methods.
| Fraction | Method | Accuracy | Macro-F1 | AUROC | Curvature | Dim. ratio | Utility |
|---|---|---|---|---|---|---|---|
| 0.01 | ERM | \(0.1323\!\pm\!0.0189\) | \(0.0578\!\pm\!0.0203\) | \(0.6817\!\pm\!0.0100\) | \(0.1488\!\pm\!0.2100\) | \(0.0209\!\pm\!0.0027\) | \(0.0178\!\pm\!0.0223\) |
| 0.01 | VIB | \(0.1947\!\pm\!0.1040\) | \(0.1380\!\pm\!0.1325\) | \(0.6921\!\pm\!0.1027\) | \(16.6281\!\pm\!28.7688\) | \(0.0243\!\pm\!0.0050\) | \(0.1720\!\pm\!0.2936\) |
| 0.01 | V-GIB | \(0.3405\!\pm\!0.0126\) | \(0.3277\!\pm\!0.0162\) | \(0.8197\!\pm\!0.0017\) | \(40.3712\!\pm\!1.2477\) | \(0.0358\!\pm\!0.0019\) | \(0.5000\!\pm\!0.0231\) |
| 0.01 | V-GIB-no-curv | \(0.3409\!\pm\!0.0044\) | \(0.3321\!\pm\!0.0045\) | \(0.8206\!\pm\!0.0043\) | \(125.9102\!\pm\!27.7020\) | \(0.0334\!\pm\!0.0061\) | \(0.5276\!\pm\!0.0132\) |
| 0.01 | V-GIB-no-dim | \(0.3388\!\pm\!0.0113\) | \(0.3283\!\pm\!0.0108\) | \(0.8184\!\pm\!0.0027\) | \(40.4671\!\pm\!1.1849\) | \(0.0370\!\pm\!0.0038\) | \(0.4847\!\pm\!0.0436\) |
| 0.05 | ERM | \(0.4560\!\pm\!0.0147\) | \(0.4422\!\pm\!0.0284\) | \(0.8788\!\pm\!0.0046\) | \(175.4830\!\pm\!56.9196\) | \(0.0480\!\pm\!0.0076\) | \(0.7950\!\pm\!0.0229\) |
| 0.05 | VIB | \(0.4614\!\pm\!0.0064\) | \(0.4531\!\pm\!0.0148\) | \(0.8825\!\pm\!0.0060\) | \(241.1659\!\pm\!50.9281\) | \(0.0452\!\pm\!0.0021\) | \(0.7603\!\pm\!0.0117\) |
| 0.05 | V-GIB | \(0.4625\!\pm\!0.0225\) | \(0.4537\!\pm\!0.0282\) | \(0.8806\!\pm\!0.0092\) | \(42.0389\!\pm\!2.0973\) | \(0.0500\!\pm\!0.0060\) | \(0.7589\!\pm\!0.0393\) |
| 0.05 | V-GIB-no-curv | \(0.4639\!\pm\!0.0344\) | \(0.4565\!\pm\!0.0344\) | \(0.8791\!\pm\!0.0121\) | \(256.4504\!\pm\!20.5095\) | \(0.0462\!\pm\!0.0024\) | \(0.7700\!\pm\!0.0351\) |
| 0.05 | V-GIB-no-dim | \(0.4545\!\pm\!0.0088\) | \(0.4462\!\pm\!0.0142\) | \(0.8798\!\pm\!0.0080\) | \(41.4887\!\pm\!1.6473\) | \(0.0501\!\pm\!0.0059\) | \(0.7430\!\pm\!0.0241\) |
| 0.10 | ERM | \(0.5123\!\pm\!0.0192\) | \(0.5135\!\pm\!0.0167\) | \(0.9014\!\pm\!0.0083\) | \(190.5787\!\pm\!49.6237\) | \(0.0555\!\pm\!0.0033\) | \(0.8845\!\pm\!0.0949\) |
| 0.10 | VIB | \(0.4683\!\pm\!0.0483\) | \(0.4531\!\pm\!0.0585\) | \(0.8857\!\pm\!0.0207\) | \(264.2385\!\pm\!134.3197\) | \(0.0517\!\pm\!0.0068\) | \(0.7803\!\pm\!0.0868\) |
| 0.10 | V-GIB | \(0.5303\!\pm\!0.0069\) | \(0.5298\!\pm\!0.0079\) | \(0.9071\!\pm\!0.0017\) | \(44.6591\!\pm\!3.0361\) | \(0.0563\!\pm\!0.0083\) | \(0.8847\!\pm\!0.0228\) |
| 0.10 | V-GIB-no-curv | \(0.5035\!\pm\!0.0222\) | \(0.4935\!\pm\!0.0305\) | \(0.9005\!\pm\!0.0073\) | \(260.4513\!\pm\!85.7759\) | \(0.0543\!\pm\!0.0046\) | \(0.9122\!\pm\!0.0371\) |
| 0.10 | V-GIB-no-dim | \(0.5254\!\pm\!0.0121\) | \(0.5242\!\pm\!0.0113\) | \(0.9071\!\pm\!0.0027\) | \(44.0562\!\pm\!2.7326\) | \(0.0566\!\pm\!0.0086\) | \(0.8879\!\pm\!0.0392\) |
| 0.20 | ERM | \(0.5269\!\pm\!0.0308\) | \(0.5156\!\pm\!0.0382\) | \(0.9152\!\pm\!0.0127\) | \(195.9634\!\pm\!43.0609\) | \(0.0557\!\pm\!0.0026\) | \(0.9849\!\pm\!0.0534\) |
| 0.20 | VIB | \(0.5531\!\pm\!0.0383\) | \(0.5395\!\pm\!0.0454\) | \(0.9195\!\pm\!0.0082\) | \(396.3132\!\pm\!76.1090\) | \(0.0634\!\pm\!0.0063\) | \(0.9712\!\pm\!0.1258\) |
| 0.20 | V-GIB | \(0.5929\!\pm\!0.0375\) | \(0.5926\!\pm\!0.0335\) | \(0.9275\!\pm\!0.0105\) | \(37.9214\!\pm\!0.7389\) | \(0.0593\!\pm\!0.0060\) | \(1.0726\!\pm\!0.0842\) |
| 0.20 | V-GIB-no-curv | \(0.5675\!\pm\!0.0334\) | \(0.5645\!\pm\!0.0325\) | \(0.9216\!\pm\!0.0068\) | \(338.5462\!\pm\!47.5527\) | \(0.0568\!\pm\!0.0056\) | \(1.0460\!\pm\!0.0649\) |
| 0.20 | V-GIB-no-dim | \(0.5882\!\pm\!0.0398\) | \(0.5879\!\pm\!0.0355\) | \(0.9266\!\pm\!0.0115\) | \(38.0066\!\pm\!0.9320\) | \(0.0596\!\pm\!0.0063\) | \(1.0575\!\pm\!0.0923\) |
3pt
Table 10 reports the pairwise differences for the full V-GIB model on CIFAR-10.
| \(\Delta\) vs. ERM | \(\Delta\) vs. VIB | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| 2-6 (lr)7-11 Fraction | Acc. | Macro-F1 | AUROC | Utility | Curv. | Acc. | Macro-F1 | AUROC | Utility | Curv. |
| 0.01 | \(+0.2082\) | \(+0.2699\) | \(+0.1380\) | \(+0.4822\) | \(+40.2224\) | \(+0.1458\) | \(+0.1897\) | \(+0.1277\) | \(+0.3280\) | \(+23.7431\) |
| 0.05 | \(+0.0065\) | \(+0.0115\) | \(+0.0018\) | \(-0.0361\) | \(-133.4440\) | \(+0.0011\) | \(+0.0006\) | \(-0.0019\) | \(-0.0014\) | \(-199.1269\) |
| 0.10 | \(+0.0181\) | \(+0.0163\) | \(+0.0057\) | \(+0.0002\) | \(-145.9196\) | \(+0.0620\) | \(+0.0767\) | \(+0.0214\) | \(+0.1044\) | \(-219.5794\) |
| 0.20 | \(+0.0660\) | \(+0.0770\) | \(+0.0123\) | \(+0.0877\) | \(-158.0420\) | \(+0.0398\) | \(+0.0532\) | \(+0.0080\) | \(+0.1014\) | \(-358.3918\) |
4pt
CovType is included as supporting optimization evidence. The available CovType summaries report training loss rather than full validation or test performance, so these tables are used only to document optimization behavior and not as primary predictive
comparisons. The results show that training loss decreases with increasing label fraction, ERM remains lowest because it solves the least constrained objective, and the curvature ablation improves the optimization profile relative to full
V-GIB on this dataset. The per-seed results are reported in Table 11, while additional single-setting diagnostics are reported in Table 12.
| Method | Fraction | Seed 13 | Seed 29 | Seed 47 |
|---|---|---|---|---|
| ERM | 0.01 | 1.184251 | 1.181215 | 1.163115 |
| ERM | 0.05 | 0.788937 | 0.790554 | 0.787388 |
| ERM | 0.10 | 0.707573 | 0.707035 | 0.703328 |
| VIB | 0.01 | 1.237100 | 1.257122 | 1.251469 |
| VIB | 0.05 | 0.862581 | 0.867222 | 0.864158 |
| VIB | 0.10 | 0.782067 | 0.778620 | 0.775489 |
| V-GIB | 0.01 | 1.298463 | 1.314734 | 1.296654 |
| V-GIB | 0.05 | 0.908371 | 0.915618 | 0.911602 |
| V-GIB | 0.10 | 0.827656 | 0.826674 | 0.826155 |
| V-GIB-no-curv | 0.01 | 1.248284 | 1.262177 | 1.247819 |
| V-GIB-no-curv | 0.05 | 0.863893 | 0.871109 | 0.868530 |
| V-GIB-no-curv | 0.10 | 0.783400 | 0.781450 | 0.780440 |
| V-GIB-no-dim | 0.01 | 1.295303 | 1.311815 | 1.293614 |
| V-GIB-no-dim | 0.05 | 0.905226 | 0.912881 | 0.908773 |
| V-GIB-no-dim | 0.10 | 0.825054 | 0.824012 | 0.823633 |
6pt
Table 12 reports additional CovType training-loss diagnostics.
| Setting | Value | Count/Seed Info |
|---|---|---|
| V-GIB, fraction \(0.02\) | 1.092046 \(\pm\) 0.002860 | 3 seeds |
| ERM, fraction \(0.20\) | 0.642478 | seed 13 only |
| VIB, fraction \(0.20\) | 0.713492 | seed 13 only |
| V-GIB, fraction \(0.20\) | 0.763080 | seed 13 only |
| V-GIB-no-curv, fraction \(0.20\) | 0.718018 | seed 13 only |
| V-GIB-no-dim, fraction \(0.20\) | 0.760319 | seed 13 only |
6pt
rkatende@kab.ac.ug↩︎