arXiv reaDer
さまざまな細胞タイプとイメージング モダリティの生細胞画像のユニバーサル インスタンス セグメンテーション ネットワークのトレーニング
Training a universal instance segmentation network for live cell images of various cell types and imaging modalities
さまざまな細胞の種類とイメージング モダリティのユニバーサル セグメンテーション ネットワークをトレーニングするために、最近の調査結果を共有します。私たちの方法は、各コンポーネントを個別に評価できる一般化された U-Net アーキテクチャに基づいて構築されました。従来のバイナリ トレーニング ターゲットを変更して、直接インスタンス セグメンテーション用の 3 つのクラスを含めました。トレーニング スキーム、トレーニング設定、ネットワーク バックボーン、およびセグメンテーション パフォーマンスに関する個々のモジュールに関して、詳細な実験が行われました。提案されたトレーニング スキームは、各データセットから順番にミニバッチを引き出し、最適化ステップの前に勾配を蓄積します。ユニバーサル ネットワークをトレーニングするための鍵は、すべてのデータセットを常に監視することであり、偏りのない方法で各データセットをサンプリングする必要があることがわかりました。私たちの実験はまた、細胞タイプとイメージングモダリティ全体で細胞境界を定義する共通の機能が存在する可能性があることを示唆しています。これにより、訓練されたモデルをまったく目に見えないデータセットに適用できるようになります。クロスエントロピー損失関数の不均一なクラスの重み、適切に設計された学習率スケジューラ、コンテキスト情報のためのより大きな画像トリミング、不均衡なクラスの追加の損失項など、いくつかのトレーニング トリックにより、セグメンテーションのパフォーマンスをさらに向上させることができます。また、グループ正規化レイヤーと Atrous Spatial Pyramid Pooling モジュールは、それぞれより信頼性の高い統計推定と改善されたセマンティック理解のおかげで、セグメンテーションのパフォーマンスが向上することもわかりました。開発したバリアントの 1 つを使用して、IEEE International Symposium on Biomedical Imaging (ISBI) 2021 で開催された 6th Cell Tracking Challenge (CTC) に参加しました。私たちの方法は、一次トラックの最初の提出時に最高の次点者として評価され、要約出版に備えた追加ラウンドの競争でも3位を確保しました.
We share our recent findings in an attempt to train a universal segmentation network for various cell types and imaging modalities. Our method was built on the generalized U-Net architecture, which allows the evaluation of each component individually. We modified the traditional binary training targets to include three classes for direct instance segmentation. Detailed experiments were performed regarding training schemes, training settings, network backbones, and individual modules on the segmentation performance. Our proposed training scheme draws minibatches in turn from each dataset, and the gradients are accumulated before an optimization step. We found that the key to training a universal network is all-time supervision on all datasets, and it is necessary to sample each dataset in an unbiased way. Our experiments also suggest that there might exist common features to define cell boundaries across cell types and imaging modalities, which could allow application of trained models to totally unseen datasets. A few training tricks can further boost the segmentation performance, including uneven class weights in the cross-entropy loss function, well-designed learning rate scheduler, larger image crops for contextual information, and additional loss terms for unbalanced classes. We also found that segmentation performance can benefit from group normalization layer and Atrous Spatial Pyramid Pooling module, thanks to their more reliable statistics estimation and improved semantic understanding, respectively. We participated in the 6th Cell Tracking Challenge (CTC) held at IEEE International Symposium on Biomedical Imaging (ISBI) 2021 using one of the developed variants. Our method was evaluated as the best runner up during the initial submission for the primary track, and also secured the 3rd place in an additional round of competition in preparation for the summary publication.
updated: Thu Jul 28 2022 18:57:30 GMT+0000 (UTC)
published: Thu Jul 28 2022 18:57:30 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アソシエイト