Publish onnx-android artifact

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+model.ort filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,47 @@
+---
+license: apache-2.0
+pipeline_tag: token-classification
+library_name: openmed
+tags:
+- openmed
+- medical-nlp
+---
+
+# OpenMed-NER-PathologyDetect-TinyMed-82M-v1-onnx-android
+
+This model card is rendered from the OpenMed model manifest. Update `models.jsonl` and rerun the publish step instead of editing this file directly.
+
+## Manifest Summary
+
+| Field | Value |
+|---|---|
+| Repository | `OpenMed/OpenMed-NER-PathologyDetect-TinyMed-82M-v1-onnx-android` |
+| Family | NER |
+| Task | token-classification |
+| Languages | en |
+| Tier | Tiny |
+| Parameters | 82M (82,000,000) |
+| Architecture | roberta |
+| Base model | OpenMed/OpenMed-NER-PathologyDetect-TinyMed-82M |
+| Formats | onnx-android, int8, ort-android |
+| License | apache-2.0 |
+| arXiv | [arXiv:2508.01630](https://arxiv.org/abs/2508.01630) |
+| Reproducibility hash | `sha256:8d301f2d5c81b794f5303713cfde89db6299968727de074358898f1b8fa04578` |
+| Released | 2026-07-11 |
+
+## Benchmark
+
+| Dataset | Micro F1 | Recall |
+|---|---:|---:|
+| Not reported | Not reported | Not reported |
+
+## Canonical Labels
+
+`Disease`
+
+## Artifact Format
+
+| Field | Value |
+|---|---|
+| Runtime artifacts | onnx-android, ort-android |
+| Quantization | int8 |

config.json

@@ -0,0 +1,47 @@
+{
+  "transformers_version": "5.12.1",
+  "architectures": [
+    "RobertaForTokenClassification"
+  ],
+  "output_hidden_states": false,
+  "return_dict": true,
+  "dtype": "bfloat16",
+  "chunk_size_feed_forward": 0,
+  "is_encoder_decoder": false,
+  "id2label": {
+    "0": "O",
+    "1": "B-Disease",
+    "2": "I-Disease"
+  },
+  "label2id": {
+    "B-Disease": 1,
+    "I-Disease": 2,
+    "O": 0
+  },
+  "problem_type": null,
+  "vocab_size": 50265,
+  "hidden_size": 768,
+  "num_hidden_layers": 6,
+  "num_attention_heads": 12,
+  "intermediate_size": 3072,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.2,
+  "attention_probs_dropout_prob": 0.2,
+  "max_position_embeddings": 514,
+  "type_vocab_size": 1,
+  "initializer_range": 0.02,
+  "layer_norm_eps": 1e-07,
+  "pad_token_id": 1,
+  "bos_token_id": 0,
+  "eos_token_id": 2,
+  "use_cache": true,
+  "classifier_dropout": 0.2,
+  "is_decoder": false,
+  "add_cross_attention": false,
+  "tie_word_embeddings": true,
+  "_name_or_path": "OpenMed/OpenMed-NER-PathologyDetect-TinyMed-82M",
+  "model_type": "roberta",
+  "position_embedding_type": "absolute",
+  "output_attentions": false,
+  "max_sequence_length": 512
+}

id2label.json

@@ -0,0 +1,5 @@
+{
+  "0": "O",
+  "1": "B-Disease",
+  "2": "I-Disease"
+}

model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a85d39cc2578fb68fd5d8d380534d5b9500eb7d4ba6ed8fc4bb0496bcadb37f
+size 326264048

model.ort

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0914cc56a10f1405e69c49c0da0d99622f8a886c7cb2770b3f2cd8e963be33c0
+size 326319008

model.required_operators_and_types.config

@@ -0,0 +1,11 @@
+# Generated from model/s:
+# - /Users/maziyar/Developer/openmed/dist/onnx-android/OpenMed/OpenMed-NER-PathologyDetect-TinyMed-82M/model.ort
+ai.onnx;1;Not
+ai.onnx;7;And
+ai.onnx;11;Range{"inputs": {"0": ["int64_t"]}}
+ai.onnx;13;Cast{"inputs": {"0": ["bool", "int32_t", "int64_t"]}, "outputs": {"0": ["bool", "int32_t", "int64_t"]}},Concat{"inputs": {"0": ["int64_t"]}},Equal{"inputs": {"0": ["int64_t"]}},Expand{"inputs": {"0": ["bool", "int64_t"]}},Flatten,Gather{"inputs": {"0": ["bool", "float", "int64_t"], "1": ["int64_t"]}},GatherElements{"inputs": {"0": ["int64_t"], "1": ["int64_t"]}},IsNaN{"inputs": {"0": ["float"]}},MatMul{"inputs": {"0": ["float"]}},Softmax{"inputs": {"0": ["float"]}},Transpose{"inputs": {"0": ["float"]}},Unsqueeze
+ai.onnx;14;Add{"inputs": {"0": ["float", "int64_t"]}},CumSum{"inputs": {"0": ["int32_t"]}},Mul{"inputs": {"0": ["int32_t", "int64_t"]}},Reshape
+ai.onnx;15;Shape
+ai.onnx;16;GreaterOrEqual,Where{"inputs": {"1": ["float", "int64_t"]}}
+ai.onnx;17;LayerNormalization
+com.microsoft;1;BiasGelu,FusedMatMul

model_fp16.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84d7f8a43b5a7337c14bf8e65649452f4125e758b8d556c639e3236ae52e91c8
+size 163257319

model_int8.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b101547f13ad973e9ac30f3f370ee67f11be322b2aece3fb240c804a543c144d
+size 198964422

openmed-onnx.json

@@ -0,0 +1,477 @@
+{
+  "format": "openmed-onnx",
+  "format_version": 1,
+  "formats": [
+    "onnx-android",
+    "onnx-int8",
+    "ort-android"
+  ],
+  "task": "token-classification",
+  "family": "roberta",
+  "source_model_id": "OpenMed/OpenMed-NER-PathologyDetect-TinyMed-82M",
+  "config_path": "config.json",
+  "label_map_path": "id2label.json",
+  "artifacts": [
+    {
+      "format": "onnx-android",
+      "path": "model.onnx",
+      "precision": "float32",
+      "metadata": {
+        "profile": "android",
+        "opset": 18,
+        "execution_providers": [
+          "NNAPI",
+          "XNNPACK"
+        ],
+        "inputs": [
+          {
+            "name": "input_ids",
+            "dtype": "int64",
+            "axes": [
+              "batch",
+              "sequence"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              }
+            ]
+          },
+          {
+            "name": "attention_mask",
+            "dtype": "int64",
+            "axes": [
+              "batch",
+              "sequence"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              }
+            ]
+          }
+        ],
+        "outputs": [
+          {
+            "name": "logits",
+            "dtype": "float32",
+            "axes": [
+              "batch",
+              "sequence",
+              "labels"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              },
+              {
+                "kind": "static",
+                "value": 3
+              }
+            ]
+          }
+        ],
+        "operators": [
+          "Constant",
+          "Equal",
+          "Not",
+          "Cast",
+          "CumSum",
+          "Mul",
+          "Add",
+          "Shape",
+          "Gather",
+          "Unsqueeze",
+          "Concat",
+          "Reshape",
+          "ConstantOfShape",
+          "Where",
+          "Expand",
+          "GatherElements",
+          "LayerNormalization",
+          "Range",
+          "GreaterOrEqual",
+          "And",
+          "Flatten",
+          "MatMul",
+          "Transpose",
+          "Softmax",
+          "IsNaN",
+          "Div",
+          "Erf"
+        ],
+        "unsupported_ops": [
+          "CumSum",
+          "GreaterOrEqual",
+          "Flatten",
+          "IsNaN"
+        ],
+        "warnings": [
+          "CumSum is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "GreaterOrEqual is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "Flatten is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "IsNaN is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set"
+        ]
+      }
+    },
+    {
+      "format": "onnx-android",
+      "path": "model_fp16.onnx",
+      "precision": "float16",
+      "metadata": {
+        "profile": "android",
+        "opset": 18,
+        "execution_providers": [
+          "NNAPI",
+          "XNNPACK"
+        ],
+        "inputs": [
+          {
+            "name": "input_ids",
+            "dtype": "int64",
+            "axes": [
+              "batch",
+              "sequence"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              }
+            ]
+          },
+          {
+            "name": "attention_mask",
+            "dtype": "int64",
+            "axes": [
+              "batch",
+              "sequence"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              }
+            ]
+          }
+        ],
+        "outputs": [
+          {
+            "name": "logits",
+            "dtype": "float32",
+            "axes": [
+              "batch",
+              "sequence",
+              "labels"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              },
+              {
+                "kind": "static",
+                "value": 3
+              }
+            ]
+          }
+        ],
+        "operators": [
+          "Constant",
+          "Equal",
+          "Not",
+          "Cast",
+          "CumSum",
+          "Mul",
+          "Add",
+          "Shape",
+          "Gather",
+          "Unsqueeze",
+          "Concat",
+          "Reshape",
+          "ConstantOfShape",
+          "Where",
+          "Expand",
+          "GatherElements",
+          "LayerNormalization",
+          "Range",
+          "GreaterOrEqual",
+          "And",
+          "Flatten",
+          "MatMul",
+          "Transpose",
+          "Softmax",
+          "IsNaN",
+          "Div",
+          "Erf"
+        ],
+        "unsupported_ops": [
+          "CumSum",
+          "GreaterOrEqual",
+          "Flatten",
+          "IsNaN"
+        ],
+        "warnings": [
+          "CumSum is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "GreaterOrEqual is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "Flatten is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "IsNaN is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set"
+        ]
+      }
+    },
+    {
+      "format": "onnx-int8",
+      "path": "model_int8.onnx",
+      "precision": "int8",
+      "metadata": {
+        "format": "onnx-int8",
+        "quantization": {
+          "scheme": "dynamic",
+          "weight_type": "qint8",
+          "op_types": [
+            "MatMul",
+            "Gemm"
+          ],
+          "calibration": "none",
+          "target": "arm64-cpu",
+          "output": "model_int8.onnx"
+        },
+        "profile": "android",
+        "opset": 18,
+        "execution_providers": [
+          "NNAPI",
+          "XNNPACK"
+        ],
+        "inputs": [
+          {
+            "name": "input_ids",
+            "dtype": "int64",
+            "axes": [
+              "batch",
+              "sequence"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              }
+            ]
+          },
+          {
+            "name": "attention_mask",
+            "dtype": "int64",
+            "axes": [
+              "batch",
+              "sequence"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              }
+            ]
+          }
+        ],
+        "outputs": [
+          {
+            "name": "logits",
+            "dtype": "float32",
+            "axes": [
+              "batch",
+              "sequence",
+              "labels"
+            ],
+            "shape": [
+              {
+                "kind": "dynamic",
+                "name": "batch"
+              },
+              {
+                "kind": "dynamic",
+                "name": "sequence"
+              },
+              {
+                "kind": "static",
+                "value": 3
+              }
+            ]
+          }
+        ],
+        "operators": [
+          "Constant",
+          "Cast",
+          "Shape",
+          "Gather",
+          "Equal",
+          "ConstantOfShape",
+          "Flatten",
+          "Not",
+          "Unsqueeze",
+          "Concat",
+          "CumSum",
+          "Range",
+          "Reshape",
+          "Mul",
+          "Add",
+          "Where",
+          "Expand",
+          "GatherElements",
+          "LayerNormalization",
+          "DynamicQuantizeLinear",
+          "MatMulInteger",
+          "Transpose",
+          "GreaterOrEqual",
+          "MatMul",
+          "And",
+          "Softmax",
+          "IsNaN",
+          "Div",
+          "Erf"
+        ],
+        "unsupported_ops": [
+          "Flatten",
+          "CumSum",
+          "DynamicQuantizeLinear",
+          "MatMulInteger",
+          "GreaterOrEqual",
+          "IsNaN"
+        ],
+        "warnings": [
+          "Flatten is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "CumSum is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "DynamicQuantizeLinear is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "MatMulInteger is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "GreaterOrEqual is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set",
+          "IsNaN is not in the Android ONNX Runtime Mobile NNAPI/XNNPACK safe set"
+        ]
+      }
+    },
+    {
+      "format": "ort-android",
+      "path": "model.ort",
+      "precision": "float32",
+      "metadata": {
+        "format": "ort-android",
+        "profile": "android",
+        "source_onnx_path": "model.onnx",
+        "optimization_style": "Fixed",
+        "target_platform": "arm",
+        "op_config_type": "required_operators_and_types",
+        "enable_type_reduction": true,
+        "ort_path": "model.ort",
+        "op_config_path": "model.required_operators_and_types.config",
+        "operators": [
+          "Constant",
+          "Equal",
+          "Not",
+          "Cast",
+          "CumSum",
+          "Mul",
+          "Add",
+          "Shape",
+          "Gather",
+          "Unsqueeze",
+          "Concat",
+          "Reshape",
+          "ConstantOfShape",
+          "Where",
+          "Expand",
+          "GatherElements",
+          "LayerNormalization",
+          "Range",
+          "GreaterOrEqual",
+          "And",
+          "Flatten",
+          "MatMul",
+          "Transpose",
+          "Softmax",
+          "IsNaN",
+          "Div",
+          "Erf"
+        ],
+        "opset": 18
+      }
+    }
+  ],
+  "minimum_opset": 18,
+  "max_sequence_length": 512,
+  "dynamic_shapes": {
+    "batch_axis": "dynamic",
+    "sequence_axis": "dynamic",
+    "shape_buckets": {
+      "strategy": "next_configured_bucket_or_exact",
+      "axes": {
+        "input": [
+          "batch",
+          "sequence"
+        ],
+        "output": [
+          "batch",
+          "sequence"
+        ]
+      },
+      "buckets": [
+        8,
+        16,
+        32,
+        64,
+        128,
+        256,
+        512,
+        1024,
+        2048
+      ],
+      "min_length": 8,
+      "max_length": 2048,
+      "overflow": "exact_length"
+    }
+  },
+  "optimization": {
+    "enabled": false,
+    "minimum_opset": 18
+  },
+  "validation": {},
+  "operator_fallbacks": [],
+  "tokenizer": {
+    "path": ".",
+    "files": [
+      "tokenizer.json",
+      "tokenizer_config.json"
+    ]
+  }
+}

tokenizer_config.json

@@ -0,0 +1,25 @@
+{
+  "add_prefix_space": true,
+  "backend": "tokenizers",
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "errors": "replace",
+  "is_local": false,
+  "local_files_only": false,
+  "mask_token": "<mask>",
+  "max_length": 512,
+  "model_max_length": 512,
+  "pad_to_multiple_of": null,
+  "pad_token": "<pad>",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "sep_token": "</s>",
+  "stride": 0,
+  "tokenizer_class": "RobertaTokenizer",
+  "trim_offsets": true,
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
+  "unk_token": "<unk>"
+}