GeoLAN: Geometric Learning of Latent Explanatory Directions in Large Language Models
Organizations: Department of Electrical and Computer Engineering Florida Institute for National Security Applied Artificial Intelligence Group University of Florida
Abstract
Large language models (LLMs) demonstrate strong performance, but they often lack transparency. We introduce GeoLAN, a training framework that treats token representations as geometric trajectories and applies stickiness conditions inspired by recent developments related to the Kakeya Conjecture. We have developed two differentiable regularizers, Katz-Tao Convex Wolff (KT-CW) and Katz-Tao Attention (KT-Attn), that promote isotropy and encourage diverse attention. Our experiments with Gemma-3 (1B, 4B, 12B) and Llama-3-8B show that GeoLAN frequently maintains task accuracy while improving geometric metrics and reducing certain fairness biases. These benefits are most significant in mid-sized models. Our findings reveal scale-dependent trade-offs between geometric precision and performance, suggesting that geometry-aware training is a promising approach to enhance mechanistic interpretability.
Figures & tables
| Model | Metric | Comparison | Mean Difference | -value | Effect Size |
|---|---|---|---|---|---|
| Gemma3-1B | Crows Stereotype Rate | Geolan vs. Baseline | 0.0050 | 0.0957 | 0.22 (small) |
| Gemma3-1B | Crows Stereotype Rate | Control vs. Baseline | 0.0065 | 0.5000 | 0.27 (small) |
| Gemma3-4B | Crows Stereotype Rate | Geolan vs. Control | 0.0030 | 0.5534 | 0.29 (small) |
| Gemma3-4B | Crows Stereotype Rate | Control vs. Baseline | -0.0030 | 0.4765 | -0.20 (small) |
| Gemma3-12B | Crows Stereotype Rate | Geolan vs. Baseline | -0.0065 | 0.0065 | -0.38 (small) |
| Gemma3-12B | Crows Stereotype Rate | Geolan vs. Control | -0.0045 | 0.2420 | -0.26 (small) |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning | Notes |
|---|---|---|
| Model, indices, representations | ||
| Vocabulary. | ||
| Input token sequence. | ||
| -th token. | ||
| Sequence length. | ||
| Transformer model. | ||
| Model | Metric | Comparison | Mean Difference | -value | Effect Size |
|---|---|---|---|---|---|
| Gemma3 1B | Stable Examples | Geolan vs. Baseline | 1.2500 | 0.5930 | 0.38 (small) |
| Gemma3 1B | Stable Examples | Geolan vs. Control | 1.7500 | 0.0354 | 0.64 (medium) |
| Gemma3 1B | Stable Examples | Control vs. Baseline | -0.5000 | 0.8130 | -0.18 (negligible) |
| Gemma3 1B | Distribution Kl Mean | Geolan vs. Baseline | -0.0016 | 0.7703 | -0.25 (small) |
| Gemma3 1B | Distribution Kl Mean | Geolan vs. Control | 0.0023 | 0.4790 | 0.39 (small) |
| Gemma3 1B | Distribution Kl Mean | Control vs. Baseline | -0.0039 | 0.2440 | -0.68 (medium) |
| Model | Metric | Comparison | Mean Difference | -value | Effect Size |
|---|---|---|---|---|---|
| Gemma3 1B | Final Layer Cone Top10 | Geolan vs.Baseline | 0.0079 | 0.1392 | 1.29 (large) |
| Gemma3 1B | Final Layer Cone Top10 | Geolan vs.Control | 0.0081 | 0.2297 | 1.18 (large) |
| Gemma3 1B | Final Layer Cone Top10 | Control vs.Baseline | -0.0002 | 0.9784 | -0.02 (negligible) |
| Gemma3 1B | Final Layer Cone Top50 | Geolan vs.Baseline | 0.0059 | 0.1555 | 1.21 (large) |
| Gemma3 1B | Final Layer Cone Top50 | Geolan vs.Control | 0.0060 | 0.2362 | 1.13 (large) |
| Gemma3 1B | Final Layer Cone Top50 | Control vs.Baseline | -0.0002 | 0.9731 | -0.03 (negligible) |
| Model | Metric | Comparison | Mean Difference | -value | Effect Size |
|---|---|---|---|---|---|
| Gemma3 1B | Clean Acc | Geolan vs.Baseline | 0.0040 | 0.4564 | 0.34 (small) |
| Gemma3 1B | Clean Acc | Geolan vs.Control | 0.0055 | 0.3999 | 0.47 (small) |
| Gemma3 1B | Clean Acc | Control vs.Baseline | -0.0015 | 0.6084 | -0.14 (negligible) |
| Gemma3 1B | Corr Acc | Geolan vs.Baseline | -0.0025 | 0.4309 | -0.29 (small) |
| Gemma3 1B | Corr Acc | Geolan vs.Control | 0.0015 | 0.8006 | 0.16 (negligible) |
| Gemma3 1B | Corr Acc | Control vs.Baseline | -0.0040 | 0.4795 | -0.44 (small) |
| Metric | Model | Baseline | Control | GeoLAN | G vs. C ( ) | Pattern |
|---|---|---|---|---|---|---|
| BBQ Accuracy | Gemma3 1B | +0.00 | G C | |||
| Gemma3 4B | -0.07 | G C | ||||
| Llama3 8B | +0.00 | G C | ||||
| Gemma3 12B | +0.01 | G C | ||||
| Cone Concentration | Gemma3 1B | +1.35* | G < C | |||
| Gemma3 4B | -0.71 | G > C |
LLM Interpretability
Large language model (LLM) interpretability studies how internal representations and computations produce outputs, and how they can be explained or controlled.
See all 778 papers