arXiv reaDer
ブレイン ネットワーク トランスフォーマー
Brain Network Transformer
人間の脳は、一般に、関心領域 (ROI) のネットワークと、脳機能と精神障害を理解するためのそれらの接続としてモデル化されています。最近、Transformer ベースのモデルは、グラフを含むさまざまな種類のデータで研究されており、パフォーマンスが大幅に向上することが示されています。この作業では、脳ネットワーク分析のための Transformer ベースのモデルを研究します。データの固有の特性に基づいて、ブレイン ネットワークを固定サイズと順序のノードを持つグラフとしてモデル化します。これにより、(1) 接続プロファイルをノード機能として使用して、自然で低コストの位置情報を提供し、(2) ペアを学習することができます。ダウンストリーム分析タスクを予測する個人全体の効率的な注意の重みを使用した、ROI 間の接続強度。さらに、自己教師ありソフト クラスタリングと正規直交射影に基づく正規直交クラスタリング読み出し操作を提案します。この設計は、ROI のグループ間で同様の動作を決定する基礎となる機能モジュールを考慮しており、区別可能なクラスター認識ノードの埋め込みと有益なグラフの埋め込みにつながります。最後に、ABIDE の唯一の公開されている大規模な脳ネットワーク データセットで評価パイプラインを再標準化し、異なるモデルの意味のある比較を可能にします。実験結果は、パブリック ABIDE と制限付き ABCD データセットの両方で、提案された Brain Network Transformer の明確な改善を示しています。実装は https://github.com/Wayfear/BrainNetworkTransformer で入手できます。
Human brains are commonly modeled as networks of Regions of Interest (ROIs) and their connections for the understanding of brain functions and mental disorders. Recently, Transformer-based models have been studied over different types of data, including graphs, shown to bring performance gains widely. In this work, we study Transformer-based models for brain network analysis. Driven by the unique properties of data, we model brain networks as graphs with nodes of fixed size and order, which allows us to (1) use connection profiles as node features to provide natural and low-cost positional information and (2) learn pair-wise connection strengths among ROIs with efficient attention weights across individuals that are predictive towards downstream analysis tasks. Moreover, we propose an Orthonormal Clustering Readout operation based on self-supervised soft clustering and orthonormal projection. This design accounts for the underlying functional modules that determine similar behaviors among groups of ROIs, leading to distinguishable cluster-aware node embeddings and informative graph embeddings. Finally, we re-standardize the evaluation pipeline on the only one publicly available large-scale brain network dataset of ABIDE, to enable meaningful comparison of different models. Experiment results show clear improvements of our proposed Brain Network Transformer on both the public ABIDE and our restricted ABCD datasets. The implementation is available at https://github.com/Wayfear/BrainNetworkTransformer.
updated: Thu Oct 13 2022 02:30:06 GMT+0000 (UTC)
published: Thu Oct 13 2022 02:30:06 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アソシエイト