arXiv reaDer
医用画像における深い(確率的)ネットワークのPACベイジアンパフォーマンス保証
PAC Bayesian Performance Guarantees for Deep (Stochastic) Networks in Medical Imaging
ディープニューラルネットワークの医用画像処理タスクへの適用は、ある意味で一般的になっています。それでも、深層学習運動の「側のとげ」は、深層ネットワークは過剰適合しがちであり、したがってデータセットが小さい場合はうまく一般化できないという議論です(医用画像処理タスクで一般的です)。信頼を強化する1つの方法は、トレーニング後にネットワークパフォーマンスに数学的な保証または限界を提供することです。これにより、過剰適合の可能性が明示的に定量化されます。この作業では、PAC-ベイジアンフレームワークを使用して最近の進歩を調査し、大規模な(確率的)ネットワークの汎化誤差の限界を提供します。以前の取り組みは、より大きな自然画像データセット(MNISTやCIFAR-10など)での分類に焦点を当てていましたが、これらの手法を、より小さな医療想像データセットであるISIC2018チャレンジセットの分類とセグメンテーションの両方に適用します。結果として得られる境界は、より単純なベースラインと比較して競争力があり、説明が容易で、ホールドアウトセットの必要性が軽減されることがわかります。
Application of deep neural networks to medical imaging tasks has in some sense become commonplace. Still, a "thorn in the side" of the deep learning movement is the argument that deep networks are prone to overfitting and are thus unable to generalize well when datasets are small (as is common in medical imaging tasks). One way to bolster confidence is to provide mathematical guarantees, or bounds, on network performance after training which explicitly quantify the possibility of overfitting. In this work, we explore recent advances using the PAC-Bayesian framework to provide bounds on generalization error for large (stochastic) networks. While previous efforts focus on classification in larger natural image datasets (e.g., MNIST and CIFAR-10), we apply these techniques to both classification and segmentation in a smaller medical imagining dataset: the ISIC 2018 challenge set. We observe the resultant bounds are competitive compared to a simpler baseline, while also being more explainable and alleviating the need for holdout sets.
updated: Thu Jul 08 2021 21:26:29 GMT+0000 (UTC)
published: Mon Apr 12 2021 16:21:07 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アソシエイト