GeoLAN: Geometric Learning of Latent Explanatory Directions in Large Language Models

Tianyu Pan, Damon L. Woodard


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.
Anthology ID:
2026.findings-acl.271
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
5493–5521
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.271/
DOI:
Bibkey:
Cite (ACL):
Tianyu Pan and Damon L. Woodard. 2026. GeoLAN: Geometric Learning of Latent Explanatory Directions in Large Language Models. In Findings of the Association for Computational Linguistics: ACL 2026, pages 5493–5521, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
GeoLAN: Geometric Learning of Latent Explanatory Directions in Large Language Models (Pan & Woodard, Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.271.pdf
Checklist:
 2026.findings-acl.271.checklist.pdf