Simply Trainable Nearest Neighbour Machine Translation with GPU Inference

Hossam Amer, Abdelrahman Abouelenin, Mohamed Maher +3

Nearest neighbor machine translation is a successful approach for fast domain adaption, which interpolates the pre-trained transformers with domain-specific token-level k-nearest-n…