Bilingual Speech-Text Matryoshka Embedding Model
- The paper demonstrates a novel bilingual speech-text Matryoshka embedding that maps both speech and text data into a shared, nested lower-dimensional space for direct cross-lingual retrieval.
- It introduces nested sub-embeddings via vector slicing to enable flexible accuracy–efficiency trade-offs in tasks like retrieval and open-vocabulary keyword spotting.
- The approach leverages joint InfoNCE training with PCA-guided prefix alignment to overcome modality and language barriers while optimizing inference efficiency.
A bilingual speech-text Matryoshka embedding model is a cross-lingual representation system in which speech queries and text documents are mapped into a shared embedding space whose leading coordinates define nested lower-dimensional sub-embeddings. In current arXiv usage, this design appears in two closely related forms: a French–Wolof speech-to-text retrieval model that is described as the first bilingual speech-text Matryoshka embedding model, and a matryoshka audio–text framework for open-vocabulary keyword spotting whose detailed bilingual adaptation is explicitly specified for bilingual keyword spotting (Sy et al., 23 Feb 2026, Jung et al., 20 Jan 2026).
1. Problem setting and research context
Bilingual speech-text Matryoshka models address a joint language barrier and modality barrier. In the French–Wolof case, the motivating problem is that most online knowledge is in French text, while Wolof is primarily oral; conventional information-retrieval pipelines therefore depend on cascaded ASR→MT systems that are costly and suffer error propagation. The proposed alternative is to retrieve French text documents directly from Wolof speech queries through a cross-lingual speech-text embedding model (Sy et al., 23 Feb 2026).
The same representational idea is also relevant to open-vocabulary keyword spotting. In that setting, text-based enrollment replaces fixed-phrase triggers, and audio utterances are matched against keyword text embeddings. Prior utterance-level matching methods learn embeddings at a single fixed dimensionality, whereas Matryoshka Audio-Text Embeddings depart from this design by encoding multiple embedding granularities within a single vector via nested sub-embeddings, or “prefixes” (Jung et al., 20 Jan 2026).
The common premise is that a single embedding should support multiple operating points. This supports flexible accuracy–efficiency trade-offs at inference, which is especially relevant in low-resource or deployment-constrained settings. In retrieval, this means selecting a dimension post hoc according to storage and latency constraints. In keyword spotting, this means preserving a full-dimensional operating point for production inference while concentrating salient cues in lower-dimensional prefixes for regularization and analysis (Sy et al., 23 Feb 2026, Jung et al., 20 Jan 2026).
2. Matryoshka representation principle
The defining operation is nested slicing of a single full vector. In the French–Wolof model, if is the full vector and , then the dimension- representation is
Training is performed jointly across dimensions through the Matryoshka loss
and, for retrieval, the per-dimension objective is InfoNCE with in-batch negatives:
Here is cosine similarity, and the paper does not report the value of (Sy et al., 23 Feb 2026).
In MATE, the same nesting principle is formulated as prefix extraction from a -dimensional utterance representation. With prefix sizes and 0, the prefixes are
1
The design schedule is a power-of-two halving schedule,
2
so that for 3 and 4, 5. The stated rationale is that lower-dimensional prefixes concentrate salient keyword cues, while progressively higher dimensions add detail (Jung et al., 20 Jan 2026).
A distinctive element of MATE is PCA-guided prefix alignment. The teacher used in the paper is derived from a corpus-wide text inner-dependency matrix 6 estimated each epoch,
7
followed by SVD, 8, and the projection
9
The prefix teacher is then
0
Both audio and text prefixes are aligned to this teacher with a sum of MSE and KL terms:
1
2
The total loss is
3
with delayed activation: 4 for epochs 5–6, and 7 for epoch 8 (Jung et al., 20 Jan 2026).
3. Architectural realizations
The current literature contains two principal realizations: a bilingual retrieval model centered on a frozen text Matryoshka encoder with speech fusion, and a dual-encoder audio–text model for keyword spotting that includes an explicit bilingual adaptation path.
| Realization | Core components | Matryoshka dimensions |
|---|---|---|
| Late-Fusion bilingual retrieval | Qwen3-0.6-Embedding, 12-layer Wolof HuBERT, CNN downsampler, projection matrix 9 | 0 |
| Dual bilingual retrieval | Wolof HuBERT, attention pooling with learnable query 1, dimension-specific projections 2, frozen text LLM | 3 |
| MATE audio–text KWS | ECAPA-TDNN audio encoder, G2P front-end, 2-layer bi-LSTM text encoder, CCSP pooling | typically 4 for 5 |
In the French–Wolof Late-Fusion model, the text backbone is Qwen3-0.6-Embedding, an MRL text-embedding LLM capable of emitting embeddings at dimensions 6, 7, 8, 9, 0, and 1, though the paper uses 2. The text-only model is first fine-tuned on bilingual text data using InfoNCE with in-batch negatives. Speech is then integrated by taking features from all 3 Wolof HuBERT layers, concatenating them, reducing the speech sequence length with a CNN by a factor of 4, mapping the result into the LLM embedding space with a learned matrix 5, concatenating the projected speech sequence with prompt token embeddings, and forwarding the combined sequence through the frozen text Matryoshka LLM. Only the CNN and 6 are trained; HuBERT and the text LLM are frozen (Sy et al., 23 Feb 2026).
The alternative bilingual Dual architecture pools speech features 7 into a single vector through attention pooling with a learnable query 8,
9
and then applies dimension-specific linear projections 0. In this setup, HuBERT is unfrozen. Two training variants are reported: Dual–Retrieval, which uses InfoNCE on speech→document retrieval, and Dual–Query Alignment, which distills speech to match its own transcription embedding via cosine similarity and L1 losses (Sy et al., 23 Feb 2026).
MATE uses a more classical dual-encoder KWS architecture. The audio encoder 1 is ECAPA-TDNN with 2 channels, approximately 3M parameters, taking 4-d log-Mel filterbanks with a 5 ms window and 6 ms shift, plus utterance-level mean normalization, and producing a phoneme-level sequence 7. Channel- and context-dependent statistics pooling produces an utterance vector 8. The text encoder 9 consists of a G2P front-end to phoneme tokens, a trainable 0-d embedding lookup, a 1-layer bi-LSTM with 2 hidden units, global average pooling, and a linear layer to produce 3. Both embeddings are 4-normalized before metric learning and inference, with 5 by default (Jung et al., 20 Jan 2026).
For bilingual keyword spotting, the paper gives explicit adaptation options rather than experimental results. Option A is multilingual G2P into a shared phoneme or IPA inventory while retaining the bi-LSTM encoder and adding a learned language-ID token or embedding to the text sequence. Option B is to replace the text encoder with a pretrained multilingual sentence encoder such as LaBSE, XLM-R, or mBERT adapters, followed by projection to 6. The audio encoder is kept unchanged, while training pairs include within-language positives, cross-lingual positives via translations or transliterations, and hard negatives across languages (Jung et al., 20 Jan 2026).
4. Training data, supervision, and inference procedures
The French–Wolof retrieval model is built on large-scale bilingual text and speech curation. The text-only training set combines French mMARCO with queries translated to Wolof, Senegalese French webpages with synthetic French queries translated to Wolof, and French QA datasets with questions translated to Wolof, together with Wolof–French translation pairs for cross-lingual transfer. The resulting scale is 7 Wolof query–French document pairs and 8 French document tokens (Sy et al., 23 Feb 2026).
Speech queries are collected from 9 TB of natural, spontaneous speech from podcasts and radio, excluding read content. The filtering pipeline uses source separation with UVR-MDX-NET, diarization with pyannote, VAD with Silero-VAD, and retains only 0–1 s utterances with DNSMOS quality 2, yielding 3 hours of high-quality Wolof speech queries. For synthetic French documents, the speech is transcribed with a Wolof Speech LLM, filtered by perplexity and lexical diversity, retaining approximately one quarter of the original 4 h speech-text pairs, then translated to French and expanded into synthetic French documents with Gemini-2.5-Flash. An instruction-following dataset is then organized around document retrieval, speech-translation retrieval, and transcription retrieval so that the model can be prompted at inference (Sy et al., 23 Feb 2026).
All French–Wolof models use Sentence-Transformers and are trained for 5 epoch with batch size 6, max length 7, and learning rate 8. The text-only model uses joint InfoNCE over Matryoshka dimensions. The Late-Fusion model uses the same joint InfoNCE loss on the instruction-following speech–text dataset. The Dual–Retrieval model also uses InfoNCE, whereas Dual–Query Alignment uses a joint cosine similarity loss and L1 loss to align speech query vectors with text query vectors (Sy et al., 23 Feb 2026).
In MATE, the training configuration is different because the target task is utterance-level KWS. Training uses King-ASR-066, approximately 9k hours of word-level segments and approximately 0k word classes, with RIR convolution and MUSAN noises as augmentation. The optimizer is AdamW with learning rate 1, weight decay 2, and 3 epochs. Batch composition is 4 utterances from 5 keywords, two utterances per keyword. Per-epoch recomputation of 6, 7, and its SVD supplies the PCA-like teachers used for prefix alignment (Jung et al., 20 Jan 2026).
Inference follows the training design. In French–Wolof retrieval, Wolof speech queries and French documents are encoded at the chosen dimension and ranked by cosine similarity. In MATE KWS, text-based enrollment produces a normalized full-dimensional keyword vector, runtime scoring computes the audio vector and uses
8
with acceptance if 9. For segment-level KWS, a sliding window is applied and the maximum similarity across windows is thresholded. Although optional prefix-level scoring can be computed, all reported MATE results use only the full-dimensional score (Jung et al., 20 Jan 2026).
5. Empirical behavior across retrieval, keyword spotting, and transfer
The strongest published bilingual results are obtained by the French–Wolof Late-Fusion model. On Kallaama-Retrieval-Eval, Late-Fusion at 0 achieves nDCG@5 1 and nDCG@10 2; at 3, 4 and 5; at 6, 7 and 8; and at 9, 00 and 01. The pipelined baseline at 02 reaches 03 and 04, NLLB-LLM2Vec at 05 reaches 06 and 07, Dual–Retrieval at 08 reaches 09 and 10, and Dual–Query Alignment at 11 reaches 12 and 13. On Fleurs-Retrieval-Eval, Late-Fusion at 14 reaches nDCG@5 15 and nDCG@10 16, compared with 17 and 18 for NLLB-LLM2Vec, 19 and 20 for Dual–Retrieval, and 21 and 22 for Dual–Query Alignment (Sy et al., 23 Feb 2026).
The same French–Wolof model generalizes beyond document retrieval. On Urban Bus keyword spotting, Late-Fusion beats Dual across dimensions; for Late-Fusion, F1/Recall at 23 is 24, at 25 is 26, at 27 is 28, and at 29 is 30. On WolBanking77 speech intent detection, a SetFit-style contrastive stage followed by a classifier head yields 31-shot F1/Recall of 32 at 33, and 34-shot performance of 35. The paper states that higher dimensions adapt faster in low-shot regimes, while smaller dimensions catch up with more examples (Sy et al., 23 Feb 2026).
In the monolingual KWS setting, MATE provides complementary evidence for the effectiveness of speech–text Matryoshka representations. On WSJ at 36, MATE improves Average Precision over baselines across multiple deep metric learning objectives: Proxy-BD 37, Proxy-MS 38, CLAT 39, AsyP 40, AdaMS 41, and RPL 42. On LibriPhrase, MATE with RPL and 43 reports EER/AUC of 44 and 45, compared with 46 and 47 for RPL, and the gains are stated to be especially notable on LP_H (Jung et al., 20 Jan 2026).
Ablation results clarify what the Matryoshka mechanism is and is not doing. Under RPL on WSJ, Full-only RPL gives 48, Per-prefix RPL gives 49, Per-prefix RPL + PCA-guided alignment gives 50, and MATE gives 51. The reported interpretation is that applying 52 to all prefixes conflicts with PCA-guided compression, whereas using compression-guided alignment as a prefix regularizer while keeping 53 only at full dimension is best. For the number of prefixes, 54 with 55 peaks on WSJ at 56, while 57 with 58 reaches 59. For alignment composition, MSE-only gives 60, KL-only gives 61, and 62 MSE:KL gives 63 (Jung et al., 20 Jan 2026).
6. Efficiency, limitations, and open research questions
The principal efficiency claim of Matryoshka designs is that one encoder can serve multiple dimensional budgets. In French–Wolof retrieval, smaller dimensions reduce storage and query-time latency, while larger dimensions provide stronger retrieval and low-shot adaptation. In MATE, a single forward pass per modality and a single 64-dimensional similarity score are sufficient at inference; evaluating prefixes requires only slicing the already computed vector, so runtime complexity remains 65 per embedding, and MATE adds no inference overhead (Sy et al., 23 Feb 2026, Jung et al., 20 Jan 2026).
At the same time, current results do not support the misconception that arbitrary truncation preserves all useful information. The French–Wolof paper analyzes cumulative energy ratio,
66
where 67 are eigenvalues of the covariance of embeddings sorted in descending order. The reported finding is that higher dimensions are lower-rank and reach full energy with a small fraction of components, yet still outperform small dimensions in retrieval, suggesting that compression via naive slicing may discard critical information at small 68. The same analysis reports that documents have higher rank than queries, and that at 69 and 70 vectors are effectively full-rank, potentially limiting representation of document information at small dimensions (Sy et al., 23 Feb 2026).
Limitations are task- and language-specific. The French–Wolof study focuses on Wolof–French and notes that generalization to other under-represented languages with different phonologies or scripts is untested. It also relies heavily on synthetic training documents, which may limit robustness and presumes availability of reasonable ASR and MT systems. MATE, by contrast, is evaluated on English-only datasets with English G2P, so generalization to other languages and scripts is not validated. The paper also notes that pronunciation quality and acoustic variability for rare words still matter, and that low-dimensional prefixes may increase false positives on short, phonetically similar distractors if they become over-concentrated (Sy et al., 23 Feb 2026, Jung et al., 20 Jan 2026).
Several research directions are already explicitly formulated. For bilingual KWS, MATE proposes adding cross-lingual alignment terms of the form
71
with a symmetric term for 72 and total objective
73
It further distinguishes shared PCA, language-specific PCA, and hybrid PCA for prefix supervision, and suggests investigating whether low-dimensional prefixes remain language-agnostic while higher dimensions capture language-specific detail. The French–Wolof paper, from a different angle, identifies dynamic structured sparsity and alternative compression schemes as future work because the present rigid slicing strategy appears suboptimal at small dimensions (Jung et al., 20 Jan 2026, Sy et al., 23 Feb 2026).