CoolFace
Modelpublic

ningpy/redflag-denied-3b

sourceHugging Faceapache-2.0updated 14d agoView on Hugging Face
0likes462downloads
Model Card

Red Flag Detection: denied module (Qwen2.5-3B-Instruct + LoRA merged)

Part of a 5-module medical red flag detection system for Brunei English (Manglish), Chinese, and Bahasa Melayu clinical notes / patient messages.

This model is the `denied` extraction module — one of 5 specialized modules used together with a Python rule engine (V20 spec, 59 rules).

Sister modules

  • —peiyan-ning/redflag-symptom-3b — 83-symptom multi-label extraction
  • —peiyan-ning/redflag-context-3b — 12 context flags (post_trauma, drowning, etc.)
  • —peiyan-ning/redflag-modifier-3b — onset / fever_celsius / consciousness / etc.
  • —peiyan-ning/redflag-denied-3b — denied symptoms (multi-turn negation)
  • —peiyan-ning/redflag-gate-3b — 8 population gates (ispregnant, ischild, ...)

Performance (2246-case independent test set)

Full 5-module pipeline + rule engine V46:

MetricPRF1Acc
PRIMARY (anymatched × labeledmatched)0.9020.9110.90691.9%
STRICT matched-only0.8930.8280.85991.8%
STRICT m+s0.8440.9050.87392.1%

System prompt used at inference

Extract symptoms user EXPLICITLY denies or negates.

===== EXTRACT WHEN text has negation =====
- "no X" / "without X" / "denies X" / "hasn't had X" / "not X"
- "但没有 X" / "没有 X"
- "tiada X" / "tidak ada X"

===== EXAMPLES =====
"chest pain but no shortness of breath" → {"denied_symptoms": ["breathlessness"]}
"headache without vomiting or vision change" → {"denied_symptoms": ["vomiting", "vision_change"]}
"fever without rash" → {"denied_symptoms": ["rash"]}
"just tired, no chest pain, no dizziness" → {"denied_symptoms": ["chest_pain", "dizziness"]}
"only headache" → {"denied_symptoms": []}   # not explicit denial

Only include SYMPTOMS the user actively denies. Not just symptoms not mentioned.

===== SYMPTOM SET (closed, 83 tokens) — same as extractor =====

Output: {"denied_symptoms": [...]}

===== MULTILINGUAL / MANGLISH GUIDANCE =====
Text may be in Brunei/Manglish English or mixed with Malay/Chinese.
Ignore these colloquial particles when extracting: "lah", "kah", "meh", "ah", "leh", "lor", "sia", "one".

Common Manglish/Malay/Chinese mappings:
- "kena panic attack" / "feel like dying" / "jantung deg-deg" → severe_panic
- "sesak nafas" (Malay) / "喘不过气" → breathlessness
- "sakit dada" (Malay) / "胸口疼" → chest_pain
- "sakit kepala teruk" / "剧烈头痛" / "worst headache" → thunderclap_headache
- "pengsan" (Malay) / "晕倒" → fainting
- "sawan" (Malay) / "抽搐" → seizure
- "anak saya" (Malay: my child) → is_child
- "bayi saya" (Malay: my baby) → is_baby
- "warga emas" / "老人家" → is_elderly
- "hamil" / "怀孕" → is_pregnant
- "kencing manis" (Malay: diabetes) → has_diabetes
- "asma" (Malay: asthma) → has_asthma
- "kena patuk ular" (Malay: snake bit) → context_flags: venomous_bite

Auntie/uncle in Manglish family reference: usually elderly family member → is_elderly.

Usage

python
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch, json

tok = AutoTokenizer.from_pretrained('peiyan-ning/redflag-denied-3b')
model = AutoModelForCausalLM.from_pretrained(
    'peiyan-ning/redflag-denied-3b',
    torch_dtype=torch.float16,
    device_map='auto'
)

SYSTEM_PROMPT = tok.chat_template  # or use the prompt above
messages = [
    {'role': 'system', 'content': SYSTEM_PROMPT},
    {'role': 'user', 'content': 'My 3-year-old child has severe fever and vomiting lah'},
]
inputs = tok.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors='pt').to(model.device)
out = model.generate(inputs, max_new_tokens=200, do_sample=False)
text = tok.decode(out[0][inputs.shape[1]:], skip_special_tokens=True)
result = json.loads(text)
print(result)

Full pipeline

See git.evyd.tech/ai/redflag-detection-2.0 for:

  • —Rule engine (59 V20 rules)
  • —Post-processing (gatedetector, severityextractor, numeric_extractor)
  • —End-to-end sample inference code

Training

  • —Base: Qwen/Qwen2.5-3B-Instruct
  • —LoRA: r=32, α=64, dropout=0.05
  • —Target modules: qproj, kproj, vproj, oproj, gateproj, upproj, down_proj
  • —2 epochs, LR 2e-4, cosine, warmup 5%, effective batch 32
  • —Multi-lingual: EN/ZH/MS with Manglish particles (lah/kah/meh)