Compositional Networks Enable Systematic Generalization for Grounded Language Understanding

TL;DR

This paper introduces a compositional network architecture that improves systematic generalization in grounded language understanding tasks, matching state-of-the-art performance while enhancing interpretability and robustness.

Contribution

The authors propose a general-purpose, compositional network mechanism that enables better generalization in grounded language understanding without sacrificing performance.

Findings

Achieves state-of-the-art performance on gSCAN dataset

Improves generalization to novel linguistic compositions

Provides interpretability of network components

Abstract

Humans are remarkably flexible when understanding new sentences that include combinations of concepts they have never encountered before. Recent work has shown that while deep networks can mimic some human language abilities when presented with novel sentences, systematic variation uncovers the limitations in the language-understanding abilities of networks. We demonstrate that these limitations can be overcome by addressing the generalization challenges in the gSCAN dataset, which explicitly measures how well an agent is able to interpret novel linguistic commands grounded in vision, e.g., novel pairings of adjectives and nouns. The key principle we employ is compositionality: that the compositional structure of networks should reflect the compositional structure of the problem domain they address, while allowing other parameters to be learned end-to-end. We build a general-purpose mechanism that enables agents to generalize their language understanding to compositional domains. Crucially, our network has the same state-of-the-art performance as prior work while generalizing its knowledge when prior work does not. Our network also provides a level of interpretability that enables users to inspect what each part of networks learns. Robust grounded language understanding without dramatic failures and without corner cases is critical to building safe and fair robots; we demonstrate the significant role that compositionality can play in achieving that goal.