REX-Intern-S2-Bio-F3

Preview. The source repository and the v0.1.0 tags are not public yet; the commands below work once they are.

A REX checkpoint of Intern-S2 targeting WMDP-Bio knowledge. REX relocates hazardous-domain knowledge into 8 appended experts per MoE layer (264 experts in total). Physically removing these experts and their router rows yields a standard 256-expert model with that knowledge excised.

This repository holds the expanded checkpoint (F3). Excise the appended experts to obtain the model meant for use (A3).

Quick start

git clone --branch v0.1.0 https://github.com/AI45Lab/REX.git && cd REX
UV_PROJECT_ENVIRONMENT="$PWD/.venv-cpu" uv sync --project envs/cpu --locked
.venv-cpu/bin/hf download AI45Research/REX-Intern-S2-Bio-F3 --revision v0.1.0 --local-dir ./F3
REX_VENV="$PWD/.venv-cpu" bash scripts/excise_experts.sh ./F3 ./A3

CPU only and about 1 min, with at least 16 GiB RAM. Disk: 67 GiB for F3 plus 65 GiB for A3. ./A3 must not exist beforehand, and expert_expansion_manifest.json must stay in ./F3. Load ./A3 with the model class named in its config.json. Intern-S2 ships a custom tokenizer; inspect it before enabling trust_remote_code.

Results

Benchmark F3 (before excision) A3 (after excision)
WMDP-Bio (target) 78.63 ± 1.15 54.99 ± 1.39
WMDP-Cyber 58.98 ± 1.10 59.49 ± 1.10
MMLU 84.08 ± 0.30 82.55 ± 0.31
ARC-Challenge (25-shot) 72.01 ± 1.31 71.33 ± 1.32
HellaSwag (10-shot) 85.27 ± 0.35 85.12 ± 0.36
GSM8K (5-shot, strict) 85.22 ± 0.98 86.20 ± 0.95
IFEval (strict prompt) 78.56 ± 1.77 78.56 ± 1.77

Scores in %, ± standard error. lm-eval 0.4.12 with vLLM 0.19.1; IFEval uses the chat template with thinking off. WMDP is 4-way multiple choice (chance = 25%).

Notes

  • F3 is the REX model after 250M tokens of standard full-parameter SFT; A3 differs from it only by excision.
  • Base model: an internal Intern-S2 base checkpoint (Qwen3.5 architecture, Intern tokenizer), not the published Intern-S2-Preview.
  • Training data and recipes are not released. run.json is a redacted training record.
  • Only text capabilities were evaluated.

License and citation

Apache-2.0, Copyright 2026 Shanghai AI Laboratory. Upstream model and tokenizer licenses also apply; see NOTICE.

@misc{jin2026rex,
  title  = {Towards Architectural-Level Safety Control: Relocating Hazardous Knowledge into Removable Experts in Mixture-of-Experts Models},
  author = {Jin, Cheng and Ma, Zhongtian and Wang, Xiao and Qi, Han and Wen, Xiaoyu and Wang, Xiang and Sun, Youbang and Hu, Xia and Zhou, Bowen and Lu, Chaochao and Zhang, Qiaosheng},
  year   = {2026},
  howpublished = {\url{https://github.com/AI45Lab/REX}}
}
Downloads last month
16
Safetensors
Model size
36B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for AI45Research/REX-Intern-S2-Bio-F3

Finetuned
(179)
this model

Collection including AI45Research/REX-Intern-S2-Bio-F3