Huazhong Agricultural University
Drug-drug interaction event text generation from two molecular structures, conditioned on biological functions selected for each drug in the pair.
When patients take several drugs at once, the resulting drug-drug interactions (DDIs) can alter absorption, metabolism, or pharmacodynamic effect in ways that matter clinically. Computational DDI methods have overwhelmingly framed this as classification: assign a drug pair to one of a fixed set of interaction-event labels, then look the label up in a separate table to recover what it means. That framing breaks down against databases such as DDInter2.0, whose event descriptions are detailed free text rather than a countable set of classes, and it forces clinicians to translate a label back into a mechanism.
PKAG-DDI reframes the task as generation: given two drug molecules, produce the natural-language description of the interaction event directly. Its central observation is that each drug plays a distinct role in an interaction — one may be a gastric alkalinizer, the other pH-sensitive — and that naming those two biological functions as explicit context substantially improves what a language model can say about the pair. Because such functions are specialist knowledge that is usually unavailable for a new drug pair, the model learns to select them itself. It was developed by Ziyan Wang, Zhankun Xiong, Feng Huang, and Wen Zhang at Huazhong Agricultural University and published at ACL 2025.
PKAG-DDI is a task-specialized system rather than a general-purpose backbone. It composes existing pretrained components — a frozen molecular graph encoder and graph-to-sequence adapter taken from MolTC, and a Galactica 1.3B language model — and adds two new trained modules on top. Trained once, it generates interaction text for drug pairs never seen during training without per-pair retraining.
Drug pairs enter as both SMILES strings and 2D molecular graphs. The graph encoder and adapter are frozen at MolTC's pretrained weights; the adapter and the Galactica 1.3B language model are fine-tuned on DDI-event data using LoRA, with training on two A800 GPUs. The selector is trained separately, supervised by BM25-matched gold biological functions, and combines graph-derived node prototype representations with molecular fingerprints through an MLP classifier. The authors recommend K = 2, since roughly 97% of drugs in the data carry fewer than three annotated biological functions.
Two datasets were constructed from professional databases. The MecDDI set covers 1,685 drugs, 1,061 biological function types, and 152,922 DDIs spanning 103 event types; the DDInter2.0 set covers 1,683 drugs and 152,887 DDIs with longer, less summarizable descriptions. On DDInter2.0 the model reaches 92.39 BLEU-2 under random split versus 83.18 for MolTC, with larger relative margins in the inductive settings (44.18 vs. 37.34 BLEU-2 cold-start). Scored as classification on MecDDI it remains competitive with dedicated classifiers (95.05% accuracy random split, 44.39% cold-start). An oracle variant supplied with ground-truth biological functions scores far higher, quantifying how much headroom remains in the selection step.
The intended user is a pharmacologist or clinical decision-support developer who needs an interpretable statement about why two drugs interact, not just that they do — particularly for compounds too new to appear in curated interaction databases. Stratifying DDInter2.0 by clinical risk level, generation quality was highest for major-risk interactions, the tier where an explanation is most consequential.
PKAG-DDI is an early demonstration that DDI-event prediction can be posed as text generation without sacrificing the classification accuracy that established methods deliver, and that explicitly modeling each drug's asymmetric role is what makes that work. Its limits are stated plainly by the authors: the selector operates over a fixed knowledge set and cannot be updated dynamically, so a drug whose biological function falls outside that set is out of scope, and all evaluation is retrospective against database annotations rather than prospective clinical validation. Code is public; trained checkpoints and the processed datasets are distributed through a HuggingFace dataset repository, though the repository carries no license file.
Papers that recently cited this model.
The most-cited papers that cite this model.
Providers that host PKAG-DDI for inference, fine-tuning, or weight download.
No providers recorded yet. Browse all providers
Not enough data