Geometry as a Missing Axis of Representation Quality: The Variational Geometric Information Bottleneck under Data Scarcity


Abstract

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.

1 Introduction↩︎

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.

  1. 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.

  2. 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.

  3. We prove consistency of the empirical surrogate optimized by V-GIB under explicit concentration assumptions for the information, curvature, and intrinsic-dimension estimators.

  4. 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.

1.1 Related Work↩︎

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.

2 Geometry as a Missing Axis of Representation Quality↩︎

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.

2.1 Setup↩︎

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.

2.2 Geometry-aware 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.

2.3 Assumptions↩︎

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.

2.4 Mechanism↩︎

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.

2.5 Information–geometry frontier↩︎

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.

2.6 Estimator concentration and surrogate consistency↩︎

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.

2.7 What the theory establishes↩︎

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.

3 Experimental Validation↩︎

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?

3.1 Protocol↩︎

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.

3.2 Breast Cancer↩︎

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.

Table 1: Breast Cancer test accuracy under label scarcity. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\). No-curv denotes V-GIB-no-curv. The last two columns report the full V-GIB accuracy difference relative to ERM and VIB.
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

a
b
c

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.

3.3 FashionMNIST↩︎

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).

Table 2: FashionMNIST test accuracy under label scarcity. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\). Bold indicates the best mean at each label fraction.
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\)
a
b
c
d

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.

3.4 CIFAR-10↩︎

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.

Table 3: CIFAR-10 test accuracy under label scarcity. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\). No-curv denotes V-GIB-no-curv. The last two columns report the full V-GIB accuracy difference relative to ERM and VIB.
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

a
b
c

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.

3.5 Supporting CovType optimization check↩︎

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.

Table 4: CovType supporting optimization summary. Entries are mean \(\pm\) standard deviation of training loss over seeds \(13,29,47\). No-curv denotes V-GIB-no-curv; No-dim denotes V-GIB-no-dim.
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

3.6 Synthesis↩︎

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.

Funding↩︎

The author received no financial support for the research, authorship, or publication of this article.

Competing interests↩︎

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.

Consent for publication↩︎

Not applicable.

Data availability↩︎

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.

Materials availability↩︎

Not applicable.

Code availability↩︎

The code used to run the experiments and generate the reported tables and figures is publicly available at https://github.com/karjxenval/V-GIB.

Author contributions↩︎

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.

4 Proofs for the geometry-sensitive bounds↩︎

This appendix gives the proofs for the geometry-sensitive results in Sec. 2. The statements are conditional on Assumptions 1 and 2.

4.1 Proof of the latent covering bound↩︎

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. ◻

4.2 Proof of the geometry-sensitive uniform deviation bound↩︎

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. ◻

4.3 Sample-complexity consequence↩︎

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. ◻

5 Optimization and surrogate-consistency proofs↩︎

This appendix gives the proofs for Theorems 2 and 3, together with the curvature concentration statement used in the surrogate analysis.

5.1 Proof of the information–geometry frontier↩︎

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. ◻

5.2 Curvature estimator concentration↩︎

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. ◻

5.3 Proof of uniform surrogate consistency↩︎

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. ◻

5.4 Geometry penalties and diagnostics↩︎

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.

5.5 Training protocol↩︎

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.

5.6 Checkpoint selection↩︎

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.

5.7 Evaluation metrics↩︎

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.

5.8 Saved outputs↩︎

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.

6 Additional Experimental Results↩︎

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.

6.1 Breast Cancer↩︎

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.

a
b
c

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.

a
b
c

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.

Table 5: Full Breast Cancer test summary. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\).
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.

Table 6: Pairwise differences for the full V-GIB model on Breast Cancer. Positive values mean V-GIB is higher than the comparator.
\(\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\)

6.2 FashionMNIST↩︎

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.

a
b
c
d
e

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.

Table 7: FashionMNIST predictive metrics. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\).
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.

Table 8: FashionMNIST geometric and utility diagnostics. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\).
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

6.3 CIFAR-10↩︎

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.

a
b
c

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.

a
b
c

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.

Table 9: Full CIFAR-10 test summary. Values are mean \(\pm\) standard deviation over seeds \(13,29,47\).
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.

Table 10: Pairwise differences for the full V-GIB model on CIFAR-10. Positive predictive and utility differences mean V-GIB is higher than the comparator. Negative curvature differences mean V-GIB has lower curvature proxy than the comparator.
\(\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

6.4 CovType↩︎

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.

Table 11: Per-seed CovType training loss for the fully available fractions.
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.

Table 12: Additional CovType training-loss results. Fraction \(0.02\) is available for V-GIB, while fraction \(0.20\) is reported as a single-seed diagnostic check.
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

References↩︎

[1]
Brenden M. Lake, Ruslan Salakhutdinov, and Joshua B. Tenenbaum. Human-level concept learning through probabilistic program induction. Science, 350(6266):1332–1338, 2015.
[2]
Oriol Vinyals, Charles Blundell, Timothy Lillicrap, Koray Kavukcuoglu, and Daan Wierstra. Matching networks for one shot learning. In Advances in Neural Information Processing Systems 29, pages 3630–3638, 2016.
[3]
Chelsea Finn, Pieter Abbeel, and Sergey Levine. Model-agnostic meta-learning for fast adaptation of deep networks. In Proceedings of the 34th International Conference on Machine Learning, volume 70 of Proceedings of Machine Learning Research, pages 1126–1135. PMLR, 2017.
[4]
Jake Snell, Kevin Swersky, and Richard S. Zemel. Prototypical networks for few-shot learning. In Advances in Neural Information Processing Systems 30, 2017.
[5]
Ting Chen, Simon Kornblith, Mohammad Norouzi, and Geoffrey Hinton. A simple framework for contrastive learning of visual representations. In Proceedings of the 37th International Conference on Machine Learning, volume 119 of Proceedings of Machine Learning Research, pages 1597–1607. PMLR, 2020.
[6]
Kaiming He, Haoqi Fan, Yuxin Wu, Saining Xie, and Ross Girshick. Momentum contrast for unsupervised visual representation learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 9729–9738, 2020.
[7]
Naftali Tishby, Fernando C. Pereira, and William Bialek. The information bottleneck method. In Proceedings of the 37th Annual Allerton Conference on Communication, Control and Computing, pages 368–377, 1999.
[8]
Alexander A. Alemi, Ian Fischer, Joshua V. Dillon, and Kevin Murphy. Deep variational information bottleneck. In Proceedings of the 5th International Conference on Learning Representations, 2017.
[9]
Joshua B. Tenenbaum, Vin de Silva, and John C. Langford. A global geometric framework for nonlinear dimensionality reduction. Science, 290(5500):2319–2323, 2000.
[10]
Mikhail Belkin, Partha Niyogi, and Vikas Sindhwani. Manifold regularization: A geometric framework for learning from labeled and unlabeled examples. Journal of Machine Learning Research, 7:2399–2434, 2006.
[11]
Partha Niyogi, Stephen Smale, and Shmuel Weinberger. Finding the homology of submanifolds with high confidence from random samples. Discrete & Computational Geometry, 39(1–3):419–441, 2008.
[12]
Charles Fefferman, Sanjoy Mitter, and Hariharan Narayanan. Testing the manifold hypothesis. Journal of the American Mathematical Society, 29(4):983–1049, 2016.
[13]
Michael M. Bronstein, Joan Bruna, Yann LeCun, Arthur Szlam, and Pierre Vandergheynst. Geometric deep learning: Going beyond euclidean data. IEEE Signal Processing Magazine, 34(4):18–42, 2017.
[14]
Michael M. Bronstein, Joan Bruna, Taco Cohen, and Petar Velivckovic. Geometric deep learning: Grids, groups, graphs, geodesics, and gauges. arXiv preprint arXiv:2104.13478, 2021.
[15]
Marco Tulio Ribeiro, Sameer Singh, and Carlos Guestrin. “why should i trust you?”: Explaining the predictions of any classifier. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pages 1135–1144, New York, 2016. Association for Computing Machinery.
[16]
Been Kim, Martin Wattenberg, Justin Gilmer, Carrie J. Cai, James Wexler, Fernanda B. Viegas, and Rory Sayres. Interpretability beyond feature attribution: Quantitative testing with concept activation vectors. In Proceedings of the 35th International Conference on Machine Learning, volume 80 of Proceedings of Machine Learning Research, pages 2668–2677. PMLR, 2018.
[17]
David Bau, Jun-Yan Zhu, Hendrik Strobelt, Agata Lapedriza, Bolei Zhou, and Antonio Torralba. Understanding the role of individual units in a deep neural network. Proceedings of the National Academy of Sciences of the United States of America, 117(48):30071–30078, 2020.
[18]
Yoshua Bengio, Aaron Courville, and Pascal Vincent. Representation learning: A review and new perspectives. IEEE Transactions on Pattern Analysis and Machine Intelligence, 35(8):1798–1828, 2013.
[19]
Elizaveta Levina and Peter J. Bickel. Maximum likelihood estimation of intrinsic dimension. In Advances in Neural Information Processing Systems 17, pages 777–784. MIT Press, 2005.
[20]
Thomas M. Cover and Joy A. Thomas. Elements of Information Theory. Wiley-Interscience, Hoboken, NJ, 2 edition, 2006.
[21]
Herbert Federer. Curvature measures. Transactions of the American Mathematical Society, 93(3):418–491, 1959.
[22]
Peter Petersen. Riemannian Geometry, volume 171 of Graduate Texts in Mathematics. Springer, Cham, 3 edition, 2016.
[23]
Peter L. Bartlett and Shahar Mendelson. Rademacher and gaussian complexities: Risk bounds and structural results. Journal of Machine Learning Research, 3:463–482, 2002.
[24]
Mehryar Mohri, Afshin Rostamizadeh, and Ameet Talwalkar. Foundations of Machine Learning. MIT Press, Cambridge, MA, 2 edition, 2018.

  1. rkatende@kab.ac.ug↩︎