首页 > AI前沿 > BLADE: Distilled LLM Regularization for Calibrated Knowledge Graph Completion

BLADE: Distilled LLM Regularization for Calibrated Knowledge Graph Completion

arXiv机器学习 2026-08-26 10:33 4 阅读 查看原文

Knowledge graph completion models optimize ranking, although many downstream applications require calibrated probabilities.

We present BLADE, a variational model that separates latent truth from graph recording and distills offline language-model judgments into a frozen teacher regularizer.

The LLM is absent during inference.

Posterior samples provide predictive probabilities and epistemic uncertainty, while the compact teacher remains available only as an optional triage factor.

Performance on Benchmarks

Across five benchmarks, BLADE remains competitive under a common ranking protocol and reduces adaptive ECE by a macro-average of 60.1% relative to deep ensembles and 78.1% relative to temperature-scaled RotatE.

On identical FB15k-237 candidate sets, BLADE also improves ECE, Brier score, and NLL over validation-selected histogram binning and a matched generative ComplEx2 model, with these improvements persisting on a prespecified near-miss pool.

Controlled Missingness

Under controlled injected missingness, the full triage score achieves a mean AUC-PR of 0.863, compared with 0.805 for its strongest non-teacher variant.

Leakage Stress Tests

Leakage stress tests show that aligned semantics matter, but they cannot exclude knowledge acquired during LLM pretraining.

We therefore claim calibration only for the declared candidate distributions, not for all unobserved triples.