Informative Relational Learning for Adverse Reaction Prediction with Enhanced Generalization to Novel Drugs
Shuge Sun, Dalin Zhang, Hongjun Chu, Xinyi GongAbstract
Motivation
Accurate prediction of adverse drug reactions (ADRs) is essential for drug safety surveillance, and recent advances in machine learning with heterogeneous biomedical information have improved predictive performance. However, two challenges remain: current methods often learn inadequate ADR representations that fail to capture dependencies among ADRs, and generalize poorly to novel drugs.
Results
To obtain informative ADR embeddings, we construct a multi-source, multi-relational ADR graph that integrates hierarchical structure and empirical ADR co-occurrence, and apply a relational graph convolutional network (R-GCN) to learn relation-aware ADR representations. To enhance generalization to novel drugs, we exploit the hierarchical structure of the Anatomical Therapeutic Chemical (ATC) classification to link drugs via shared higher-level categories for effective knowledge transfer and model these relations with an R-GCN. We further introduce a Conditional Domain Adversarial Network (CDAN) to reduce distribution shifts between known and novel drugs by aligning features conditioned on predicted ADR labels, learning domain-invariant yet task-relevant representations. Additionally, to exploit similar ADR patterns among related drugs, we introduce a dual-branch mixture-of-experts (Dual-MoE) module where each expert captures ADR commonalities within a drug category in one branch, while a separate branch models global patterns. Extensive experiments show that our method consistently outperforms seven baselines, achieving F1 improvements of 4.3% and 4.7% over the best baseline on two datasets, respectively, with more balanced precision–recall trade-offs. It also improves AUC on uncommon ADRs by 7% more than on common ADRs, and remains more robust under data sparsity, with more gradual performance degradation as training data decreases.
Availability and implementation
The code of our model is available at https://github.com/fzsdb/Knowledge-guided-ADR-prediction.git.