본문 바로가기
  • Home

Leveraging the Mamba Architecture for Multi-Class Drug-Drug Interaction Prediction

  • Journal of The Korea Society of Computer and Information
  • Abbr : JKSCI
  • 2025, 30(6), pp.47~54
  • Publisher : The Korean Society Of Computer And Information
  • Research Area : Engineering > Computer Science
  • Received : May 16, 2025
  • Accepted : June 5, 2025
  • Published : June 30, 2025

Jong-Hyeon Ko 1 Kyuri Jo 1

1충북대학교

Accredited

ABSTRACT

In this study, we propose a multi-class drug–drug interaction (DDI) prediction framework based on the Mamba sequence modeling architecture. The proposed model consists of three modules—a global autoencoder (AE1), a Siamese-based encoder (AE2), and a dual encoding module using Mamba (MambaSeparate)—each of which independently captures diverse drug pair features and integrates them for final classification. A subsequent Mamba layer further enhances the representation of drug–drug interactions by enabling information fusion across modules. Notably, the Mamba architecture leverages selective state space modeling to dynamically control information flow based on input significance, offering efficient long-range dependency modeling with lower computational complexity compared to transformers. The use of intermediate embeddings from each module facilitates representation learning at multiple semantic levels. Our framework effectively captures non-linear relationships and sequential interaction patterns between drugs. Experimental results on benchmark datasets demonstrate that the proposed model consistently outperforms existing methods across Tasks 1 to 3, achieving superior accuracy and generalization performance.

Citation status

* References for papers published after 2023 are currently being built.

This paper was written with support from the National Research Foundation of Korea.