Query-aware routing for Cross-lingual performance gains in Encoders
Organizations: ConfidentialMind
Abstract
Multilingual encoders can exhibit reduced retrieval effectiveness when queries and relevant documents differ in language, despite strong same-language performance. We investigate whether Finnish and Swedish cross-lingual retrieval can improve while preserving an encoder's existing same-language performance and document index. We combine a query-only low-rank adapter, trained against frozen document embeddings, with deterministic routing based on query and index languages. Cross-language queries use the adapter, while same-language queries use the original encoder. SampoTron, our fine-tuned low-rank (LoRA) adapter alongwith the Nemotron-3-Embed-1B model, improves average retrieval quality across six English, Finnish, and Swedish directions from 0.241 to 0.291 in normalized discounted cumulative gain (nDCG) at rank ten, a 20.9% relative gain on a sampled financial benchmark. All six cross-lingual directions improve, and routing preserves the original same-language performance, including two full-corpus Finnish evaluations. The approach enables selective cross-language specialization with reusable document embedding vectors.
Figures & tables
| Setting | Value |
|---|---|
| LoRA rank / scaling / dropout | / / |
| Target projections | q, k, v, o, gate, up, down |
| Optimizer / learning rate / weight decay | AdamW / / |
| Effective / replay batch size | 256 / 16 |
| Inverse temperature / served dimensions | 20 / 1024 |
| Maximum sequence length / random seed | 1024 tokens / 42 |
| Configuration | FIQA cross | FIQA same | MIRACL-fi | Mr. TyDi-fi |
|---|---|---|---|---|
| Original Nemotron-3-Embed-1B | 0.2406 | 0.3303 | 0.7556 | 0.7284 |
| Query adapter, ungated | 0.2908 | 0.3361 | 0.6746 | 0.6648 |
| SampoTron, routed | 0.2908 | 0.3303 | 0.7556 | 0.7284 |