Closed-Book Training to Improve Summarization Encoder Memory
This work addresses the challenge of enhancing memory capabilities in summarization models for natural language processing, representing an incremental improvement over existing pointer-generator methods.
The paper tackles the problem of improving encoder memory in neural summarization models by introducing a closed-book decoder that forces the encoder to be more selective, resulting in significant performance gains on the CNN/Daily Mail dataset with higher ROUGE and METEOR scores and better generalizability on DUC-2002.
A good neural sequence-to-sequence summarization model should have a strong encoder that can distill and memorize the important information from long input texts so that the decoder can generate salient summaries based on the encoder's memory. In this paper, we aim to improve the memorization capabilities of the encoder of a pointer-generator model by adding an additional 'closed-book' decoder without attention and pointer mechanisms. Such a decoder forces the encoder to be more selective in the information encoded in its memory state because the decoder can't rely on the extra information provided by the attention and possibly copy modules, and hence improves the entire model. On the CNN/Daily Mail dataset, our 2-decoder model outperforms the baseline significantly in terms of ROUGE and METEOR metrics, for both cross-entropy and reinforced setups (and on human evaluation). Moreover, our model also achieves higher scores in a test-only DUC-2002 generalizability setup. We further present a memory ability test, two saliency metrics, as well as several sanity-check ablations (based on fixed-encoder, gradient-flow cut, and model capacity) to prove that the encoder of our 2-decoder model does in fact learn stronger memory representations than the baseline encoder.