RobustOps-SFT-4B

GitHub

English · 简体中文 · Models & data collection

GitHub: code & reproduction · Quick start · Training · Evaluation

RobustOps combines supervised fine-tuning and reinforcement learning to improve task completion and fault recovery in 4B enterprise tool agents. The GitHub project provides source code, enterprise runtime adapters, fault injection, training data, recipes and evaluation workflows.

Full-parameter SFT for enterprise task completion and recovery from tool failures. Model repository.

Training data · Trajectories.

Training

Training recipe · Environment and dependencies

692 successful DeepSeek-V4.1-Flash teacher trajectories from 80% Train, with separate demonstration coverage of 357/1,025 clean task–tool-configuration cases (34.83%) and 335/1,025 fault cases (32.68%). Selective decision supervision applies: identified invalid decisions remain in context with their target loss masked, while valid fault encounters and recovery remain learnable. Training uses batch 8, learning rate 3e-6, cosine decay, 8% warmup and three epochs. The adopted checkpoint is epoch 2 / step 174. The initial model is Arctic-MidTool-RL-4B at revision 38b5e742fd20023058cd2cb207358dd033556773.

Training uses two 96 GB GPUs and a 32K context. SFT/RL share 15 validation records for supervised loss diagnostics, without training gradients. Weights are stored as FP32 and evaluated with BF16 compute.

Fixed Test results

Measure Result
Clean task success 28/130 (21.54%)
Actual injection coverage 121/130 (93.08%)
Actual-hit task success 29/121 (23.97%)
Base/SFT/RL/teacher common-hit success 28/116 (24.14%)
Four-model common ignored-filter OOD success 4/20 (20.00%)

Evaluation fixes tasks, state, verifiers, fault mechanisms and repetition limits. Only valid zero-hit runs under these fixed settings receive supplementation, stopping on the first valid hit even when the task fails. Recovery additionally requires matched clean success. The ignored-filter family is held out of project SFT/RL.

Loading

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "haohuiliang/RobustOps-SFT-4B"
revision = "main"
tokenizer = AutoTokenizer.from_pretrained(repo, revision=revision)
model = AutoModelForCausalLM.from_pretrained(
    repo, revision=revision, dtype=torch.bfloat16, device_map="auto",
)

Use the supplied tool-call chat template and tool definitions, with thinking disabled. Running enterprise tasks also requires the RobustOps runtime and adapters.

Scope and limitations

Research in a stateful enterprise simulator. Aggregate evaluation results are reported in the main README.

License and attribution

Apache-2.0 with upstream attribution. Base model · EnterpriseOps-Gym · veRL.

Downloads last month
891
Safetensors
Model size
4B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for haohuiliang/RobustOps-SFT-4B

Finetuned
(1)
this model
Finetunes
2 models
Quantizations
1 model

Collection including haohuiliang/RobustOps-SFT-4B