Large Multimodal Model Compression via Efficient Pruning and Distillation at AntGroup

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Maolin, Zhao, Yao, Liu, Jiajia, Chen, Jingdong, Zhuang, Chenyi, Gu, Jinjie, Guo, Ruocheng, Zhao, Xiangyu
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914847625576448
author Wang, Maolin
Zhao, Yao
Liu, Jiajia
Chen, Jingdong
Zhuang, Chenyi
Gu, Jinjie
Guo, Ruocheng
Zhao, Xiangyu
author_facet Wang, Maolin
Zhao, Yao
Liu, Jiajia
Chen, Jingdong
Zhuang, Chenyi
Gu, Jinjie
Guo, Ruocheng
Zhao, Xiangyu
contents The deployment of Large Multimodal Models (LMMs) within AntGroup has significantly advanced multimodal tasks in payment, security, and advertising, notably enhancing advertisement audition tasks in Alipay. However, the deployment of such sizable models introduces challenges, particularly in increased latency and carbon emissions, which are antithetical to the ideals of Green AI. This paper introduces a novel multi-stage compression strategy for our proprietary LLM, AntGMM. Our methodology pivots on three main aspects: employing small training sample sizes, addressing multi-level redundancy through multi-stage pruning, and introducing an advanced distillation loss design. In our research, we constructed a dataset, the Multimodal Advertisement Audition Dataset (MAAD), from real-world scenarios within Alipay, and conducted experiments to validate the reliability of our proposed strategy. Furthermore, the effectiveness of our strategy is evident in its operational success in Alipay's real-world multimodal advertisement audition for three months from September 2023. Notably, our approach achieved a substantial reduction in latency, decreasing it from 700ms to 90ms, while maintaining online performance with only a slight performance decrease. Moreover, our compressed model is estimated to reduce electricity consumption by approximately 75 million kWh annually compared to the direct deployment of AntGMM, demonstrating our commitment to green AI initiatives. We will publicly release our code and the MAAD dataset after some reviews\footnote{https://github.com/MorinW/AntGMM$\_$Pruning}.
format Preprint
id arxiv_https___arxiv_org_abs_2312_05795
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Large Multimodal Model Compression via Efficient Pruning and Distillation at AntGroup
Wang, Maolin
Zhao, Yao
Liu, Jiajia
Chen, Jingdong
Zhuang, Chenyi
Gu, Jinjie
Guo, Ruocheng
Zhao, Xiangyu
Artificial Intelligence
The deployment of Large Multimodal Models (LMMs) within AntGroup has significantly advanced multimodal tasks in payment, security, and advertising, notably enhancing advertisement audition tasks in Alipay. However, the deployment of such sizable models introduces challenges, particularly in increased latency and carbon emissions, which are antithetical to the ideals of Green AI. This paper introduces a novel multi-stage compression strategy for our proprietary LLM, AntGMM. Our methodology pivots on three main aspects: employing small training sample sizes, addressing multi-level redundancy through multi-stage pruning, and introducing an advanced distillation loss design. In our research, we constructed a dataset, the Multimodal Advertisement Audition Dataset (MAAD), from real-world scenarios within Alipay, and conducted experiments to validate the reliability of our proposed strategy. Furthermore, the effectiveness of our strategy is evident in its operational success in Alipay's real-world multimodal advertisement audition for three months from September 2023. Notably, our approach achieved a substantial reduction in latency, decreasing it from 700ms to 90ms, while maintaining online performance with only a slight performance decrease. Moreover, our compressed model is estimated to reduce electricity consumption by approximately 75 million kWh annually compared to the direct deployment of AntGMM, demonstrating our commitment to green AI initiatives. We will publicly release our code and the MAAD dataset after some reviews\footnote{https://github.com/MorinW/AntGMM$\_$Pruning}.
title Large Multimodal Model Compression via Efficient Pruning and Distillation at AntGroup
topic Artificial Intelligence
url https://arxiv.org/abs/2312.05795