Knowledge, Rules and Their Embeddings: Two Paths towards Neuro-Symbolic JEPA
Abstract
Modern self-supervised predictive architectures excel at capturing complex statistical correlations from high-dimensional data but lack mechanisms to internalize verifiable human logic, leaving them susceptible to spurious correlations and shortcut learning. Conversely, traditional rule-based inference systems offer rigorous, interpretable logic but suffer from discrete boundaries and NP-hard combinatorial explosion. To bridge this divide, we propose a bidirectional neuro-symbolic framework centered around Rule-informed Joint-Embedding Predictive Architectures (RiJEPA). In the first direction, we inject structured inductive biases into JEPA training via Energy-Based Constraints (EBC) and a multi-modal dual-encoder architecture. This fundamentally reshapes the representation manifold, replacing arbitrary statistical correlations with geometrically sound logical basins. In the second direction, we demonstrate that by relaxing rigid, discrete symbolic rules into a continuous, differentiable logic, we can bypass traditional combinatorial search for new rule generation. By leveraging gradient-guided Langevin diffusion within the rule energy landscape, we introduce novel paradigms for continuous rule discovery, which enable unconditional joint generation, conditional forward and abductive inference, and marginal predictive translation. Empirical evaluations on both synthetic topological simulations and a high-stakes clinical use case confirm the efficacy of our approach. Ultimately, this framework establishes a powerful foundation for robust, generative, and interpretable neuro-symbolic representation learning.
Figures & tables
| Model | ID Energy (Valid) | OOD Energy (Invalid) |
|---|---|---|
| Classic JEPA | 0.003 | 0.150 |
| RiJEPA (EBC) | 0.006 | 12.713 |
| Variable | Type | Description & Encoding Schema |
|---|---|---|
| age | Integer | Patient age in years |
| sex | Categorical | Patient sex ( , ) |
| cp | Categorical | Chest pain type ( ) |
| trestbps | Integer | Resting blood pressure on hospital admission (in mm Hg) |
| chol | Integer | Serum cholesterol measurement (in mg/dl) |
| fbs | Categorical | Fasting blood sugar mg/dl ( , ) |
| Risk Profile | Symbolic Logic (Antecedent Consequent) | Support | Confidence |
|---|---|---|---|
| Low Risk | age_group=Middle AND thalach_level=High target_risk=0.0 | 21.5% | 70.8% |
| Low Risk | cp=1.0 AND restecg=2.0 target_risk=0.0 | 4.2% | 83.3% |
| High Risk | cp=4.0 AND slope=3.0 target_risk=1.0 | 4.2% | 100.0% |
| Evaluation Metric | Classic JEPA | RiJEPA | Clinical Significance |
|---|---|---|---|
| Downstream Linear Probing | 100% | 100% | Both models allow for ex-post high-dimensional linear separation. |
| Zero-Shot Logical Accuracy | NA | 100% | Only RiJEPA forms explicit logical clusters, enabling zero-shot diagnosis. |
| Fallback Latent Norm (Ablation) | NA | 5.87 | RiJEPA maintains geometric stability even without symbolic input. |
| Paradigm | Query Logic | Discovered Clinical Profile |
|---|---|---|
| Joint Discovery | Unconditional | IF Normal Chol + Elevated BP + LV Hypertrophy + Downsloping ST + Typical Angina High Risk + Downsloping ST + High FBS + (Mixed Age Markers) |
| Forward Inference | IF Senior THEN… | High Risk + Medium Heart Rate + Exercise Angina + Asymptomatic Pain + LV Hypertrophy |
| Abductive Reasoning | High Risk BECAUSE… | Exercise Angina + Low/Medium Heart Rate + Downsloping ST + Senior Age |
| Marginal-Predictive | Real Patient Vitals (e.g., Young, High Max HR) | Low Risk + Normal ECG + Upsloping ST + High Heart Rate + No Angina |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Pairwise Predictive (RbJEPA) | Energy-Based Constraint (RiJEPA) |
|---|---|
| Positive-only training | Positive + Negative (contrastive) training |
| Learns a conditional mapping | Learns a rule validity landscape |
| Regression objective | Energy-Based / Margin objective |
| Stable and computationally simple | More powerful but harder to optimize |
| No explicit modeling of invalid logic | Explicit modeling of invalid logic |