Improving Joint Training of Inference Networks and Structured Prediction Energy Networks

Lifu Tu, Richard Yuanzhe Pang, Kevin Gimpel


Abstract
Deep energy-based models are powerful, but pose challenges for learning and inference (Belanger and McCallum, 2016). Tu and Gimpel (2018) developed an efficient framework for energy-based models by training “inference networks” to approximate structured inference instead of using gradient descent. However, their alternating optimization approach suffers from instabilities during training, requiring additional loss terms and careful hyperparameter tuning. In this paper, we contribute several strategies to stabilize and improve this joint training of energy functions and inference networks for structured prediction. We design a compound objective to jointly train both cost-augmented and test-time inference networks along with the energy function. We propose joint parameterizations for the inference networks that encourage them to capture complementary functionality during learning. We empirically validate our strategies on two sequence labeling tasks, showing easier paths to strong performance than prior work, as well as further improvements with global energy terms.
Anthology ID:
2020.spnlp-1.8
Volume:
Proceedings of the Fourth Workshop on Structured Prediction for NLP
Month:
November
Year:
2020
Address:
Online
Editors:
Priyanka Agrawal, Zornitsa Kozareva, Julia Kreutzer, Gerasimos Lampouras, André Martins, Sujith Ravi, Andreas Vlachos
Venue:
spnlp
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
62–73
Language:
URL:
https://aclanthology.org/2020.spnlp-1.8
DOI:
10.18653/v1/2020.spnlp-1.8
Bibkey:
Cite (ACL):
Lifu Tu, Richard Yuanzhe Pang, and Kevin Gimpel. 2020. Improving Joint Training of Inference Networks and Structured Prediction Energy Networks. In Proceedings of the Fourth Workshop on Structured Prediction for NLP, pages 62–73, Online. Association for Computational Linguistics.
Cite (Informal):
Improving Joint Training of Inference Networks and Structured Prediction Energy Networks (Tu et al., spnlp 2020)
Copy Citation:
PDF:
https://aclanthology.org/2020.spnlp-1.8.pdf
Video:
 https://slideslive.com/38940143