Names Don’t Matter: Symbol-Invariant Transformer for Open-Vocabulary Learning
İlker Işık, Wenchao Li
Abstract
Current neural architectures lack a principled way to handle interchangeable tokens, i.e., symbols that are semantically equivalent yet distinguishable, such as bound variables. As a result, models trained on fixed vocabularies often struggle to generalize to unseen symbols, even when the underlying semantics remain unchanged. We propose a novel Transformer-based mechanism that is provably invariant to the renaming of interchangeable tokens. Our approach employs parallel embedding streams to isolate the contribution of each interchangeable token in the input, combined with an aggregated attention mechanism that enables structured information sharing across streams. Experimental results confirm the theoretical guarantees of our method and demonstrate substantial performance gains on open-vocabulary tasks that require generalization to novel symbols. Project page: https://bu-depend-lab.github.io/Symbol-Invariant-Transformer/
Ask Lune
Ask anything. The answer quotes the paper.
Lune reads all of it and the papers around it, then shows you the passages behind every answer.
Builds on11
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- COGS: A Compositional Generalization Challenge Based on Semantic InterpretationNajoung Kim, Tal LinzenEMNLP 2020 · 149 citations
- Proof Artifact Co-Training for Theorem Proving with Language ModelsJesse Michael Han, Jason Rute, Yuhuai Wu, Edward W. Ayers et al.ICLR 2022 · 149 citations
- Teaching Temporal Logics to Neural NetworksChristopher Hahn, Frederik Schmitt, Jens U. Kreber, Markus Norman Rabe et al.ICLR 2021 · 78 citations
- Mathematical Reasoning via Self-supervised Skip-tree TrainingMarkus Norman Rabe, Dennis Lee, Kshitij Bansal, Christian SzegedyICLR 2021 · 65 citations
Related papers
- Interchangeable Token Embeddings for Extendable Vocabulary and Alpha-EquivalenceIlker Isik, Ramazan Gokberk Cinbis, Ebru Aydin GolICML 2025
- How Do Transformers Learn Variable Binding in Symbolic Programs?Yiwei Wu, Atticus Geiger, Raphaël MillièreICML 2025
- Lexinvariant Language ModelsQian Huang, Eric Zelikman, Sarah Li Chen, Yuhuai Wu et al.NeurIPS 2023 · 3 citations
- In-Context AlgebraEric Todd, Jannik Brinkmann, Rohit Gandikota, David BauICLR 2026 · 3 citations
- From Characters to Words: Hierarchical Pre-trained Language Model for Open-vocabulary Language UnderstandingLi Sun, Florian Luisier, Kayhan Batmanghelich, Dinei A. F. Florêncio et al.ACL 2023
