Model Choice Crucial for Causal Inference in Pharmacovigilance

Csaba Kiss, Roland Molontay, Gabriele Pergola· June 17, 2026 View original

Summary

A study evaluated various classification models within the InferBERT framework for pharmacovigilance, finding that domain-specific pre-trained models like BioBERT significantly outperform simpler baselines and larger general LLMs. The research highlights the critical role of model selection and domain-specific pre-training for accurately identifying causal adverse drug events.

Identifying causal adverse drug events (ADEs) from mere correlations is a persistent challenge in pharmacovigilance. The InferBERT framework, which combines transformer models with Do-calculus for causal inference, relies heavily on the performance of its underlying classification model. This study aimed to determine the optimal model choice within this framework. Researchers conducted a comparative analysis using two benchmarks: Analgesics-induced Acute Liver Failure (AILF) and Tramadol-related Mortalities (TRAM). They evaluated four distinct models: XGBoost as a baseline, ALBERT (the original InferBERT model), BioBERT (a transformer pre-trained on biomedical text), and Med-LLaMA (a medical large language model). The findings revealed that BioBERT consistently achieved the highest accuracy on both datasets, demonstrating a clear advantage for domain-specific pre-training. Surprisingly, Med-LLaMA, despite its larger size, underperformed. While post-hoc calibration improved calibration error, its effect on accuracy and causal discovery was mixed. The study concludes that investing in manageable, domain-aware models is more effective for computational pharmacovigilance than simply scaling model size, as BioBERT's superiority also led to stronger concordance with traditional pharmacovigilance signals.

Why it matters

This research provides crucial guidance for developing effective AI systems in pharmacovigilance, emphasizing that domain-specific model pre-training is more impactful than sheer model size for accurate causal inference, leading to safer drug monitoring and better patient outcomes.

How to implement this in your domain

  1. 1Prioritize domain-specific pre-trained models (e.g., BioBERT) over general or larger LLMs for pharmacovigilance tasks.
  2. 2Integrate causal inference frameworks like InferBERT, ensuring the selection of a robust underlying classification model.
  3. 3Conduct thorough comparative analyses of different model architectures and pre-training strategies for specific clinical applications.
  4. 4Evaluate the impact of post-hoc calibration on model performance and causal discovery in real-world pharmacovigilance.

Who benefits

PharmaceuticalsHealthcareMedical ResearchRegulatory Affairs

Key takeaways

  • Model selection is critical for accurate causal inference in pharmacovigilance using frameworks like InferBERT.
  • Domain-specific pre-trained models, such as BioBERT, significantly outperform general LLMs and simpler baselines.
  • Larger model size (e.g., Med-LLaMA) does not guarantee superior performance in specialized domains.
  • Investing in domain-aware models is more effective than simply scaling model size for computational pharmacovigilance.

Original post by Csaba Kiss, Roland Molontay, Gabriele Pergola

"arXiv:2606.17113v1 Announce Type: new Abstract: Distinguishing causal adverse drug events (ADEs) from spurious correlations remains a central challenge in pharmacovigilance. The InferBERT framework integrates transformer models with Do-calculus, but its success hinges on the unde…"

View on X

Originally posted by Csaba Kiss, Roland Molontay, Gabriele Pergola on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses