Efficient and Interpretable Grammatical Error Correction with Mixture of Experts

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Qorib, Muhammad Reza, Aji, Alham Fikri, Ng, Hwee Tou
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913568444645376
author Qorib, Muhammad Reza
Aji, Alham Fikri
Ng, Hwee Tou
author_facet Qorib, Muhammad Reza
Aji, Alham Fikri
Ng, Hwee Tou
contents Error type information has been widely used to improve the performance of grammatical error correction (GEC) models, whether for generating corrections, re-ranking them, or combining GEC models. Combining GEC models that have complementary strengths in correcting different error types is very effective in producing better corrections. However, system combination incurs a high computational cost due to the need to run inference on the base systems before running the combination method itself. Therefore, it would be more efficient to have a single model with multiple sub-networks that specialize in correcting different error types. In this paper, we propose a mixture-of-experts model, MoECE, for grammatical error correction. Our model successfully achieves the performance of T5-XL with three times fewer effective parameters. Additionally, our model produces interpretable corrections by also identifying the error type during inference.
format Preprint
id arxiv_https___arxiv_org_abs_2410_23507
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient and Interpretable Grammatical Error Correction with Mixture of Experts
Qorib, Muhammad Reza
Aji, Alham Fikri
Ng, Hwee Tou
Computation and Language
Error type information has been widely used to improve the performance of grammatical error correction (GEC) models, whether for generating corrections, re-ranking them, or combining GEC models. Combining GEC models that have complementary strengths in correcting different error types is very effective in producing better corrections. However, system combination incurs a high computational cost due to the need to run inference on the base systems before running the combination method itself. Therefore, it would be more efficient to have a single model with multiple sub-networks that specialize in correcting different error types. In this paper, we propose a mixture-of-experts model, MoECE, for grammatical error correction. Our model successfully achieves the performance of T5-XL with three times fewer effective parameters. Additionally, our model produces interpretable corrections by also identifying the error type during inference.
title Efficient and Interpretable Grammatical Error Correction with Mixture of Experts
topic Computation and Language
url https://arxiv.org/abs/2410.23507