perplexity-ai/pplx-embed-v2-late-0.6b · Hugging Face
Key Points
- 1The `pplx-embed-v2-late` family consists of multimodal ColBERT-based retrievers built on Qwen3.5 that utilize bidirectional attention to produce 128-dimensional token-level embeddings.
- 2These models, available in 0.6B and 9B parameter sizes, share a unified embedding space and leverage a distilled LEAF-style objective to achieve high performance on ViDoRe and text-based benchmarks.
- 3Implementation is streamlined via the `sentence-transformers` library, allowing users to perform efficient MaxSim similarity scoring across text and image inputs without requiring custom Python code.
The pplx-embed-v2-late series represents a family of multimodal, late-interaction (ColBERT-style) retriever models built upon the Qwen3.5 architecture. Designed for retrieval tasks involving text, images, and visual documents, these models operate by generating granular, 128-dimensional vector representations for each individual token, which are subsequently compared using the MaxSim similarity metric.
Core Methodology
The models employ a late-interaction paradigm, which facilitates high-fidelity retrieval by preserving token-level information rather than compressing entire documents into a single dense vector. The core similarity mechanism between a query and a document is defined by the MaxSim operator, which computes the maximum cosine similarity of each query vector against the set of document vectors:This approach allows the model to capture fine-grained contextual alignment between query terms and document segments.
Training and Architecture
The family comprises two primary variants:- pplx-embed-v2-late-0.6b (340M active parameters): A compact model that is fully fine-tuned to ensure high efficiency.
- pplx-embed-v2-late-9b (7.4B active parameters): A larger, more robust variant where the final eight transformer layers are fully fine-tuned, and the preceding layers and vision encoder are adapted via Low-Rank Adaptation (LoRA).
Both models share a unified embedding space, which enables cross-model compatibility; an index constructed with the 9B model can be queried using the 0.6B model. The models were trained via distillation from a proprietary 18B ColBERT teacher model, utilizing pair and triplet data. The learning process incorporated a token-level LEAF-style objective to align the student model with the teacher’s fine-grained interaction patterns.
Technical Implementation and Constraints
The models utilize bidirectional attention for processing inputs. A critical implementation detail is the placement of special Query/Document tokens; while other systems like PyLate place these at the second position, thepplx-embed-v2-late models expect them at the first position. Furthermore, the models support input batches of either text-only or image-only content, but explicitly do not support mixed text-and-image inputs within a single batch.