SIP: Injecting a Structural Inductive Bias into a Seq2Seq Model by Simulation

Matthias Lindemann, Alexander Koller, Ivan Titov


Abstract
Strong inductive biases enable learning from little data and help generalization outside the training distribution. Popular neural architectures such as Transformers lack strong structural inductive biases for seq2seq NLP tasks on their own. Consequently, they struggle with systematic generalization beyond the training distribution, e.g. with extrapolating to longer inputs, even when pre-trained on large amounts of text.We show how a structural inductive bias can be efficiently injected into a seq2seq model by pre-training it to simulate structural transformations on synthetic data. Specifically, we inject an inductive bias towards Finite State Transducers (FSTs) into a Transformer by pre-training it to simulate FSTs given their descriptions. Our experiments show that our method imparts the desired inductive bias, resulting in improved systematic generalization and better few-shot learning for FST-like tasks. Our analysis shows that fine-tuned models accurately capture the state dynamics of the unseen underlying FSTs, suggesting that the simulation process is internalized by the fine-tuned model.
Anthology ID:
2024.acl-long.355
Volume:
Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
August
Year:
2024
Address:
Bangkok, Thailand
Editors:
Lun-Wei Ku, Andre Martins, Vivek Srikumar
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
6570–6587
Language:
URL:
https://aclanthology.org/2024.acl-long.355
DOI:
Bibkey:
Cite (ACL):
Matthias Lindemann, Alexander Koller, and Ivan Titov. 2024. SIP: Injecting a Structural Inductive Bias into a Seq2Seq Model by Simulation. In Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 6570–6587, Bangkok, Thailand. Association for Computational Linguistics.
Cite (Informal):
SIP: Injecting a Structural Inductive Bias into a Seq2Seq Model by Simulation (Lindemann et al., ACL 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.acl-long.355.pdf