Cost-effective Deployment of BERT Models in Serverless Environment
Marek Suppa, Katarína Benešová, Andrej Švec
Abstract
In this study, we demonstrate the viability of deploying BERT-style models to AWS Lambda in a production environment. Since the freely available pre-trained models are too large to be deployed in this environment, we utilize knowledge distillation and fine-tune the models on proprietary datasets for two real-world tasks: sentiment analysis and semantic textual similarity. As a result, we obtain models that are tuned for a specific domain and deployable in the serverless environment. The subsequent performance analysis shows that this solution does not only report latency levels acceptable for production use but that it is also a cost-effective alternative to small-to-medium size deployments of BERT models, all without any infrastructure overhead.
BibTeX
@inproceedings{suppa-etal-2021-cost,
title = "Cost-effective Deployment of {BERT} Models in Serverless Environment",
author = "Suppa, Marek and
Bene{\v{s}}ov{\'a}, Katar{\'i}na and
{\v{S}}vec, Andrej",
editor = "Kim, Young-bum and
Li, Yunyao and
Rambow, Owen",
booktitle = "Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies: Industry Papers",
month = jun,
year = "2021",
address = "Online",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2021.naacl-industry.24/",
doi = "10.18653/v1/2021.naacl-industry.24",
pages = "187--195"
}