← Search

Yichen Jiang

8 accepted papers

2024

Inducing Systematicity in Transformers by Attending to Structurally Quantized Embeddings

ACL 2024long

Transformers generalize to novel compositions of structures and entities after being trained on a complex dataset, but easily overfit on datasets of insufficient complexity. We observe that when the training set is sufficiently complex, the model encodes structurally equivalent sentences using a sys…

2024

ReMix: Optimizing Data Mixtures for Large Scale Imitation Learning

CoRL 2024poster

Increasingly large robotics datasets are being collected to train larger foundation models in robotics. However, despite the fact that data selection has been of utmost importance to scaling in vision and natural language processing (NLP), little work in robotics has questioned what data such models…

Cited by 16SourcecodeScholar
2022

Mutual Exclusivity Training and Primitive Augmentation to Induce Compositionality

EMNLP 2022main

Recent datasets expose the lack of the systematic generalization ability in standard sequence-to-sequence models. In this work, we analyze this behavior of seq2seq models and identify two contributing factors: a lack of mutual exclusivity bias (one target sequence can only be mapped to one source se…

2021

Enriching Transformers with Structured Tensor-Product Representations for Abstractive Summarization

NAACL 2021long

Abstractive summarization, the task of generating a concise summary of input documents, requires: (1) reasoning over the source document to determine the salient pieces of information scattered across the long document, and (2) composing a cohesive text by reconstructing these salient facts into a s…

2021

Inducing Transformer’s Compositional Generalization Ability via Auxiliary Sequence Prediction Tasks

EMNLP 2021main

Systematic compositionality is an essential mechanism in human language, allowing the recombination of known parts to create novel expressions. However, existing neural models have been shown to lack this basic ability in learning symbolic structures. Motivated by the failure of a Transformer model…