arXiv reaDer
適合率が証明可能な適合率曲線の下の領域の確率的最適化
Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergence
ROC (AUROC) および適合率-再現率曲線 (AUPRC) の下の領域は、不均衡な問題の分類パフォーマンスを評価するための一般的なメトリックです。 AUROC と比較して、AUPRC は非常に不均衡なデータセットに対してより適切なメトリックです。 AUROC の確率的最適化は広く研究されていますが、AUPRC の原理に基づく確率的最適化はめったに検討されていません。この作業では、深層学習用に AUPRC を最適化するための原理的な技術的方法を提案します。私たちのアプローチは、AUPRC の不偏点推定量である平均精度 (AP) の最大化に基づいています。目的を、外部レベルの確率変数に依存する内部関数を持つ従属構成関数の合計にキャストします。確率的組成最適化の最近の進歩を活用することにより、穏やかな条件下で証明可能な収束が保証された効率的な適応および非適応の確率アルゴリズムを提案します。画像とグラフのデータセットに関する広範な実験結果は、提案された方法が、AUPRC に関して不均衡な問題に対して以前の方法よりも優れていることを示しています。私たちの知る限り、私たちの仕事は、証明可能な収束で AUPRC を最適化する最初の試みを表しています。
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 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.
updated: Sun Jun 06 2021 15:20:25 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アソシエイト