Can we teach natural language understanding models to track their beliefs through intermediate points in text? We propose a representation learning framework called breakpoint modeling that allows for learning of this type. Given any text encoder and data marked with intermediate states (breakpoints) along with corresponding textual queries viewed as true/false propositions (i.e., the candidate beliefs of a model, consisting of information changing through time) our approach trains models in an efficient and end-to-end fashion to build intermediate representations that facilitate teaching and direct querying of beliefs at arbitrary points alongside solving other end tasks. To show the benefit of our approach, we experiment with a diverse set of NLU tasks including relational reasoning on CLUTRR and narrative understanding on bAbI. Using novel belief prediction tasks for both tasks, we show the benefit of our main breakpoint transformer, based on T5, over conventional representation learning approaches in terms of processing efficiency, prediction accuracy and prediction consistency, all with minimal to no effect on corresponding QA end-tasks. To show the feasibility of incorporating our belief tracker into more complex reasoning pipelines, we also obtain SOTA performance on the three-tiered reasoning challenge for the TRIP benchmark (around 23-32% absolute improvement on Tasks 2-3).
|Number of pages||17|
|State||Published - 2022|
|Event||2022 Conference on Empirical Methods in Natural Language Processing, EMNLP 2022 - Abu Dhabi, United Arab Emirates|
Duration: 7 Dec 2022 → 11 Dec 2022
|Conference||2022 Conference on Empirical Methods in Natural Language Processing, EMNLP 2022|
|Country/Territory||United Arab Emirates|
|Period||7/12/22 → 11/12/22|
Bibliographical noteFunding Information:
The authors thank the Aristo team for valuable feedback. This work was supported by the European Research Council (ERC) under the European Union's Horizon 2020 research and innovation pro-gramme (grant no. 852686, SIAM, Shahaf). Part of this research was also supported by the European Research Council, ERC-StG grant no. 677352 (Tsarfaty), which is gratefully acknowledged.
© 2022 Association for Computational Linguistics.