arXiv reaDer
PS-DeVCEM:弱くラベル付けされたデータに基づくビデオカプセル内視鏡検査のための病理学に敏感な深層学習モデル
PS-DeVCEM: Pathology-sensitive deep learning model for video capsule endoscopy based on weakly labeled data
ビデオカプセル内視鏡(VCE)データにおけるさまざまな結腸疾患のフレームレベルの異常検出とマルチラベル分類のための新しい病理学に敏感な深層学習モデル(PS-DeVCEM)を提案します。私たちの提案するモデルは、いくつかのタイプの疾患によって引き起こされる結腸の見かけの不均一性という重要な課題に対処することができます。私たちのモデルは、注意ベースの深い複数インスタンス学習によって駆動され、詳細なフレームごとの注釈の代わりにビデオラベルを使用して、弱くラベル付けされたデータでエンドツーエンドでトレーニングされます。空間的および時間的特徴は、それぞれResNet50および残余の長短期記憶(残余のLSTM)ブロックを介して取得されます。さらに、学習された時間的注意モジュールは、最終的なラベル予測に対する各フレームの重要性を提供します。さらに、病理学のクラス間の距離を最大化するための自己監視方法を開発しました。定性的および定量的実験を通じて、提案された弱教師あり学習モデルが、3つの最先端のビデオ分析方法と比較して、それぞれ61.6%および55.1%の優れた精度とF1スコア到達を提供することを示します。また、トレーニング中にフレーム注釈情報を使用せずに、病理を伴うフレームを一時的にローカライズするモデルの機能も示します。さらに、最初で最大のVCEデータセットを収集し、ビデオラベルのみで注釈を付けました。データセットには、28,304フレームと14クラスの結腸直腸疾患およびアーチファクトを含む455の短いビデオセグメントが含まれています。この出版物をサポートするデータセットとコードは、当社のホームページで利用できるようになります。
We propose a novel pathology-sensitive deep learning model (PS-DeVCEM) for frame-level anomaly detection and multi-label classification of different colon diseases in video capsule endoscopy (VCE) data. Our proposed model is capable of coping with the key challenge of colon apparent heterogeneity caused by several types of diseases. Our model is driven by attention-based deep multiple instance learning and is trained end-to-end on weakly labeled data using video labels instead of detailed frame-by-frame annotation. The spatial and temporal features are obtained through ResNet50 and residual Long short-term memory (residual LSTM) blocks, respectively. Additionally, the learned temporal attention module provides the importance of each frame to the final label prediction. Moreover, we developed a self-supervision method to maximize the distance between classes of pathologies. We demonstrate through qualitative and quantitative experiments that our proposed weakly supervised learning model gives superior precision and F1-score reaching, 61.6% and 55.1%, as compared to three state-of-the-art video analysis methods respectively. We also show our model's ability to temporally localize frames with pathologies, without frame annotation information during training. Furthermore, we collected and annotated the first and largest VCE dataset with only video labels. The dataset contains 455 short video segments with 28,304 frames and 14 classes of colorectal diseases and artifacts. Dataset and code supporting this publication will be made available on our home page.
updated: Sun Nov 22 2020 15:33:37 GMT+0000 (UTC)
published: Sun Nov 22 2020 15:33:37 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アソシエイト