Text Classification
PEFT
Safetensors
English
intent-classification
ecommerce
lora
qlora
function-calling
Instructions to use while-ai/ecommerce-1b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use while-ai/ecommerce-1b with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("unsloth/gemma-3-1b-it-bnb-4bit") model = PeftModel.from_pretrained(base_model, "while-ai/ecommerce-1b") - Notebooks
- Google Colab
- Kaggle
Zero Proof AI commited on
Model card: leaner client-facing rewrite, e-commerce positioning, rubric_short
Browse files
README.md
CHANGED
|
@@ -14,48 +14,26 @@ language:
|
|
| 14 |
- en
|
| 15 |
---
|
| 16 |
|
| 17 |
-
# zeroproof-ecommerce-1b
|
| 18 |
|
| 19 |
-
|
| 20 |
|
| 21 |
-
|
| 22 |
|
| 23 |
-
|
| 24 |
|
| 25 |
-
|
| 26 |
|
| 27 |
-
|
| 28 |
|
| 29 |
-
|
| 30 |
|
| 31 |
-
|
| 32 |
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
Held-out eval: **1,977 conversations**, zero train/eval leakage, **macro-averaged** (equal weight per intent, so `none` cannot inflate the score). Every model below is scored **identically**, exact-match against gold. Frontier panel is a 412-row balanced subset of the same eval.
|
| 38 |
-
|
| 39 |
-
| Metric (macro) | gemma-1b | **zeroproof-ecommerce-1b** | GPT-5 | Sonnet 5 | Opus 4.8 |
|
| 40 |
-
|---|---|---|---|---|---|
|
| 41 |
-
| Valid JSON | 72.7% | **99.2%** | 100% | 96.1% | 100% |
|
| 42 |
-
| Intent detected | 54.9% | **74.9%** | 85.8% | 89.9% | 92.4% |
|
| 43 |
-
| **Correct core_type** | 12.1% | **70.0%** | 79.5% | 87.0% | 88.2% |
|
| 44 |
-
| Details filled | 1.8% | **44.3%** | 41.4% | 55.7% | 53.2% |
|
| 45 |
-
|
| 46 |
-
- Core-type accuracy goes **12% → 70%** with fine-tuning, reaching **~88% of GPT-5** and **~80% of Opus** on this task, at a fraction of the size, cost, and latency.
|
| 47 |
-
- On **details extraction it matches GPT-5** (44% vs 41%).
|
| 48 |
-
- The eval contains **no easy rows**, every conversation is hard (negation/ambiguity) or long (16+ turns). zeroproof-ecommerce-1b scores **65%** on hard and **78%** on long slices (base: 18% / 11%).
|
| 49 |
-
|
| 50 |
-
Per-intent core_type (zeroproof-ecommerce-1b): `none` 95 · `exchange` 73 · `reverse` 73 · `send` 70 · `recur` 68 · `bill` 57 · `spend` 55.
|
| 51 |
-
|
| 52 |
-
## Cost & latency (measured)
|
| 53 |
-
|
| 54 |
-
Benchmarked on a **Modal L4 GPU** (~$0.80/hr) running `gemma-3-1b` + this LoRA adapter under vLLM (greedy, 256 max new tokens, 64 held-out prompts):
|
| 55 |
-
|
| 56 |
-
- **$0.18 / 1M output tokens** at batched capacity (1,205 tok/s), roughly **55–140× cheaper** than frontier list prices ($10–25/1M).
|
| 57 |
-
- **~1 s per turn** single-request latency (p50 984 ms, p95 1,191 ms).
|
| 58 |
-
- Single-stream with the GPU idle between requests is pricier (~$1.96/1M); real cost sits between, depending on load.
|
| 59 |
|
| 60 |
## Usage
|
| 61 |
|
|
@@ -67,17 +45,21 @@ base = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it")
|
|
| 67 |
tok = AutoTokenizer.from_pretrained("google/gemma-3-1b-it")
|
| 68 |
model = PeftModel.from_pretrained(base, "zero-proof-ai/zeroproof-ecommerce-1b")
|
| 69 |
|
| 70 |
-
# Feed the conversation so far; the model
|
| 71 |
```
|
| 72 |
|
| 73 |
-
The system prompt and
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
|
| 75 |
-
##
|
| 76 |
|
| 77 |
-
-
|
| 78 |
-
- **Data:** 13,282 label-blind role-play conversations, relabeled under a locked written policy and validated by a data gate (deduplicated by message signature, zero train/eval leakage). Deliberately `none`-heavy to mirror real traffic; macro-averaging keeps the reported score honest.
|
| 79 |
-
- **Recipe:** 1 epoch, effective batch 16, LR 2e-4, seq len 12288 (long-context conversations fit). Loss plateaus by ~step 800; held-out accuracy confirms no memorization.
|
| 80 |
|
| 81 |
## License
|
| 82 |
|
| 83 |
-
Adapter released under the base model's **Gemma** license terms.
|
|
|
|
| 14 |
- en
|
| 15 |
---
|
| 16 |
|
| 17 |
+
# zeroproof-ecommerce-1b
|
| 18 |
|
| 19 |
+
**Frontier-approaching e-commerce intent detection in a 1B model.**
|
| 20 |
|
| 21 |
+
`zeroproof-ecommerce-1b` reads a live customer conversation and returns a JSON object with the customer's payment intent and its structured details. It runs inline on live traffic, at a cost and latency where calling a frontier model is not an option.
|
| 22 |
|
| 23 |
+
On a hard, held-out benchmark it reaches **70% exact-match core-intent accuracy, 88% of GPT-5's**, at roughly **1/1000th the parameters** and **$0.18 per 1M tokens**.
|
| 24 |
|
| 25 |
+
Built by ZeroProof. Architecturally a QLoRA fine-tune of `gemma-3-1b-it`; the accuracy comes from our e-commerce intent dataset.
|
| 26 |
|
| 27 |
+
## Benchmarks
|
| 28 |
|
| 29 |
+

|
| 30 |
|
| 31 |
+
Held-out eval of 1,977 conversations, zero train/eval leakage, macro-averaged, exact-match against gold. The frontier panel is scored on a 412-row balanced subset of the same eval.
|
| 32 |
|
| 33 |
+
- Fine-tuning takes core-intent accuracy from **12% to 70%**, landing at **88% of GPT-5** and within 10 points of Opus 4.8.
|
| 34 |
+
- On structured detail extraction it **edges GPT-5** (44% vs 41%).
|
| 35 |
+
- No easy rows: every conversation is adversarial (negation, ambiguity) or long. The model holds **65%** on the hard slice and **78%** on the long slice, where the base model scores 18% and 11%.
|
| 36 |
+
- **$0.18 per 1M output tokens** and **~1s per response**, 55 to 140x below frontier list prices.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
## Usage
|
| 39 |
|
|
|
|
| 45 |
tok = AutoTokenizer.from_pretrained("google/gemma-3-1b-it")
|
| 46 |
model = PeftModel.from_pretrained(base, "zero-proof-ai/zeroproof-ecommerce-1b")
|
| 47 |
|
| 48 |
+
# Feed the conversation so far; the model returns one JSON object per message.
|
| 49 |
```
|
| 50 |
|
| 51 |
+
The system prompt and detail schemas ship in the repo. Use greedy decoding (`temperature=0`).
|
| 52 |
+
|
| 53 |
+
## Training and data
|
| 54 |
+
|
| 55 |
+
A 4-bit QLoRA fine-tune of `gemma-3-1b-it` (via [unsloth](https://github.com/unslothai/unsloth)), one epoch, 12288-token context.
|
| 56 |
+
|
| 57 |
+
The differentiator is the data. ZeroProof builds e-commerce intent training data as **randomized conversational simulations**: role-played customers with independently sampled personas, tones, financial situations, and life stages, including adversarial actors, distilled from frontier models under a locked labeling policy and passed through a structural data gate (deduplicated by message signature, zero train/eval leakage). This corpus, not the fine-tuning recipe, is where the accuracy comes from.
|
| 58 |
|
| 59 |
+
## Intended use
|
| 60 |
|
| 61 |
+
English-language e-commerce and payments support. Accuracy is currently strongest on refunds, exchanges, transfers, and subscriptions; checkout and bill-pay are the focus of the current data round.
|
|
|
|
|
|
|
| 62 |
|
| 63 |
## License
|
| 64 |
|
| 65 |
+
Adapter released under the base model's **Gemma** license terms.
|