Merging aligned feed-forward sublayers into tied weights can remove over a third of a Transformer's feed-forward parameters with only small performance losses, after a short fine-tuning.
PyMarian: Fast Neural Machine Translation and Evaluation in Python
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
The deep learning language of choice these days is Python; measured by factors such as available libraries and technical support, it is hard to beat. At the same time, software written in lower-level programming languages like C++ retain advantages in speed. We describe a Python interface to Marian NMT, a C++-based training and inference toolkit for sequence-to-sequence models, focusing on machine translation. This interface enables models trained with Marian to be connected to the rich, wide range of tools available in Python. A highlight of the interface is the ability to compute state-of-the-art COMET metrics from Python but using Marian's inference engine, with a speedup factor of up to 7.8$\times$ the existing implementations. We also briefly spotlight a number of other integrations, including Jupyter notebooks, connection with prebuilt models, and a web app interface provided with the package. PyMarian is available in PyPI via $\texttt{pip install pymarian}$.
fields
cs.CL 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Merging Feed-Forward Sublayers for Compressed Transformers
Merging aligned feed-forward sublayers into tied weights can remove over a third of a Transformer's feed-forward parameters with only small performance losses, after a short fine-tuning.