arXiv reaDer
RaBiT: 結腸ポリープのセグメンテーションのための逆アテンションを備えた双方向特徴ピラミッド ネットワークを使用した効率的なトランスフォーマー
RaBiT: An Efficient Transformer using Bidirectional Feature Pyramid Network with Reverse Attention for Colon Polyp Segmentation
結腸ポリープの自動かつ正確なセグメンテーションは、結腸直腸がんの早期診断に不可欠です。高度な深層学習モデルは、ポリープのセグメンテーションにおいて有望な結果を示しています。ただし、マルチスケールの特徴と汎化機能を表現するには依然として制限があります。これらの問題に対処するために、このホワイト ペーパーでは、エンコーダに軽量の Transformer ベースのアーキテクチャを組み込んで、複数レベルのグローバル セマンティック関係をモデル化するエンコーダ デコーダ モデルである RaBiT を紹介します。デコーダは、さまざまなレベルで特徴マップをより適切に融合し、ポリープ境界を段階的に改善するためのリバース アテンション モジュールを備えたいくつかの双方向特徴ピラミッド レイヤーで構成されています。また、リバース アテンション モジュールを軽量化し、マルチクラス セグメンテーションにより適したものにするためのアイデアも提案します。いくつかのベンチマーク データセットでの広範な実験により、私たちの手法は低い計算複雑性を維持しながら、すべてのデータセットにわたって既存の手法よりも優れたパフォーマンスを発揮することが示されています。さらに、私たちの方法は、トレーニングセットとテストセットが異なる特性を持っている場合でも、データセット間の実験で高い汎化能力を実証します。
Automatic and accurate segmentation of colon polyps is essential for early diagnosis of colorectal cancer. Advanced deep learning models have shown promising results in polyp segmentation. However, they still have limitations in representing multi-scale features and generalization capability. To address these issues, this paper introduces RaBiT, an encoder-decoder model that incorporates a lightweight Transformer-based architecture in the encoder to model multiple-level global semantic relationships. The decoder consists of several bidirectional feature pyramid layers with reverse attention modules to better fuse feature maps at various levels and incrementally refine polyp boundaries. We also propose ideas to lighten the reverse attention module and make it more suitable for multi-class segmentation. Extensive experiments on several benchmark datasets show that our method outperforms existing methods across all datasets while maintaining low computational complexity. Moreover, our method demonstrates high generalization capability in cross-dataset experiments, even when the training and test sets have different characteristics.
updated: Wed Jul 12 2023 19:25:10 GMT+0000 (UTC)
published: Wed Jul 12 2023 19:25:10 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アソシエイト