arXiv reaDer
遺伝的U-Net:遺伝的アルゴリズムを使用して網膜血管セグメンテーションのために自動的に設計されたディープネットワーク
Genetic U-Net: Automatically Designed Deep Networks for Retinal Vessel Segmentation Using a Genetic Algorithm
最近、手作業で設計された畳み込みニューラルネットワーク(CNN)に基づく多くの方法が、自動網膜血管セグメンテーションで有望な結果を達成しています。ただし、これらのCNNは、複雑な眼底画像で網膜血管をキャプチャする際に制約を受けたままです。セグメンテーションのパフォーマンスを向上させるために、これらのCNNには多くのパラメーターがある傾向があり、過剰適合や計算の複雑さが増す可能性があります。さらに、競合CNNの手動設計には時間がかかり、広範な経験的知識が必要です。ここでは、Genetic U-Netと呼ばれる新しい自動設計方法を提案して、より良い網膜血管セグメンテーションを実現できるが、アーキテクチャベースのパラメータが少ないU字型CNNを生成し、それによって上記の問題に対処します。まず、U字型のエンコーダーデコーダーに基づいて、凝縮された柔軟な検索空間を考案しました。次に、改良された遺伝的アルゴリズムを使用して、検索スペースでパフォーマンスの高いアーキテクチャを特定し、より少ないパラメータで優れたネットワークアーキテクチャを見つける可能性を調査しました。実験結果は、提案された方法を使用して得られたアーキテクチャが、特に元のU-Netパラメータの数の1%未満で、他の最先端モデルよりも大幅に少ないパラメータで優れたパフォーマンスを提供したことを示しています。さらに、実験結果の詳細な調査を通じて、優れた網膜血管セグメンテーションを生成するためのネットワークのいくつかの効果的な操作とパターンが特定されました。
Recently, many methods based on hand-designed convolutional neural networks (CNNs) have achieved promising results in automatic retinal vessel segmentation. However, these CNNs remain constrained in capturing retinal vessels in complex fundus images. To improve their segmentation performance, these CNNs tend to have many parameters, which may lead to overfitting and high computational complexity. Moreover, the manual design of competitive CNNs is time-consuming and requires extensive empirical knowledge. Herein, a novel automated design method, called Genetic U-Net, is proposed to generate a U-shaped CNN that can achieve better retinal vessel segmentation but with fewer architecture-based parameters, thereby addressing the above issues. First, we devised a condensed but flexible search space based on a U-shaped encoder-decoder. Then, we used an improved genetic algorithm to identify better-performing architectures in the search space and investigated the possibility of finding a superior network architecture with fewer parameters. The experimental results show that the architecture obtained using the proposed method offered a superior performance with less than 1% of the number of the original U-Net parameters in particular and with significantly fewer parameters than other state-of-the-art models. Furthermore, through in-depth investigation of the experimental results, several effective operations and patterns of networks to generate superior retinal vessel segmentations were identified.
updated: Fri Jun 11 2021 09:58:57 GMT+0000 (UTC)
published: Thu Oct 29 2020 13:31:36 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アソシエイト