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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lindemann, Matthias, Koller, Alexander, Titov, Ivan
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913425220698112
author Lindemann, Matthias
Koller, Alexander
Titov, Ivan
author_facet Lindemann, Matthias
Koller, Alexander
Titov, Ivan
contents Strong inductive biases enable learning from little data and help generalization outside of 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.
format Preprint
id arxiv_https___arxiv_org_abs_2310_00796
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle SIP: Injecting a Structural Inductive Bias into a Seq2Seq Model by Simulation
Lindemann, Matthias
Koller, Alexander
Titov, Ivan
Computation and Language
Strong inductive biases enable learning from little data and help generalization outside of 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.
title SIP: Injecting a Structural Inductive Bias into a Seq2Seq Model by Simulation
topic Computation and Language
url https://arxiv.org/abs/2310.00796