Application Specific Compression of Deep Learning Models
Large Deep Learning models are compressed and deployed for specific applications. However, current Deep Learning model compression methods do not utilize the information about the target application. As a result, the compressed models are application agnostic. Our goal is to customize the model compression process to create a compressed model that will perform better for the target application. Our method, Application Specific Compression (ASC), identifies and prunes components of the large Deep Learning model that are redundant specifically for the given target application. The intuition of our work is to prune the parts of the network that do not contribute significantly to updating the data representation for the given application. We have experimented with the BERT family of models for three applications: Extractive QA, Natural Language Inference, and Paraphrase Identification. We observe that ASC performs better than existing model compression methods and off-the-shelf compressed models.
Authors
- Amit Awekar (ORCID: https://orcid.org/0000-0003-1559-8247)
- Rohit Raj (ORCID: https://orcid.org/0009-0004-6002-1767)
- Angana Borah
Institutions
- Indian Institute of Technology Guwahati (IN)
- University of Michigan (US)
Publication Details
- Published
- 2024-12-18
- DOI
- https://doi.org/10.1145/3703323.3703339
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 0.00