Pruning Multilingual Large Language Models for Multilingual Inference

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kim, Hwichan, Suzuki, Jun, Hirasawa, Tosho, Komachi, Mamoru
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912054958358528
author Kim, Hwichan
Suzuki, Jun
Hirasawa, Tosho
Komachi, Mamoru
author_facet Kim, Hwichan
Suzuki, Jun
Hirasawa, Tosho
Komachi, Mamoru
contents Multilingual large language models (MLLMs), trained on multilingual balanced data, demonstrate better zero-shot learning performance in non-English languages compared to large language models trained on English-dominant data. However, the disparity in performance between English and non-English languages remains a challenge yet to be fully addressed. A distinctive characteristic of MLLMs is their high-quality translation capabilities, indicating an acquired proficiency in aligning between languages. This study explores how to enhance the zero-shot performance of MLLMs in non-English languages by leveraging their alignment capability between English and non-English languages. To achieve this, we first analyze the behavior of MLLMs when performing translation and reveal that there are large magnitude features that play a critical role in the translation process. Inspired by these findings, we retain the weights associated with operations involving the large magnitude features and prune other weights to force MLLMs to rely on these features for tasks beyond translation. We empirically demonstrate that this pruning strategy can enhance the MLLMs' performance in non-English language.
format Preprint
id arxiv_https___arxiv_org_abs_2409_16911
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Pruning Multilingual Large Language Models for Multilingual Inference
Kim, Hwichan
Suzuki, Jun
Hirasawa, Tosho
Komachi, Mamoru
Computation and Language
Multilingual large language models (MLLMs), trained on multilingual balanced data, demonstrate better zero-shot learning performance in non-English languages compared to large language models trained on English-dominant data. However, the disparity in performance between English and non-English languages remains a challenge yet to be fully addressed. A distinctive characteristic of MLLMs is their high-quality translation capabilities, indicating an acquired proficiency in aligning between languages. This study explores how to enhance the zero-shot performance of MLLMs in non-English languages by leveraging their alignment capability between English and non-English languages. To achieve this, we first analyze the behavior of MLLMs when performing translation and reveal that there are large magnitude features that play a critical role in the translation process. Inspired by these findings, we retain the weights associated with operations involving the large magnitude features and prune other weights to force MLLMs to rely on these features for tasks beyond translation. We empirically demonstrate that this pruning strategy can enhance the MLLMs' performance in non-English language.
title Pruning Multilingual Large Language Models for Multilingual Inference
topic Computation and Language
url https://arxiv.org/abs/2409.16911