comsat-embed-ja-0.3b-preview

comsat-embed-ja-0.3b-preview is an encoder-based embedding model developed by Sionic AI, optimized for Japanese semantic retrieval tasks. Trained on over 1.5M Japanese examples, it encodes queries and documents into vectors so that the most relevant documents can be found by similarity. The model is designed to provide high-quality text representations for real-world information retrieval scenarios, including document search, question answering, knowledge base retrieval, and enterprise semantic search. At only 0.3B parameters, it delivers robust performance across Japanese search environments where accurate semantic matching is essential.

Highlights

• Japanese-specialized — trained on 1.5M+ Japanese examples; achieves state-of-the-art average NDCG@10 (0.7785) on the 11-task JMTEB(v2) retrieval benchmark among the compared models with ≤4B parameters.
• Compact & efficient — 0.3B (310M) parameters, well suited to cost-efficient, low-latency deployment.
• Long context — handles inputs up to 8,192 tokens.
• Asymmetric encoding — queries and documents are encoded with their respective prefixes (検索クエリ: / 検索文書: ).
• Embeddings — 768-dimensional, mean-pooled and L2-normalized, compared with cosine similarity.

Usage

First install the Sentence Transformers library

pip install -U sentence-transformers

Sentence Transformers Usage

⚠️ Encode queries with the query prompt and documents with the document prompt. (Both use their own prefix; skipping the prompt slightly degrades retrieval quality.)

from sentence_transformers import SentenceTransformer

model = SentenceTransformer("sionic-ai/comsat-embed-ja-0.3b-preview")

queries   = ["日本の首都はどこですか？"]
documents = ["日本の首都は東京です。"]

# Prefixes ("検索クエリ: " / "検索文書: ") are applied automatically by prompt_name
q_emb = model.encode(queries,   prompt_name="query",    normalize_embeddings=True)
d_emb = model.encode(documents, prompt_name="document", normalize_embeddings=True)

# Option: sentence-transformers 5.x helper API (equivalent)
# q_emb = model.encode_query(queries)
# d_emb = model.encode_document(documents)

scores = q_emb @ d_emb.T   # cosine similarity
print(scores)

Transformers Usage

import torch
import torch.nn.functional as F
from torch import Tensor
from transformers import AutoTokenizer, AutoModel


def mean_pool(last_hidden_states: Tensor, attention_mask: Tensor) -> Tensor:
    mask = attention_mask.unsqueeze(-1).to(last_hidden_states.dtype)
    return (last_hidden_states * mask).sum(dim=1) / mask.sum(dim=1).clamp(min=1e-9)


# Prepend the prefixes manually when using plain Transformers
queries   = ["検索クエリ: 日本の首都はどこですか？"]
documents = ["検索文書: 日本の首都は東京です。"]
input_texts = queries + documents

tokenizer = AutoTokenizer.from_pretrained("sionic-ai/comsat-embed-ja-0.3b-preview")
model = AutoModel.from_pretrained("sionic-ai/comsat-embed-ja-0.3b-preview")

batch_dict = tokenizer(
    input_texts,
    padding=True,
    truncation=True,
    max_length=8192,
    return_tensors="pt",
)
with torch.no_grad():
    outputs = model(**batch_dict)

embeddings = mean_pool(outputs.last_hidden_state, batch_dict["attention_mask"])
embeddings = F.normalize(embeddings, p=2, dim=1)
scores = embeddings[:1] @ embeddings[1:].T   # cosine similarity
print(scores.tolist())

JMTEB Retrieval Benchmark

• NLPJournalTitleAbsRetrieval.V2: Japanese academic paper retrieval — retrieve the abstract from the paper title.
• NLPJournalTitleIntroRetrieval.V2: Japanese academic paper retrieval — retrieve the introduction from the title.
• NLPJournalAbsIntroRetrieval.V2: Japanese academic paper retrieval — retrieve the introduction from the abstract.
• NLPJournalAbsArticleRetrieval.V2: Japanese academic paper retrieval — retrieve the article body from the abstract.
• MintakaRetrieval: A multilingual open-domain QA retrieval dataset (Japanese subset).
• JaGovFaqsRetrieval: A Japanese government FAQ retrieval dataset.
• JaqketRetrieval: A Japanese open-domain quiz QA retrieval dataset.
• MultiLongDocRetrieval: A long-document retrieval dataset (Japanese subset).
• JaCWIRRetrieval: A Japanese casual web information retrieval dataset.
• MIRACLRetrieval: A Wikipedia-based retrieval dataset (Japanese subset).
• MrTidyRetrieval: A Wikipedia-based Japanese retrieval dataset.

Performance (JMTEB v2 Retrieval, NDCG@10)

Only models with ≤4B parameters are shown. All scores are NDCG@10; for multilingual tasks the Japanese subset is used (Mintaka/MultiLongDoc/MIRACL=ja, MrTidy=japanese).

Avg is the mean over the 11 JMTEB(v2) retrieval tasks (higher is better). Reproduction: evaluated with the MTEB/JMTEB retrieval pipeline (NDCG@10, full corpus); the query prompt (検索クエリ: ) is applied to queries and the document prompt (検索文書: ) to documents.

License

• Model weights: cc-by-nc-4.0 (non-commercial use).