arXiv reaDer
脳疾患診断のための接続型スタイルにガイドされた文脈表現学習
Connectional-Style-Guided Contextual Representation Learning for Brain Disease Diagnosis
構造磁気共鳴画像法 (sMRI) は大きな臨床的価値を示しており、深層学習 (DL) ベースのコンピューター支援脳疾患診断に広く使用されています。これまでのアプローチは、特定の領域内でのみ重要である可能性がある sMRI の局所的な形状とテクスチャに焦点を当てていました。学習された表現には偽の情報が含まれる可能性があり、他の病気やデータセットでは汎化能力が不十分です。意味のある堅牢な特徴の捕捉を容易にするためには、まず単一のデータ/タスク ドメイン内に制限されない脳の固有のパターンを包括的に理解する必要があります。脳が相互に結合したニューロンの複雑なコネクトームであることを考慮すると、脳内の接続特性には強い生物学的重要性があり、それは複数のドメインにわたって共有され、ほとんどの病理学的情報をカバーします。この研究では、多発性脳疾患の診断に使用される、脳の固有のパターンを捕捉する接続スタイルの文脈表現学習モデル (CS-CRL) を提案します。具体的には、ビジョン トランスフォーマー (ViT) エンコーダーを備えており、プロキシ タスクとしてマスク再構成と接続情報の表現をガイドするグラム マトリックスを活用します。これにより、グローバルなコンテキストの捕捉と、生物学的妥当性を備えた特徴の集約が容易になります。この結果は、CS-CRL が 6 つのデータセットと 3 つの疾患にわたる複数の脳疾患診断タスクにおいて優れた精度を達成し、最先端のモデルを上回るパフォーマンスを示していることを示しています。さらに、CS-CRL が脳ネットワークのような特性をより多く捕捉し、特徴をより適切に集約し、最適化が容易で、ノイズに対してより堅牢であることを実証します。これは理論上の優位性を説明します。私たちのソースコードは間もなく公開される予定です。
Structural magnetic resonance imaging (sMRI) has shown great clinical value and has been widely used in deep learning (DL) based computer-aided brain disease diagnosis. Previous approaches focused on local shapes and textures in sMRI that may be significant only within a particular domain. The learned representations are likely to contain spurious information and have a poor generalization ability in other diseases and datasets. To facilitate capturing meaningful and robust features, it is necessary to first comprehensively understand the intrinsic pattern of the brain that is not restricted within a single data/task domain. Considering that the brain is a complex connectome of interlinked neurons, the connectional properties in the brain have strong biological significance, which is shared across multiple domains and covers most pathological information. In this work, we propose a connectional style contextual representation learning model (CS-CRL) to capture the intrinsic pattern of the brain, used for multiple brain disease diagnosis. Specifically, it has a vision transformer (ViT) encoder and leverages mask reconstruction as the proxy task and Gram matrices to guide the representation of connectional information. It facilitates the capture of global context and the aggregation of features with biological plausibility. The results indicate that CS-CRL achieves superior accuracy in multiple brain disease diagnosis tasks across six datasets and three diseases and outperforms state-of-the-art models. Furthermore, we demonstrate that CS-CRL captures more brain-network-like properties, better aggregates features, is easier to optimize and is more robust to noise, which explains its superiority in theory. Our source code will be released soon.
updated: Thu Jun 08 2023 15:39:27 GMT+0000 (UTC)
published: Thu Jun 08 2023 15:39:27 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アソシエイト