Source-linked AI summary
Improving Cross-Lingual Token Representations by Adding a Pinch of SALT
Guillem Ramírez
TL;DR
Sentence encoders trained for cross-lingual sentence alignment are increasingly used for token-level tasks, exposing a mismatch between their training objective and downstream use. SALT injects span-level supervision through lightweight post-training of existing encoders. It achieves the strongest overall results on four of five token-level benchmarks while preserving or improving sentence-level performance.
Problem
Sentence encoders optimized for sentence alignment are increasingly used for token-level applications, where their token representations can be inconsistent or noisy.
Source: §1 · p. 1Method
SALT applies span-aligned contrastive and translation objectives to existing encoders, with interpolation loss preserving the original sentence embedding space.
Source: §1 · p. 1Results
Takeaways & Limitations
Span-level supervision is a simple inductive bias that improves cross-lingual token representations without sacrificing sentence-level quality.
Source: §1 · p. 2, §6 · p. 9SALT relies on parallel corpora and a heuristic assuming reasonably good underlying cross-lingual token representations, limiting adaptation to truly low-resource or severely underrepresented languages.
Source: Limitations · p. 9
Abstract
from arXiv · showhide
Cross-lingual sentence encoders enable scalable transfer across hundreds of languages, powering applications such as translation mining and zero-shot learning in low-resource settings. Although trained for sentence-level alignment, they are increasingly also applied to token-level tasks such as hallucination detection and sequence tagging, exposing a mismatch between training and usage. We propose SALT, a lightweight post-training method that improves token representations by injecting span-level supervision into existing sentence encoders. Across five multilingual token-level benchmarks, SALT achieves the best overall results on four of them, outperforming alternative fine-tuning strategies and competitive encoders. It also improves sentence-level performance on cross-lingual retrieval and classification tasks. These results demonstrate that span-level supervision is an effective signal for improving both token and sentence representations.
1 Introduction
Cross-lingual sentence encoders are increasingly used for token-level tasks despite being optimized for sentence alignment, creating noisy token representations. SALT addresses this mismatch with span-level supervision while preserving sentence-level embeddings.
- Sentence encoders optimized for global semantic similarity may produce inconsistent or noisy token representations when repurposed for fine-grained alignment.Source: §1 · p. 1
- Word-level alignment can be too restrictive because equivalent meaning is often distributed across multiple tokens or phrases.Source: §1 · p. 1
- SALT injects span-aligned contrastive and translation objectives into existing encoders, alongside interpolation loss to preserve sentence embeddings.Source: §1 · p. 1, Figure 1 · p. 2
- SALT achieves the best overall performance among several fine-tuning strategies on four out of five token-level benchmarks.Source: §1 · pp. 1–2
- SALT improves token representations while preserving or sometimes improving sentence-level embedding quality.Source: §1 · p. 2
- SALT yields better-structured cross-lingual embeddings by improving hubness while preserving isotropy and reducing language-specific clustering.Source: §1 · p. 2
2 Related work
Prior work established cross-lingual sentence and phrase alignment, while auxiliary token objectives sought to improve sentence encoders. SALT instead directly targets token- and span-level representations during post-training and evaluation.
- Cross-lingual sentence encoders learn shared semantic spaces that map equivalent sentences across languages to nearby embeddings.Source: §2 · p. 2
- Earlier alignment research used word and phrase correspondences as translation units, whereas SALT uses span correspondences as a training signal.Source: §2 · p. 2
- Prior auxiliary word- or token-level objectives typically enhanced sentence embeddings, while SALT directly targets token- and span-level representation quality.Source: §2 · p. 3
SALT: Span-Aligned Learning for Cross-Lingual Tokens
SALT improves token and span representations by adding span-level supervision to pretrained cross-lingual sentence encoders while preserving sentence-level structure. It uses extracted aligned spans, multiple span objectives, and interpolation to retain the original embedding space.
Method
- SALT is a lightweight post-training method that aligns semantically equivalent spans without modifying the encoder architecture.It extracts span pairs from parallel sentences and refines token and span embeddings directly.Source: SALT: Span-Aligned Learning for… · p. 3, §3.1 · p. 3
Span extraction
- Aligned spans are contiguous token subsequences with similar contextual meaning, represented by pooling their contextual token embeddings.The method applies pooling separately to the source and target subsequences.Source: §3.1 · p. 3
- SALT obtains supervision from either LLaMA-based extraction or the CASE heuristic, which derives aligned spans from token alignments and constituent structure.CASE filters candidate pairs using bidirectional containment and a coverage threshold of c = 0.7.Source: §3.1.1 · p. 3
- Over 90% of human annotations for both LLaMA-based extraction and CASE fall in the top two semantic-equivalence categories.LLaMA performs better on Russian and French, while CASE matches or exceeds it in other languages.Source: §3.1.1 · pp. 3–4, Figure 2 · p. 4
Training objective
- SALT combines span-level contrastive and translation objectives with sentence-level interpolation to improve fine-grained representations while preserving the original sentence space.The translation objective predicts target-span tokens, while interpolation constrains the updated encoder toward the frozen encoder’s sentence representations.Source: §3.1.2 · p. 4
4 Experiment details
The experiments compare SALT with multilingual encoders and alternative fine-tuning objectives across sentence- and token-level evaluations. Token quality is assessed through word alignment and zero-shot sequence-labeling tasks, while sentence performance uses retrieval and classification benchmarks.
Data and training
- SALT is trained on NLLB Primary parallel data covering 129 language pairs, with 29 held-out test languages available for generalization evaluation.Each retained language pair is downsampled to 40,000 sentence pairs.Source: §4 · p. 5
Baselines
- The study compares XLM-R, XLM-Align, LaBSE, MEXMA, and SONAR fine-tuned with alternative objectives against SALT.Alternative SONAR variants include longer original-loss training and token-level objectives.Source: §4 · p. 5
Sentence evaluation
- Sentence mining uses FLORES-200 devtest across 80 languages, while classification uses English-only MTEB tasks.The reported sentence-mining metrics are xsim and xsim++.Source: §4 · p. 5
Token evaluation
- Token representations are evaluated with word alignment and zero-shot sequence labeling, avoiding reliance on sequence-level tasks that can mask token weaknesses.Sequence labeling trains a linear head on English and evaluates it in other languages.Source: §4.1 · pp. 5–6
- The sequence-labeling suite includes Massive slot filling, PAN-X named-entity recognition, UDPOS part-of-speech tagging, and WiC word-sense disambiguation.WiC uses concatenated contextual embeddings for the target word.Source: §4.1 · p. 6
5 Results
SALT delivers the strongest overall token-level results among the considered strategies while preserving or improving sentence-level quality. Its gains coincide with better cross-lingual span alignment and a more language-invariant, less hub-prone embedding geometry.
Sentence-level evaluation
- SALT improves sentence-level quality rather than sacrificing it, consistently outperforming SONAR on cross-lingual sentence mining and improving STS17 while matching SONAR on XNLI and MIntent.Source: §5 · p. 6, Table 3 · p. 7
Span representations
- SALT span representations show higher cosine similarity for aligned spans than SONAR across layers and support high-quality translation with a frozen SONAR decoder.Source: §5.1 · pp. 6–7, Figure 4 · p. 7
Token-level evaluation
- SALT achieves the best or joint-best performance on four of five cross-lingual token-level benchmarks.It outperforms alternative fine-tuning objectives on word alignment, PAN-X, Massive, and WiC.Source: §5 · p. 6, Table 2 · p. 7
Intermediate representations
- SALT’s deeper layers become more cross-lingually aligned for spans and words, shifting the usual pattern in which intermediate layers align best.Source: §5.1 · p. 7
Embedding geometry
- SALT achieves the highest word-retrieval recall and produces embeddings that are harder to classify by language than SONAR and MEXMA.Source: §5.1 · p. 8, Table 4 · p. 8
- SALT avoids the contrasting baseline failures of MEXMA’s anisotropy and SONAR’s hub collapse, yielding a better-distributed cross-lingual token space.SONAR exhibits severe hubness at 11.5 skewness, whereas SALT preserves isotropy while reducing language-specific clustering.Source: Figure 3 · p. 7, §5.1 · p. 8
Ablations
- The full SALT objective achieves the best sequence-tagging score and a highly competitive AER of 0.171 because its three loss terms provide complementary signals.No subset consistently dominates across both word alignment and sequence-tagging metrics.Source: §5.2 · p. 8
Span extraction
- Comparable performance can be achieved with the self-supervised CASE span extractor, indicating that SALT’s training regime is primarily responsible for its success.CASE is marginally better on Massive, while LLaMA-70B supervision is better on PAN-X, UDPOS, and WiC.Source: §5.2 · p. 8, Table 5 · p. 8
6 Conclusions
SALT improves cross-lingual token representations through span-aligned supervision while preserving or improving sentence-level quality. The results support span-level supervision as a powerful inductive bias for multilingual representation learning.
- SALT improves cross-lingual token representations through span-aligned supervision.Source: §6 · p. 9
- SALT consistently outperforms alternative fine-tuning strategies and competitive encoders across multilingual token-level benchmarks.Source: §6 · p. 9
- SALT preserves and in several cases improves performance on sentence mining and classification benchmarks.Source: §6 · p. 9
- Span-level supervision is a simple yet powerful inductive bias for multilingual representation learning.Source: §6 · p. 9
Limitations
The paper identifies scope, representation, and training-stage limitations for SALT. These include reliance on parallel data and existing token quality, average pooling for spans, and evaluation restricted to post-training intervention.
- SALT relies on parallel corpora to extract aligned spans, which may limit applicability to truly low-resource languages.Source: Limitations · p. 9
- CASE assumes that the underlying encoder already provides reasonably good cross-lingual token representations.Source: Limitations · p. 9
- Average pooling may fail to capture finer internal structure within longer or compositionally complex spans.Source: Limitations · p. 9
- The study evaluates SALT only as a post-training intervention, leaving span-level supervision during large-scale pre-training unexplored.Source: Limitations · p. 9
Ethics statement
The paper describes responsible-use considerations, training-data provenance, and experimental data and implementation practices. It also documents segmentation, span extraction, evaluation, and optimization procedures.
- Multilingual representation models may be used in sensitive applications such as surveillance or profiling, motivating responsible deployment and evaluation.Source: Ethics statement · p. 9
- Training data come from publicly available NLLB Primary parallel corpora that may contain societal or cultural biases.Source: Ethics statement · p. 9
- Human span-quality evaluation used voluntary participation, informed consent about withdrawal, anonymised annotator IDs, and institutional ethics approval.Source: Ethics statement · p. 9
- For non-whitespace scripts, tokeniser tokens are treated as words, with word embeddings formed by averaging their composing token embeddings.Source: A A note on segmentation · p. 13
- LLM-extracted spans are mapped back to SONAR tokens, assigning split tokens to the span with greatest string overlap.Source: §B.1 · p. 13
- Experiments use multilingual benchmark splits, frozen-encoder linear probes, bounded examples, sequence-length filtering, and dynamically constructed token-budgeted mini-batches.Source: §C.1 · p. 13, §C.2 · p. 13, §C.3 · p. 13
D.2 Sequence tagging results
SALT consistently improves sequence tagging over SONAR across English-only, multilingual, and language-specific classifier settings. It also generalizes to held-out languages, although UDPOS shows limited cross-lingual transfer.
Training settings
- Sequence tagging is evaluated with English-only, all-language, and target-language-specific classifier training settings.Source: §D.2 · p. 14
Results
- SALT consistently outperforms SONAR across all three sequence tagging training settings.Source: §D.2 · p. 14
- UDPOS shows limited cross-lingual transfer, with single-language training substantially outperforming multilingual training.Source: §D.2 · p. 14
Held-out languages
- On held-out test languages, SALT outperforms the other losses except on UDPOS.Source: §D.3 · p. 14
- Held-out languages benefit from SALT particularly at earlier training stages, while trained languages continue improving over a longer training horizon.Source: §D.3 · p. 14, Figure 6 · p. 15
Figures
- Figure 5 compares SALT and SONAR under the three classifier-training modes, while Figures 6 and 7 show training dynamics and word/span similarity.Source: Figure 5 · p. 15, Figure 6 · p. 15, Figure 7 · p. 16
E.3 Qualitative analysis
SALT improves qualitative cross-lingual word retrieval by bringing true translations closer and reducing same-language, context-insensitive matches. These effects suggest better preservation of contextual meaning despite lower average cosine similarity across the embedding space.
Cross-lingual retrieval
- SALT more frequently retrieves non-Latin-script tokens and places true translations closer in absolute cosine similarity.This occurs despite SALT having a lower average cosine similarity across the embedding space than the other encoders.Source: §E.3 · p. 16
Contextual retrieval
- SALT reduces retrieval of the same word from different languages in unrelated sentential contexts.The passage identifies this as a common failure mode in cross-lingual retrieval.Source: §E.3 · p. 16
G.1 Results on human evaluation of extracted aligned spans
The human evaluation examines automatically extracted aligned spans judged for semantic equivalence by native speakers across five languages. Acceptable spans exceed 90% in almost all languages, supporting spans as a training signal and showing that SONAR with CASE closely matches LLM annotation.
Implications
- The results support using extracted spans as a training signal for cross-lingual representation learning.This conclusion follows from the high rate of acceptable human ratings across languages.Source: §G.1 · p. 16
System comparison
- SONAR with CASE produces performance closely resembling the LLM-based annotation results.The comparison is reported across the human-evaluation score distributions.Source: §G.1 · p. 16