On the Compression of Language Models for Code: An Empirical Study on CodeBERT

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: d'Aloisio, Giordano, Traini, Luca, Sarro, Federica, Di Marco, Antinisca
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915069737041920
author d'Aloisio, Giordano
Traini, Luca
Sarro, Federica
Di Marco, Antinisca
author_facet d'Aloisio, Giordano
Traini, Luca
Sarro, Federica
Di Marco, Antinisca
contents Language models have proven successful across a wide range of software engineering tasks, but their significant computational costs often hinder their practical adoption. To address this challenge, researchers have begun applying various compression strategies to improve the efficiency of language models for code. These strategies aim to optimize inference latency and memory usage, though often at the cost of reduced model effectiveness. However, there is still a significant gap in understanding how these strategies influence the efficiency and effectiveness of language models for code. Here, we empirically investigate the impact of three well-known compression strategies -- knowledge distillation, quantization, and pruning -- across three different classes of software engineering tasks: vulnerability detection, code summarization, and code search. Our findings reveal that the impact of these strategies varies greatly depending on the task and the specific compression method employed. Practitioners and researchers can use these insights to make informed decisions when selecting the most appropriate compression strategy, balancing both efficiency and effectiveness based on their specific needs.
format Preprint
id arxiv_https___arxiv_org_abs_2412_13737
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On the Compression of Language Models for Code: An Empirical Study on CodeBERT
d'Aloisio, Giordano
Traini, Luca
Sarro, Federica
Di Marco, Antinisca
Software Engineering
Artificial Intelligence
Performance
Language models have proven successful across a wide range of software engineering tasks, but their significant computational costs often hinder their practical adoption. To address this challenge, researchers have begun applying various compression strategies to improve the efficiency of language models for code. These strategies aim to optimize inference latency and memory usage, though often at the cost of reduced model effectiveness. However, there is still a significant gap in understanding how these strategies influence the efficiency and effectiveness of language models for code. Here, we empirically investigate the impact of three well-known compression strategies -- knowledge distillation, quantization, and pruning -- across three different classes of software engineering tasks: vulnerability detection, code summarization, and code search. Our findings reveal that the impact of these strategies varies greatly depending on the task and the specific compression method employed. Practitioners and researchers can use these insights to make informed decisions when selecting the most appropriate compression strategy, balancing both efficiency and effectiveness based on their specific needs.
title On the Compression of Language Models for Code: An Empirical Study on CodeBERT
topic Software Engineering
Artificial Intelligence
Performance
url https://arxiv.org/abs/2412.13737