arXiv reaDer
PlutoNet: 修正部分デコーダと一貫性トレーニングを備えた効率的なポリープ セグメンテーション ネットワーク
PlutoNet: An Efficient Polyp Segmentation Network with Modified Partial Decoder and Consistency Training
深層学習モデルを使用して、専門家が気付かないポリープの数を最小限に抑え、介入中に検出されたポリープを正確にセグメント化します。最先端のモデルは優れたパフォーマンスを発揮しますが、必要なパラメーターが多すぎるため、リアルタイム アプリケーションで問題が発生する可能性があります。この問題に対処するために、PlutoNet と呼ばれる新しいセグメンテーション モデルを提案します。これは、2,626,537 個のパラメーターのみを必要とし、いくつかのデータセットで最先端のモデルよりも優れています。 PlutoNet を使用して、一貫性を強化するために損失を組み合わせてトレーニングされる共有エンコーダー、変更された部分デコーダー、および補助デコーダーで構成される新しい一貫性トレーニング アプローチを提案します。軽量アーキテクチャを使用し、修正された部分デコーダーを提案します。これは、部分デコーダーと、必要なパラメーターの数をさらに減らし、セマンティックの詳細をキャプチャする本格的な接続を組み合わせたものです。非対称畳み込みを使用してさまざまなポリプ サイズを処理し、スクイーズおよび励起ブロックを使用して各特徴マップを重み付けします。次に、変更された部分デコーダーと補助デコーダーの損失を組み合わせることで一貫性を強化します。これにより、学習した関連するセマンティックの詳細を失うことなく、エンコーダーの表現を改善できます。このようにして、偽陽性率を減らすことができます。大規模な実験を行って、モデルが最先端のモデルよりも優れており、いくつかのデータセットにうまく一般化できることを示します。私たちのモデルは、EndoScene データセット、Etis データセット、2018 Data Science Bowl Challenge データセット、および Kvasir Instrument データセットで最先端のモデルよりも優れています。また、モデルの各コンポーネントの有効性を示すためにアブレーション研究も行います。
Deep learning models are used to minimize the number of polyps that goes unnoticed by the experts and to accurately segment the detected polyps during interventions. Although state-of-the-art models perform well, they require too many parameters, which can pose a problem with real-time applications. To address this problem, we propose a novel segmentation model called PlutoNet which requires only 2,626,537 parameters while outperforming state-of-the-art models on several datasets. With PlutoNet, we propose a novel consistency training approach that consists of a shared encoder, the modified partial decoder, and the auxiliary decoder that are trained with a combined loss to enforce consistency. We use a lightweight architecture and propose the modified partial decoder, which is a combination of the partial decoder and full-scale connections that further reduces the number of parameters required, as well as captures semantic details. We use asymmetric convolutions to handle varying polyp sizes and then we weigh each feature map by using a squeeze and excitation block. Then, we enforce consistency by combining the loss of the modified partial decoder and the auxiliary decoder, which helps improve the encoder's representations without losing learned relevant semantic details. This way we are able to reduce false positive rates. We perform extensive experiments to show that our model outperforms the state-of-the-art models and is able to generalize well to several datasets. Our model outperforms the state-of-the-art models on the EndoScene dataset, Etis dataset, 2018 Data Science Bowl Challenge dataset, and Kvasir Instrument dataset. We also perform an ablation study to show the effectiveness of each component of our model.
updated: Sat Nov 12 2022 23:14:38 GMT+0000 (UTC)
published: Wed Apr 06 2022 20:29:00 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アソシエイト