arXiv reaDer
勾配ブースティング決定木と濃縮ベースの累積ゲインを使用したランク付け学習による複合仮想スクリーニング
Compound virtual screening by learning-to-rank with gradient boosting decision tree and enrichment-based cumulative gain
情報検索で広く使用されている機械学習手法であるLearning-to-rankは、新薬開発の初期段階を加速するために、最近、リガンドベースの仮想スクリーニングの問題に適用されています。ランキング予測モデルは、序数の関係に基づいて学習するため、さまざまな環境からのアッセイデータを統合するのに適しています。化合物スクリーニングにおけるランク予測の既存の研究では、一般に、RankSVMと呼ばれるランク学習法が使用されています。ただし、最近人気が高まっている勾配ブースティング決定木(GBDT)ベースのランク付け学習方法と比較または検証されていません。さらに、正規化割引累積ゲイン(NDCG)と呼ばれるランキング指標は情報検索で広く使用されていますが、予測が他のモデルよりも優れているかどうかを判断するだけです。言い換えると、NDCGは、予測モデルがランダムな結果よりも悪い結果を生成する場合を認識できません。それにもかかわらず、NDCGは、ランク付け学習を使用した化合物スクリーニングのパフォーマンス評価で引き続き使用されます。この研究では、リガンドベースの仮想スクリーニングのために、ラムダランクおよびラムダロスと呼ばれるランク付け損失関数を備えたGBDTモデルを使用しました。結果は、回帰を使用して既存のRankSVMメソッドおよびGBDTモデルと比較されました。また、ランキング予測の良さを適切に評価することを目的とした、新しいランキング指標である正規化エンリッチメント割引累積ゲイン(NEDCG)を提案しました。結果は、ランク付けを学習するGBDTモデルが、さまざまなデータセットでGBDTとRankSVMを使用する既存の回帰手法よりも優れていることを示しました。さらに、NEDCGは、回帰による予測がマルチアッセイ、マルチファミリーデータセットのランダム予測に匹敵することを示し、化合物スクリーニングのパフォーマンスをより直接的に評価するための有用性を示しました。
Learning-to-rank, a machine learning technique widely used in information retrieval, has recently been applied to the problem of ligand-based virtual screening, to accelerate the early stages of new drug development. Ranking prediction models learn based on ordinal relationships, making them suitable for integrating assay data from various environments. Existing studies of rank prediction in compound screening have generally used a learning-to-rank method called RankSVM. However, they have not been compared with or validated against the gradient boosting decision tree (GBDT)-based learning-to-rank methods that have gained popularity recently. Furthermore, although the ranking metric called Normalized Discounted Cumulative Gain (NDCG) is widely used in information retrieval, it only determines whether the predictions are better than those of other models. In other words, NDCG is incapable of recognizing when a prediction model produces worse than random results. Nevertheless, NDCG is still used in the performance evaluation of compound screening using learning-to-rank. This study used the GBDT model with ranking loss functions, called lambdarank and lambdaloss, for ligand-based virtual screening; results were compared with existing RankSVM methods and GBDT models using regression. We also proposed a new ranking metric, Normalized Enrichment Discounted Cumulative Gain (NEDCG), which aims to properly evaluate the goodness of ranking predictions. Results showed that the GBDT model with learning-to-rank outperformed existing regression methods using GBDT and RankSVM on diverse datasets. Moreover, NEDCG showed that predictions by regression were comparable to random predictions in multi-assay, multi-family datasets, demonstrating its usefulness for a more direct assessment of compound screening performance.
updated: Mon Aug 29 2022 16:25:41 GMT+0000 (UTC)
published: Wed May 04 2022 16:36:24 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト