arXiv reaDer
証明可能な収束を伴う適合率-再現率曲線下の領域の確率的最適化
Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergence
ROC (AUROC) および適合率-再現率曲線 (AUPRC) の下の領域は、不均衡な問題の分類パフォーマンスを評価するための一般的なメトリックです。 AUROC と比較して、AUPRC は非常に不均衡なデータセットに適したメトリックです。 AUROC の確率的最適化は広く研究されてきましたが、AUPRC の原理に基づく確率的最適化はほとんど検討されていません。この作業では、深層学習用に AUPRC を最適化するための原理に基づいた技術的方法を提案します。私たちのアプローチは、AUPRC の偏りのないポイント推定量である平均精度 (AP) を最大化することに基づいています。外部レベルの確率変数に依存する内部関数を使用して、目的を依存構成関数の合計にキャストします。確率的合成最適化の最近の進歩を活用することにより、穏やかな条件下で証明可能な収束を保証する SOAP という名前の効率的な適応型および非適応型確率アルゴリズムを提案します。画像とグラフのデータセットに関する広範な実験結果は、提案された方法が AUPRC に関して不均衡な問題に対する以前の方法よりも優れていることを示しています。私たちの知る限り、私たちの研究は証明可能な収束で AUPRC を最適化する最初の試みです。 SOAP は ~https://libauc.org/ の libAUC ライブラリに実装されています。
Areas under ROC (AUROC) and precision-recall curves (AUPRC) are common metrics for evaluating classification performance for imbalanced problems. Compared with AUROC, AUPRC is a more appropriate metric for highly imbalanced datasets. While stochastic optimization of AUROC has been studied extensively, principled stochastic optimization of AUPRC has been rarely explored. In this work, we propose a principled technical method to optimize AUPRC for deep learning. Our approach is based on maximizing the averaged precision (AP), which is an unbiased point estimator of AUPRC. We cast the objective into a sum of dependent compositional functions with inner functions dependent on random variables of the outer level. We propose efficient adaptive and non-adaptive stochastic algorithms named SOAP with provable convergence guarantee under mild conditions by leveraging recent advances in stochastic compositional optimization. Extensive experimental results on image and graph datasets demonstrate that our proposed method outperforms prior methods on imbalanced problems in terms of AUPRC. To the best of our knowledge, our work represents the first attempt to optimize AUPRC with provable convergence. The SOAP has been implemented in the libAUC library at~https://libauc.org/.
updated: Wed Apr 12 2023 22:42:58 GMT+0000 (UTC)
published: Sun Apr 18 2021 06:22:21 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アソシエイト