Enhanced Drug-drug Interaction Prediction Using Adaptive Knowledge Integration

Pengfei Liu, Jun Tao, Zhixiang Ren

arXiv:2603.12885·cs.LG·Published 2026-03-13

Drug-drug interaction event (DDIE) prediction is crucial for preventing adverse reactions and ensuring optimal therapeutic outcomes. However, existing methods often face challenges with imbalanced datasets, complex interaction mechanisms, and poor generalization to unknown drug combinations. To address these challenges, we propose a knowledge augmentation framework that adaptively infuses prior drug knowledge into a large language model (LLM). This framework utilizes reinforcement learning techniques to facilitate adaptive knowledge extraction and synthesis, thereby efficiently optimizing the strategy space to enhance the accuracy of LLMs for DDIE predictions. As a result of few-shot learning, we achieved a notable improvement compared to the baseline. This approach establishes an effective framework for scientific knowledge learning for DDIE predictions.

TopicsLarge Language Models & Materials

Tagsreinforcement-learning

arXiv categoriescs.LG

arXiv abstract pagePDF