Decoupled Transformer for Scalable Inference in Open-domain Question Answering

Haytham Elfdaeel, Stanislav Peshterliev


Abstract
Large transformer models, such as BERT, achieve state-of-the-art results in machine reading comprehension (MRC) for open-domain question answering (QA). However, transformers have a high computational cost for inference which makes them hard to apply to online QA systems for applications like voice assistants. To reduce computational cost and latency, we propose decoupling the transformer MRC model into input-component and cross-component. The decoupling allows for part of the representation computation to be performed offline and cached for online use. To retain the decoupled transformer accuracy, we devised a knowledge distillation objective from a standard transformer model. Moreover, we introduce learned representation compression layers which help reduce by four times the storage requirement for the cache. In experiments on the SQUAD 2.0 dataset, a decoupled transformer reduces the computational cost and latency of open-domain MRC by 30-40% with only 1.2 points worse F1-score compared to a standard transformer.
Anthology ID:
2021.ranlp-1.44
Volume:
Proceedings of the International Conference on Recent Advances in Natural Language Processing (RANLP 2021)
Month:
September
Year:
2021
Address:
Held Online
Editors:
Ruslan Mitkov, Galia Angelova
Venue:
RANLP
SIG:
Publisher:
INCOMA Ltd.
Note:
Pages:
386–393
Language:
URL:
https://aclanthology.org/2021.ranlp-1.44
DOI:
Bibkey:
Cite (ACL):
Haytham Elfdaeel and Stanislav Peshterliev. 2021. Decoupled Transformer for Scalable Inference in Open-domain Question Answering. In Proceedings of the International Conference on Recent Advances in Natural Language Processing (RANLP 2021), pages 386–393, Held Online. INCOMA Ltd..
Cite (Informal):
Decoupled Transformer for Scalable Inference in Open-domain Question Answering (Elfdaeel & Peshterliev, RANLP 2021)
Copy Citation:
PDF:
https://aclanthology.org/2021.ranlp-1.44.pdf
Data
MRPC