Ant Colony-Based Sequential Hybrid Sampling Method for Handling the Imbalanced Data Problem in Drug-Target Interaction Prediction

• Published in: 2024 7th International Conference of Computer and Informatics Engineering (IC2IE) pp. 1 - 7
• Main Authors: Harahap, Rizky Nurhaliza, Hendrawan, Rahmat, Kurniawan, Isman
• Format: Conference Proceeding
• Language: English
• Published: IEEE 12.09.2024
• Subjects: Accuracy; ACO; Ant colony optimization; Classification algorithms; Diffusion tensor imaging; Drug discovery; Drug target interaction; Drugs; Measurement; Optimization; oversampling; Robustness; Sampling methods; SHS; SMOTE; undersampling
• DOI: 10.1109/IC2IE63342.2024.10748152

Drug Target Interaction (DTI) is crucial in pharmaceutical research, aiding drug discovery, repositioning, and side effect identification. Computational techniques, particularly machine learning, are increasingly used due to cost and time efficiencies. However, DTI prediction faces challenges, notably class imbalance. This study investigates the application of oversampling followed by Ant Colony Optimization (ACO) based undersampling, to address class imbalance problem in DTI prediction using Sequential Hybrid Sampling (SHS) approach. Several oversampling methods such as SMOTE, Random Oversampling (ROS), and ADASYN were used in the experiment to get the best results. The evaluation results show that by using F1-score as the fitness function, it is found that using ACO as undersampling based with SHS approach can improve the performance of the classifier. Gradient boosting is used as the classification method and F1-score, G-Mean, and Balanced Accuracy are used as evaluation metrics. The result show that implementation SHS with SMOTE and ACO were achieved the highest F1-score of 43.24%, Geometric Mean (G-Mean) of 50.75%, and Balanced Accuracy Score (BAS) of 62.41%. These results highlight that the implementation of SHS with ACO can improve the performance of the classifier in DTI prediction compared to the classic oversampling technique. By advancing the understanding and methodologies for handling class imbalance in DTI prediction, this study contributes to the broader goal of enhancing drug discovery and development processes.