arXiv reaDer
深層学習ベースの医用画像セグメンテーションに最適なオプティマイザーの選択
Selecting the Best Optimizers for Deep Learning based Medical Image Segmentation
この作業の目標は、心臓画像のセグメンテーションのコンテキストでディープ ラーニングに最適なオプティマイザーを特定し、効果的な最適化戦略を使用してセグメンテーション ネットワークを設計する方法に関するガイダンスを提供することです。適応学習は、より大きな学習率 (LR) から開始して徐々に減少させることで、収束を高速化するのに役立ちます。 Momentum オプティマイザは、加速スキーム カテゴリ内のニューラル ネットワークを迅速に最適化するのに特に効果的です。この記事では、これら 2 種類のアルゴリズム (LR とモーメンタム オプティマイザー、つまりモメンタム レート (MR)) の間の潜在的な相互作用を明らかにすることで、1 つの設定で SGD アルゴリズムの 2 つのバリアントを調べます。基本オプティマイザーとして循環学習を使用し、学習率とモメンタム率の最適値を統合することをお勧めします。 MRIおよびCTスキャンからの心臓構造の医用画像セグメンテーションの重要な問題の下で、LRとMRの関係を調査しました。 MICCAI 2017 の ACDC チャレンジからの心臓画像データセットと、心臓画像セグメンテーションの問題で成功することが示された 4 つの異なるアーキテクチャを使用して実験を行いました。私たちの包括的な評価は、提案されたオプティマイザーが、単一および複数オブジェクトのセグメンテーション設定の両方で、同等またはより低い計算コストで、深層学習文献の他のオプティマイザーよりも優れた結果 (ダイス メトリックで 2% 以上の改善) を達成したことを示しました。加速された最適化方法と適応最適化方法の組み合わせは、医療画像セグメンテーションのパフォーマンスに劇的な影響を与える可能性があるという仮説を立てました。この目的のために、深層学習ベースの医用画像セグメンテーションにおける効率と精度の問題に対処するために、新しい巡回最適化法 (CLMR) を提案しました。提案された戦略は、適応オプティマイザーと比較してより良い一般化をもたらしました。
The goal of this work is to identify the best optimizers for deep learning in the context of cardiac image segmentation and to provide guidance on how to design segmentation networks with effective optimization strategies. Adaptive learning helps with fast convergence by starting with a larger learning rate (LR) and gradually decreasing it. Momentum optimizers are particularly effective at quickly optimizing neural networks within the accelerated schemes category. By revealing the potential interplay between these two types of algorithms (LR and momentum optimizers or momentum rate (MR) in short), in this article, we explore the two variants of SGD algorithms in a single setting. We suggest using cyclic learning as the base optimizer and integrating optimal values of learning rate and momentum rate. We investigated the relationship of LR and MR under an important problem of medical image segmentation of cardiac structures from MRI and CT scans. We conducted experiments using the cardiac imaging dataset from the ACDC challenge of MICCAI 2017, and four different architectures shown to be successful for cardiac image segmentation problems. Our comprehensive evaluations demonstrated that the proposed optimizer achieved better results (over a 2% improvement in the dice metric) than other optimizers in deep learning literature with similar or lower computational cost in both single and multi-object segmentation settings. We hypothesized that combination of accelerated and adaptive optimization methods can have a drastic effect in medical image segmentation performances. To this end, we proposed a new cyclic optimization method (CLMR) to address the efficiency and accuracy problems in deep learning based medical image segmentation. The proposed strategy yielded better generalization in comparison to adaptive optimizers.
updated: Sun Feb 05 2023 03:49:27 GMT+0000 (UTC)
published: Sun Feb 05 2023 03:49:27 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アソシエイト