{
  "model": "artifacts/google-mbert-pii-4bit/litert-int4-b16/model.int4.weight-only.tflite",
  "dataset": "data/final_splits_audited/combined_clean",
  "split": "test",
  "rows": 64,
  "max_length": 256,
  "batch_size": 16,
  "precision": 0.9577735124760077,
  "recall": 0.9577735124760077,
  "f1": 0.9577735124760077,
  "accuracy": 0.9935635598465157
}

Fixture/runtime verification:

json

{
  "status": "converted_int4_weight_only",
  "fp32_artifact": "artifacts/google-mbert-pii-4bit/litert-int4-b16/model.fp32.tflite",
  "max_length": 256,
  "batch_size": 16,
  "fp32": {
    "name": "litert_fp32",
    "shape": [
      16,
      256,
      39
    ],
    "argmax_match_rate_vs_torch": 1.0,
    "max_abs_diff_vs_torch": 0.00022459030151367188,
    "mean_abs_diff_vs_torch": 8.042213721637381e-07
  },
  "int4_artifact": "artifacts/google-mbert-pii-4bit/litert-int4-b16/model.int4.weight-only.tflite",
  "int4_recipe": "ai_edge_quantizer.recipe.weight_only_wi4_afp32",
  "int4": {
    "name": "litert_int4_weight_only",
    "shape": [
      16,
      256,
      39
    ],
    "argmax_match_rate_vs_torch": 0.91015625,
    "max_abs_diff_vs_torch": 9.644926071166992,
    "mean_abs_diff_vs_torch": 0.20304147899150848
  }
}

Runtime Contract

Use this model behind the same production wrapper as the ONNX/CoreML releases:

sliding-window inference, usually max_length=256 and stride around 96;
offset-based span reconstruction;
whitespace trimming and overlap de-duplication;
deterministic regex/rule assists for email, phone, national ID, postal code, date, card number, and IMEI exclusion;
cue-word correction around Persian labels such as کد ملی, شماره تماس, کدپستی, and ایمیل.

"""Minimal LiteRT wrapper contract.

The exported .tflite graph is fixed shape: batch 16, sequence 256. Tokenize windows to that shape, pad the final short batch by duplicating rows, run ai_edge_litert.Interpreter, then drop the duplicate row during postprocessing. """

Compact and cleaner on mixed Persian/Latin/email text.

Reza2kn/openmed-persian-pii-google-mbert-litert-int4

OpenMed Persian PII Google mBERT LiteRT INT4

Metrics

Runtime Contract

Reza2kn/
openmed-persian-pii-google-mbert-litert-int4