AfroXLMR-Comet: Multilingual Knowledge Distillation with Attention Matching for Low-Resource languages

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Raju, Joshua Sakthivel, S, Sanjay, Walia, Jaskaran Singh, Raghav, Srinivas, Marivate, Vukosi
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913706605019136
author Raju, Joshua Sakthivel
S, Sanjay
Walia, Jaskaran Singh
Raghav, Srinivas
Marivate, Vukosi
author_facet Raju, Joshua Sakthivel
S, Sanjay
Walia, Jaskaran Singh
Raghav, Srinivas
Marivate, Vukosi
contents Language model compression through knowledge distillation has emerged as a promising approach for deploying large language models in resource-constrained environments. However, existing methods often struggle to maintain performance when distilling multilingual models, especially for low-resource languages. In this paper, we present a novel hybrid distillation approach that combines traditional knowledge distillation with a simplified attention matching mechanism, specifically designed for multilingual contexts. Our method introduces an extremely compact student model architecture, significantly smaller than conventional multilingual models. We evaluate our approach on five African languages: Kinyarwanda, Swahili, Hausa, Igbo, and Yoruba. The distilled student model; AfroXLMR-Comet successfully captures both the output distribution and internal attention patterns of a larger teacher model (AfroXLMR-Large) while reducing the model size by over 85%. Experimental results demonstrate that our hybrid approach achieves competitive performance compared to the teacher model, maintaining an accuracy within 85% of the original model's performance while requiring substantially fewer computational resources. Our work provides a practical framework for deploying efficient multilingual models in resource-constrained environments, particularly benefiting applications involving African languages.
format Preprint
id arxiv_https___arxiv_org_abs_2502_18020
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AfroXLMR-Comet: Multilingual Knowledge Distillation with Attention Matching for Low-Resource languages
Raju, Joshua Sakthivel
S, Sanjay
Walia, Jaskaran Singh
Raghav, Srinivas
Marivate, Vukosi
Computation and Language
Artificial Intelligence
Information Retrieval
Machine Learning
Language model compression through knowledge distillation has emerged as a promising approach for deploying large language models in resource-constrained environments. However, existing methods often struggle to maintain performance when distilling multilingual models, especially for low-resource languages. In this paper, we present a novel hybrid distillation approach that combines traditional knowledge distillation with a simplified attention matching mechanism, specifically designed for multilingual contexts. Our method introduces an extremely compact student model architecture, significantly smaller than conventional multilingual models. We evaluate our approach on five African languages: Kinyarwanda, Swahili, Hausa, Igbo, and Yoruba. The distilled student model; AfroXLMR-Comet successfully captures both the output distribution and internal attention patterns of a larger teacher model (AfroXLMR-Large) while reducing the model size by over 85%. Experimental results demonstrate that our hybrid approach achieves competitive performance compared to the teacher model, maintaining an accuracy within 85% of the original model's performance while requiring substantially fewer computational resources. Our work provides a practical framework for deploying efficient multilingual models in resource-constrained environments, particularly benefiting applications involving African languages.
title AfroXLMR-Comet: Multilingual Knowledge Distillation with Attention Matching for Low-Resource languages
topic Computation and Language
Artificial Intelligence
Information Retrieval
Machine Learning
url https://arxiv.org/abs/2502.18020