Sentence Similarity
sentence-transformers
Safetensors
Transformers
Japanese
modernbert
feature-extraction
mirei
masked-lm
text-embedding
embeddings
retrieval
text-embeddings-inference
Instructions to use iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Transformers
How to use iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT") model = AutoModel.from_pretrained("iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
metadata
language:
- ja
license: mit
library_name: sentence-transformers
pipeline_tag: sentence-similarity
tags:
- mirei
- modernbert
- masked-lm
- transformers
- text-embedding
- embeddings
- feature-extraction
- retrieval
base_model: iamtatsuki05/ModernBERT-JP-0.5B-PT-stage2
datasets:
- cl-nagoya/ruri-dataset-v2-pt
Sentence-ModernBERT-JP-0.5B-PT
English / Japanese
Overview
Sentence-ModernBERT-JP-0.5B-PT は、 iamtatsuki05/ModernBERT-JP-0.5B-PT-stage2 を cl-nagoya/ruri-dataset-v2-pt で弱教師あり対照学習を行った 1280 次元の日本語埋め込みモデルです。
Usage
Requirements
sentence-transformers>=4.1.0
transformers>=4.51.0
accelerate>=1.6.0
sentencepiece>=0.2.0
flash-attn>=2.7.3
Sample Code
import torch
from sentence_transformers import SentenceTransformer
model_name = "iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT"
model_kwargs = {
"torch_dtype": torch.bfloat16,
"attn_implementation": "flash_attention_2",
}
model = SentenceTransformer(model_name, model_kwargs=model_kwargs)
queries = ["ハチワレはどのようなキャラクターですか?"]
docs = [
"ハチワレは、『ちいかわ』に登場する猫風のキャラクターで、明るく社交的、前向きな性格が特徴。ちいかわたちと共に日常を楽しみつつ、討伐などの冒険にも積極的に挑む存在です。",
"うさぎは、天真爛漫でマイペースな性格が特徴のキャラクターで、突飛な行動力と鋭い直感でちいかわたちを引っ張る存在。自由気ままながらも仲間思いな一面を併せ持ちます。",
]
q_emb = model.encode(queries, normalize_embeddings=True)
d_emb = model.encode(docs, normalize_embeddings=True)
scores = model.similarity(q_emb, d_emb)
print(scores)
Model Details
- ベースモデル: iamtatsuki05/ModernBERT-JP-0.5B-PT-stage2
- アーキテクチャ: ModernBERT
- 最大シーケンス長: 8,192トークン
- 埋め込み次元: 1280(Mean Pooling)
- トークナイザ: SentencePiece / 語彙数 102,400
- 位置エンコーディング: RoPE
- 対応言語: 日本語
- 類似度指標: cosine
Model Series
事前学習済みモデルに対して cl-nagoya/ruri-dataset-v2-pt で各サブセット辺り 2,000,000 件のデータで弱教師あり学習したモデル群です。
| ID | Architecture | #Param. | #Param. w/o Emb. |
|---|---|---|---|
| iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT (this model) |
ModernBERT | 679M | 548M |
| iamtatsuki05/Sentence-Llama-Bi-JP-0.5B-PT | Llama | 661M | 530M |
| iamtatsuki05/Sentence-Sarashina-Bi-0.5B-PT | Llama | 661M | 530M |
Licence
このモデルは MIT でライセンスされています。
How to Cite
@article{MIREI
title={同一条件下における Encoder/Decoder アーキテクチャによる文埋め込みの性能分析},
author={岡田 龍樹 and 杉本 徹},
journal={言語処理学会第 32 回年次大会 (NLP2026)},
year={2026}
}
