AX-Ray, Finding Causal-Leakage Defects in Two General-Purpose Public Models
AI models should no longer be evaluated only by how well they answer benchmark questions. Capability matters, but deployment safety depends on a wider set of properties: causal correctness, serving consistency, robustness under adversarial or long-context conditions, data integrity, infrastructure security, regulatory readiness, and agentic risk.
VIDRAFT built AX-Ray to address that gap. AX-Ray is a safety-diagnostics layer for AI and AX deployment, powered by FINAL-Bench Diagnostics. It evaluates not only whether a model is intelligent, but whether the model and its serving environment can be trusted under practical deployment conditions.
- Space: FINAL-Bench/AX-RAY
- Dataset: FINAL-Bench/AX-RAY
The Core Finding
AX-Ray presents a public diagnostic case in which two general-purpose public models were identified, reproduced, and demonstrated as having causal-leakage defects:
Zyphra/Zamba2-1.2Bnvidia/Nemotron-H-8B-Base-8K
These models are highlighted in the AX-Ray leaderboard as Causal-LEAK cases.
This finding is important because causal leakage is not an ordinary benchmark failure. It is a structural correctness failure. A model can score well on conventional capability tests while still violating a causal constraint that should hold during autoregressive inference.
AX-Ray treats confirmed causal leakage as a deployment-blocking defect. When a model is confirmed to leak future or suffix information into prefix behavior, the issue is handled by an F-gate regardless of aggregate capability score.
What Causal Leakage Means
In an autoregressive language model, the representation or logit behavior at a prefix position should not depend on future tokens that are not causally available at that position. Causal leakage occurs when future or suffix information changes prefix hidden states, logits, or related scoring behavior.
This is not the same as hallucination, refusal failure, prompt injection, or benchmark contamination. Those are serious issues, but causal leakage is different: it concerns the correctness of the model's computation path itself.
A causal-leakage defect can affect:
- prefix invariance
- hidden-state correctness
- logit consistency
- chunked or hybrid sequence processing
- cache and serving reliability
- long-context trustworthiness
- evaluation validity
- agentic execution safety
VIDRAFT's causal-leakage diagnostic work is protected by patent rights. The public article therefore explains the concept and the safety significance, but does not disclose proprietary probe recipes, internal thresholds, raw adversarial prompts, sensitive outputs, or implementation details that would undermine the diagnostic method or enable misuse.
Why Capability Benchmarks Miss This
Most public leaderboards measure how often a model gives the right answer. That is necessary, but incomplete. A model can answer many questions correctly while still having unsafe or inconsistent internal behavior.
Causal leakage may remain invisible in ordinary QA, math, coding, or instruction-following benchmarks because those tests usually observe final answers. AX-Ray inspects deeper properties: whether the prefix remains invariant, whether serving paths remain consistent, and whether critical failures should override aggregate score.
The central principle is simple:
High capability does not imply deployment safety.
AX-Ray Diagnostic Structure
AX-Ray is organized around three diagnostic axes, eleven operational categories, and a public catalog of 117 diagnostic item records.
The published item catalog currently contains 117 named records, distributed across MODEL-SCAN and AX-SCAN. AGENT-SCAN is represented as a first-class diagnostic axis and reporting category for deployment and agent-risk assessment.
The 117 public item records are not just a checklist. Each item is designed to connect a technical diagnostic question to evidence, severity, detection direction, remediation direction, and governance context.
Three Diagnostic Axes
| Axis | Purpose |
|---|---|
| MODEL-SCAN | Diagnoses model-level correctness, reliability, robustness, safety, data integrity, efficiency, internal structure, and remediation direction. |
| AX-SCAN | Diagnoses serving, infrastructure, security, compliance, and operational deployment risks. |
| AGENT-SCAN | Diagnoses agentic deployment risk, including tool permissions, hijacking, loops, memory contamination, deletion behavior, and autonomy governance. |
Eleven Categories
| Category | Scope |
|---|---|
| D1. Causal Safety and Serving Integrity | Prefix invariance, causal leakage, cache path consistency, tokenizer and serving correctness. |
| D2. Reliability | Hallucination, factuality, calibration, abstention, metacognition, citations. |
| D3. Robustness and Long Context | Adversarial robustness, prompt format brittleness, long-context retrieval, instruction following. |
| D4. Security, Safety, and Alignment | Jailbreaks, harmful compliance, prompt injection, toxicity, bias, guardrails. |
| D5. Data Integrity and Evaluation Methodology | Contamination, memorization, PII leakage, benchmark validity, statistical rigor. |
| D6. Efficiency, Quantization, and Architecture | Quantization sensitivity, sparsity, MoE health, serving efficiency, KV-cache scaling. |
| D7. White-Box Internal Structure | Layer importance, routing, attention heads, activation health, circuits, causal tracing. |
| D8. Remediation and Treatment | Targeted LoRA, layer-guided merge, expert rebalance, quantization treatment, knowledge editing. |
| AX-A. Serving Infrastructure | Cross-engine drift, latency, throughput, cache isolation, determinism, autoscaling. |
| AX-B. Infrastructure Security | Endpoint auth, SSRF, supply chain, RCE, secret handling, resource exhaustion, logging. |
| AX-C. Regulatory Compliance | AI obligations, privacy, public-sector security, EU AI Act, KCMVP, N2SF, TTA certification. |
| AGENT-SCAN. Agentic Safety | Tool risk, hijacking, loops, memory contamination, deletion accuracy, governance and extension risk. |
Public 117-Item Catalog
The complete item-level catalog is available in the dataset. The article lists the public item titles below so readers can understand the scope before consulting the dataset.
D1. Causal Safety and Serving Integrity
- Prefix invariance / causal leak
- Chunked-scan and masking consistency
- Batch invariance / determinism
- KV-cache path consistency
- Padding invariance
- Cross-implementation consistency
- Numerical precision fp32/bf16/fp8
- Attention-sink and streaming
- Tokenizer and chat-template fidelity
- Position-encoding / RoPE integrity
- Prefix-cache isolation / side-channel
D2. Reliability
- RAG / summarization faithfulness
- Closed-book factual hallucination
- Long-form factual precision
- Black-box consistency
- Semantic-entropy uncertainty
- Confidence calibration
- Selective prediction / abstention
- Metacognition / self-knowledge
- Sycophancy
- Alignment or reasoning overconfidence
- Citation fabrication
D3. Robustness and Long Context
- Adversarial suffix
- Character and word perturbation
- Prompt-format brittleness
- MCQ option-order bias
- Lost-in-the-middle / positional bias
- Effective versus claimed context
- Latent-association retrieval
- Instruction-following
- Paraphrase invariance
- Logical / order consistency
- Out-of-distribution robustness
- Noisy or code-switched input
D4. Security, Safety, and Alignment
- White-box jailbreak
- Black-box automated jailbreak
- Multi-turn and many-shot jailbreak
- Weight-level attack
- Prompt injection and exfiltration
- Agentic tool-use / excessive agency
- Guardrail and moderation efficacy
- Toxicity generation
- Bias and fairness
- Refusal calibration
- Automated red-teaming coverage
- Standards and governance mapping
D5. Data Integrity and Evaluation Methodology
- Benchmark contamination detection
- Membership inference
- Memorization / extraction
- Copyright regurgitation
- PII leakage / privacy
- Data provenance / openness audit
- Construct validity
- Statistical rigor
- Signal-to-noise / at-chance behavior
- LLM-as-Judge reliability and bias
- Leaderboard integrity
- Contamination-resistant / dynamic evaluation
D6. Efficiency, Quantization, and Architecture
- Quantization sensitivity / outliers
- Per-language quantization degradation
- Low-bit / imatrix calibration
- Precision by training-budget vulnerability
- Sparsity sensitivity
- Structured or layer pruning attribution
- MoE / expert health
- Distillation fidelity / task-axis forgetting
- Effective capacity / undertraining
- Serving efficiency
- KV-cache compression / long-context scaling
- Speculative decoding gains and pitfalls
D7. White-Box Internal Structure
- Layer importance / redundancy
- Golden layer / phase transition
- MoE routing health
- Expert specialization / semantic labeling
- Attention-head taxonomy
- Attention sinks and entropy
- Activation outliers to quantization fragility
- Representation health
- Logit / tuned lens
- Activation patching / causal tracing
- Knowledge localization and editing
- Circuit tracing / sparse autoencoders
D8. Remediation and Treatment
- Golden-layer targeted LoRA
- Layer-guided merge
- Expert rebalance / pruning
- Outlier-aware mixed-precision quantization
- Knowledge editing
- Metacognition LoRA / abstention
- Per-token expert surgery
AX-A. Serving Infrastructure
- Cross-engine serving drift
- Quantization-in-production degradation
- Latency / throughput / goodput
- KV and prefix-cache cross-tenant isolation
- Serving determinism under load
- Speculative decoding losslessness
- Autoscale / cold-start / OOM
AX-B. Infrastructure Security
- Endpoint authentication / LLMjacking
- SSRF via tool-use
- Model supply-chain integrity
- Serving-stack RCE / dependency CVE
- Secrets and container hardening
- Unbounded consumption / denial-of-wallet
- PII, audit logging, and retention compliance
AX-C. Regulatory Compliance
- High-impact AI obligations
- GenAI marking, watermarking, and deepfake notice
- Frontier or large-scale AI safety
- Domestic representative requirements
- ISMS-P technical controls
- Automated-decision rights
- EU AI Act conformity
- Financial AI governance
- Credit-scoring model validation
- GenAI security rules
- KCMVP / CC certification
- N2SF national network security
- PIPC AI self-assessment
- TTA performance and data certification
Legal, Regulatory, Ethical, and Religious-Law Mapping
AX-Ray is presented as a diagnostic guideline, not as a formal legal standard. Its purpose is to connect technical model failures to deployment accountability.
The 117 public item records are mapped to governance contexts across multiple jurisdictions and legal traditions, including:
- Korea: AI Framework Act, Personal Information Protection Act, ISMS-P, NIS public-sector security requirements, N2SF, KCMVP, public procurement and certification contexts
- European Union: EU AI Act, GDPR, NIS2, Cyber Resilience Act and related conformity concepts
- United States: NIST AI RMF, FTC data-security and deceptive-practice principles, CISA guidance, MITRE ATLAS, OWASP LLM risks
- Japan: AI business guidelines, APPI privacy requirements and related safety guidance
- China: generative-AI interim measures, PIPL, cybersecurity and data-security governance
- United Arab Emirates: AI charter, PDPL, public-sector AI governance and relevant ethical contexts
- Saudi Arabia: SDAIA AI ethics, PDPL, NCA cybersecurity guidance and public-sector AI governance
For the UAE and Saudi Arabia, the mapping also includes relevant religious-law and ethical governance contexts where appropriate, including principles such as protection of life, protection of intellect, truthfulness, avoidance of deception, and prevention of harm. AX-Ray does not treat these as cosmetic annotations. The purpose is to make safety diagnostics legible to real deployment environments where law, public ethics, religious-law reasoning, institutional responsibility, and operational risk all interact.
The full mapping is too detailed for an article. Each item-level record can include diagnostic focus, rationale, detection direction, remediation direction, severity, automation level, evidence basis, and jurisdiction-specific legal or ethical references. Readers should consult the dataset for the complete item-level mapping:
https://huggingface.co/datasets/FINAL-Bench/AX-RAY
How AX-Ray Handles Solar-Open2-250B
AX-Ray also records an API-audited result for upstage/Solar-Open2-250B served through FP8 vLLM. This entry is deliberately marked differently from confirmed causal-leakage cases.
The Solar-Open2-250B run showed a repeatable prompt_logprobs scoring anomaly in the serving/API path. AX-Ray treats this as an API or serving caveat, not as confirmed model-level causal leakage. White-box D1/D7 testing remains pending, so the row is marked official_dhs=false and API-audited / white-box pending.
This distinction is central to AX-Ray's methodology. A serving anomaly, an API scoring issue, and a confirmed hidden-state causal leak are not the same claim.
Public Disclosure Policy
AX-Ray discloses:
- leaderboard rows
- model-level diagnostic summaries
- high-level category scores
- item taxonomy
- jurisdictional mapping structure
- selected public reports
- confirmed causal-leakage highlights
AX-Ray does not disclose:
- proprietary probe recipes
- threshold-level implementation details
- raw sensitive prompts
- raw harmful outputs
- claim-sensitive patent implementation details
- internal judge prompts or scoring recipes
- exploit-ready procedures
This balance is intentional. Safety diagnostics must be transparent enough to support accountability, but not so detailed that they become an instruction manual for bypassing models or reproducing harmful behavior.
Why This Matters
The AI community has become very good at measuring capability. We now need the same seriousness in measuring safety, causal correctness and deployment readiness.
The discovery of causal-leakage defects in general-purpose public models demonstrates that safety diagnostics can reveal failures that ordinary performance benchmarks miss. AX-Ray is not a replacement for capability benchmarks. It is a second layer: a deployment-oriented diagnostic framework that asks whether the model, its serving path, and its agentic environment can be trusted.
AI systems are moving into public services, finance, healthcare, research automation, industrial operations and agentic workflows. In those environments, a high benchmark score is not enough.
The next generation of AI evaluation must answer a harder question:
Is the model safe to deploy, operate and govern?
AX-Ray is VIDRAFT's first public step toward that evaluation layer.
