ModernBERT-Ja-310M

This repository provides Japanese ModernBERT trained by SB Intuitions.

ModernBERT is a new variant of the BERT model that combines local and global attention, allowing it to handle long sequences while maintaining high computational efficiency. It also incorporates modern architectural improvements, such as RoPE.

Our ModernBERT-Ja-310M is trained on a high-quality corpus of Japanese and English text comprising 4.09T tokens, featuring a vocabulary size of 102,400 and a sequence length of 8,192 tokens.

How to Use

You can use our models directly with the transformers library v4.48.0 or higher:

pip install -U "transformers>=4.48.0"

Additionally, if your GPUs support Flash Attention 2, we recommend using our models with Flash Attention 2.

pip install flash-attn --no-build-isolation

Example Usage

import torch
from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline

model = AutoModelForMaskedLM.from_pretrained("sbintuitions/modernbert-ja-310m", torch_dtype=torch.bfloat16)
tokenizer = AutoTokenizer.from_pretrained("sbintuitions/modernbert-ja-310m")
fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)

results = fill_mask("おはようございます、今日の天気は<mask>です。")

for result in results:
    print(result)
# {'score': 0.54296875, 'token': 16416, 'token_str': '晴れ', 'sequence': 'おはようございます、今日の天気は晴れです。'}
# {'score': 0.17578125, 'token': 28933, 'token_str': '曇り', 'sequence': 'おはようございます、今日の天気は曇りです。'}
# {'score': 0.13671875, 'token': 92339, 'token_str': 'くもり', 'sequence': 'おはようございます、今日の天気はくもりです。'}
# {'score': 0.06494140625, 'token': 2988, 'token_str': '雨', 'sequence': 'おはようございます、今日の天気は雨です。'}
# {'score': 0.019775390625, 'token': 10547, 'token_str': '晴', 'sequence': 'おはようございます、今日の天気は晴です。'}

Model Series

We provide ModernBERT-Ja in several model sizes. Below is a summary of each model.

For all models, the vocabulary size is 102,400, the head dimension is 64, and the activation function is GELU.
The configuration for global attention and sliding window attention consists of 1 layer + 2 layers (global–local–local).
The sliding window attention window context size is 128, with global_rope_theta set to 160,000 and local_rope_theta set to 10,000.

Model Description

We constructed the ModernBERT-Ja-310M model through a three-stage training process, which follows the original ModernBERT.

First, we performed pre-training using a large corpus. Next, we conducted two phases of context length extension.

1. Pre-training

• Training with 3.51T tokens, including Japanese and English data extracted from web corpora.
• The sequence length is 1,024 with best-fit packing.
• Masking rate is 30% (with 80-10-10 rule).

2. Context Extension (CE): Phase 1

• Training with 430B tokens, comprising high-quality Japanese and English data.
• The sequence length is 8,192 with best-fit packing.
• Masking rate is 30% (with 80-10-10 rule).

3. Context Extension (CE): Phase 2

• Training with 150B tokens, comprising high-quality Japanese data.
• The sequence length is 8,192 without sequence packing.
• Masking rate is 15% (with 80-10-10 rule).

The key differences from the original ModernBERT are:

1. It is pre-trained on Japanese and English corpora, leading to a total of approximately 4.09T training tokens.
2. We observed that decreasing the mask rate in Context Extension Phase 2 from 30% to 15% improved the model's performance.

Tokenization and Vocabulary

We use the tokenizer and vocabulary from sbintuitions/sarashina2-13b.
Specifically, we employ a SentencePiece tokenizer with a unigram language model and byte fallback.

We do not apply pre-tokenization using a Japanese tokenizer.
Therefore, users can directly input raw sentences into the tokenizer without any additional preprocessing.

Intended Uses and Limitations

You can use this model for masked language modeling, but it's mostly intended to be fine-tuned on a downstream task. Note that this model is not designed for text generation. When you want to generate a text, please use a text generation model such as Sarashina.

Since the unigram language model is used as a tokenizer, the token boundaries often do not align with the morpheme boundaries, resulting in poor performance in token classification tasks such as named entity recognition and span extraction.

Evaluation

We evaluated our model on 12 datasets, including JGLUE, across various tasks:

• Knowledge-based tasks: JCommonsenseQA (JComQA), RCQA
• Japanese linguistic acceptability classification: JCoLA
• Natural Language Inference (NLI) tasks: JNLI, JSICK, JSNLI, Kyoto University RTE (KU RTE)
• Semantic Textual Similarity (STS) task: JSTS
• Various classification tasks: Livedoor news corpus (Livedoor), LLM-jp Toxicity (Toxicity), MARC-ja, WRIME v2 (WRIME)

These tasks are short-sequence evaluation tasks, and we aligned our settings with those of existing models.
While the maximum sequence length varies across tasks, it does not exceed 512.
We set the sequence length and other experimental configurations per task, ensuring that the settings remain consistent across models.

For hyperparameters, we explored the following ranges:

• Learning rate: {5e-6, 1e-5, 2e-5, 3e-5, 5e-5, 1e-4}
• Number of epochs:
  • Tasks with a large number of instances: {1, 2}
  • Tasks with fewer instances: {3, 5, 10}

In the experiments, we loaded several Japanese models that are publicly available on HuggingFace using AutoModel and constructed classification models by appending a classification head consisting of a linear layer, a GELU activation function, and another linear layer. This was done because HuggingFace's AutoModelForSequenceClassification comes with different implementations for each model, and using them directly would result in classification heads that differ from one model to another.

For the embeddings fed into the classification layer, we used the embedding of the special token at the beginning of the sentence. That is, [CLS] in BERT and <s> in RoBERTa. Note that our model does not perform the next sentence prediction (NSP) task during pretraining, so <s> is added at the beginning of the sentence, not <cls>. Therefore, we used the <s> token for classification.

We conducted evaluations using 5-fold cross-validation. That is, we trained the model on the train set and evaluated it on the validation set. After determining the optimal hyperparameters (learning rate, epochs) based on the average performance on the validation sets, we report the average performance on the test sets with the hyperparameters.

For datasets without predefined splits, we first set aside 10% of the data as the test set and then performed 5-fold cross-validation on the remaining data. For datasets such as some tasks in JGLUE, where only train and validation sets are publicly available, we treated the validation set as the test set and performed 5-fold cross-validation on the remaining data.
For datasets with predefined train, validation, and test sets, we simply trained and evaluated the model five times with different random seeds and used the model with the best average evaluation score on the validation set to measure the final score on the test set.

Evaluation Results

The evaluation results are shown in the table. #Param. represents the number of parameters in both the input embedding layer and the Transformer layers, while #Param. w/o Emb. indicates the number of parameters in the Transformer layers only.

According to our evaluation results, our ModernBERT-Ja-310M archives state-of-the-art performance across the evaluation tasks, even when compared with much larger models.
Despite being a long-context model capable of processing sequences of up to 8,192 tokens, our ModernBERT-Ja-310M also exhibited strong performance in short-sequence evaluations.

Ethical Considerations

ModernBERT-Ja-310M may produce representations that reflect biases.
When you use this model for masked language modeling, it may generate biases or harmful expressions.

License

MIT License

Citation

@misc{
    modernbert-ja,
	author = {Tsukagoshi, Hayato and Li, Shengzhe and Fukuchi, Akihiko and Shibata, Tomohide},
	title = {{ModernBERT-Ja}},
	howpublished = {\url{https://huggingface.co/collections/sbintuitions/modernbert-ja-67b68fe891132877cf67aa0a}},
    url = {https://huggingface.co/collections/sbintuitions/modernbert-ja-67b68fe891132877cf67aa0a},
	year = {2025},
}