Compressing Cross-Lingual Multi-Task Models at Qualtrics

Authors

  • Daniel Campos University of Illinois Urbana-Champaign
  • Daniel Perry Qualtrics
  • Samir Joshi Qualtrics
  • Yashmeet Gambhir Qualtrics
  • Wei Du Qualtrics
  • Zhengzheng Xing Qualtrics
  • Aaron Colak Qualtrics

DOI:

https://doi.org/10.1609/aaai.v37i13.26857

Keywords:

Transfer Learning, Model Compression, Knowledge Distillation, Multitask Modeling, Cross-lingual Modeling, Quantization, Transformers, Inference Latency

Abstract

Experience management is an emerging business area where organizations focus on understanding the feedback of customers and employees in order to improve their end-to-end experiences. This results in a unique set of machine learning problems to help understand how people feel, discover issues they care about, and find which actions need to be taken on data that are different in content and distribution from traditional NLP domains. In this paper, we present a case study of building text analysis applications that perform multiple classification tasks efficiently in 12 languages in the nascent business area of experience management. In order to scale up modern ML methods on experience data, we leverage cross lingual and multi-task modeling techniques to consolidate our models into a single deployment to avoid overhead. We also make use of model compression and model distillation to reduce overall inference latency and hardware cost to the level acceptable for business needs while maintaining model prediction quality. Our findings show that multi-task modeling improves task performance for a subset of experience management tasks in both XLM-R and mBert architectures. Among the compressed architectures we explored, we found that MiniLM achieved the best compression/performance tradeoff. Our case study demonstrates a speedup of up to 15.61x with 2.60% average task degradation (or 3.29x speedup with 1.71% degradation) and estimated savings of 44% over using the original full-size model. These results demonstrate a successful scaling up of text classification for the challenging new area of ML for experience management.

Downloads

Published

2023-09-06

How to Cite

Campos, D., Perry, D., Joshi, S., Gambhir, Y., Du, W., Xing, Z., & Colak, A. (2023). Compressing Cross-Lingual Multi-Task Models at Qualtrics. Proceedings of the AAAI Conference on Artificial Intelligence, 37(13), 15661-15667. https://doi.org/10.1609/aaai.v37i13.26857

Issue

Section

IAAI Technical Track on emerging Applications of AI