arXiv reaDer
生物医学画像セグメンテーションのための直交アンサンブルネットワーク
Orthogonal Ensemble Networks for Biomedical Image Segmentation
セマンティックセグメンテーションなどの視覚的タスクに対するディープラーニングベースのアプローチの驚くべきパフォーマンスにもかかわらず、それらは誤って調整された予測を生成することが知られており、重要な意思決定プロセスに有害である可能性があります。アンサンブル学習は、個々のモデルのパフォーマンスを向上させるだけでなく、独立した予測を平均化することにより、モデルの誤校正を減らすことも示しています。このシナリオでは、モデルの多様性が重要な要素になり、個々のモデルをさまざまな機能ソリューションに収束させることが容易になります。この作業では、直交制約によってモデルの多様性を明示的に適用するための新しいフレームワークである直交アンサンブルネットワーク(OEN)を紹介します。提案された方法は、アンサンブルの構成要素間に直交性を誘導すると、モデル全体の多様性が高まるという仮説に基づいています。シーケンシャルアンサンブルトレーニングプロセスを正規化するために使用できる新しいペアワイズ直交性制約に頼り、予測パフォーマンスの向上とモデル出力のキャリブレーションの向上を実現します。提案されたフレームワークを、2つの挑戦的な脳病変セグメンテーションタスク(MR画像における脳腫瘍と白質の高強度セグメンテーション)でベンチマークします。実験結果は、私たちのアプローチがより堅牢で十分に較正されたアンサンブルモデルを生成し、生物医学画像セグメンテーションのコンテキストで困難なタスクを処理できることを示しています。
Despite the astonishing performance of deep-learning based approaches for visual tasks such as semantic segmentation, they are known to produce miscalibrated predictions, which could be harmful for critical decision-making processes. Ensemble learning has shown to not only boost the performance of individual models but also reduce their miscalibration by averaging independent predictions. In this scenario, model diversity has become a key factor, which facilitates individual models converging to different functional solutions. In this work, we introduce Orthogonal Ensemble Networks (OEN), a novel framework to explicitly enforce model diversity by means of orthogonal constraints. The proposed method is based on the hypothesis that inducing orthogonality among the constituents of the ensemble will increase the overall model diversity. We resort to a new pairwise orthogonality constraint which can be used to regularize a sequential ensemble training process, resulting on improved predictive performance and better calibrated model outputs. We benchmark the proposed framework in two challenging brain lesion segmentation tasks --brain tumor and white matter hyper-intensity segmentation in MR images. The experimental results show that our approach produces more robust and well-calibrated ensemble models and can deal with challenging tasks in the context of biomedical image segmentation.
updated: Sat May 22 2021 23:44:55 GMT+0000 (UTC)
published: Sat May 22 2021 23:44:55 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アソシエイト