日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

週刊ニュースレター購読
医用画像基盤モデルarXiv:2610.28411

Brain2Vec:構造的脳MRIのための自己教師あり3D基盤モデル

Brain2Vec: A Self-Supervised 3D Foundation Model for Structural Brain MRI

シェア:XThreadsFacebookLINEはてブBluesky

74,425件のT1強調MRIで事前学習した3D基盤モデルを提案し、脳年齢予測の微調整や複数の下流分類で高い性能と収束速度を示した。

詳しい要約

1. どんなもの?

- 構造的脳MRI(T1-weighted MRI)向けの自己教師あり3D基盤モデル。 - 複数研究から集めた74,425件のT1-weighted MRIスキャンで事前学習。 - 用途は2通り。 - 下流タスクのfine-tuningにおける重み初期化backbone。 - そのまま使えるstandalone feature extractor。 - コードと学習済みモデルをオープンソースで公開し、NiChart cloud platformからも利用可能。

2. 先行研究と比べてどこがすごい?

- 従来のend-to-end学習はラベル不足と計算制約が課題。 - 大規模データで事前学習した基盤モデルとして、限られたラベルでの下流学習を支援。 - 脳年齢予測のfine-tuningで、random initializationより最終精度と収束速度が改善。 - 凍結embeddingが5つの下流分類タスク中4つで解剖学的volumetric featuresを上回る。 - より大量データで事前学習されたstate-of-the-art foundation modelとも競合する性能。

3. 技術・手法の肝は?

- 自己教師あり学習による3D基盤モデルの事前学習。 - 学習データは複数研究を統合した74,425件のT1-weighted MRI。 - 事前学習済み重みをfine-tuningの初期化に利用可能。 - 凍結したままfeature extractorとしてembeddingを抽出可能。 - 具体的な自己教師あり目的関数やネットワーク構成は要旨からは不明。

4. どうやって有効だと検証した?

- 3つの観点で評価。 - 下流予測性能。 - 収束速度。 - embedding品質。 - 脳年齢予測のfine-tuningで、random initializationと比較し最終精度と収束速度を検証。 - 5つの下流分類タスクで、凍結embeddingを解剖学的volumetric featuresおよびstate-of-the-art foundation modelと比較。 - 潜在featureの操作が再構築解剖に単調かつ領域特異的な効果を与えることを確認。

5. 議論はある?

- Brain2Vecは空間的に局在し神経解剖学的に意味のある脳領域の情報を主に符号化する。 - 一部のlatent featuresを操作すると、再構築解剖に単調で領域特異的な変化が生じる。 - 限られたラベルデータ下での学習支援とfeature抽出の両面で有用性を示唆。 - 事前学習データの研究間差異やバイアス、他モダリティへの一般化などは要旨からは不明。

6. 次に読むべき論文は?

- 比較対象として言及された、より大量データで事前学習されたstate-of-the-art foundation model(具体名は要旨からは不明)。 - 解剖学的volumetric featuresに関する研究。 - 脳年齢予測(brain age prediction)の関連研究。 - 自己教師あり学習による医用画像基盤モデルの代表的手法(例: masked autoencoder系)。

※ AIが要旨から生成した要約です。正確性は原文をご確認ください。

著者: Spiros Maggioros, Guray Erus, Gareth Harman, George Aidinis, Alexander Getka, Aristeidis Sotiras, Christos Davatzikos

分類: cs.CV, cs.AI, cs.LG

原文アブストラクト

Deep learning models have become essential tools in medical imaging, yet training them end-to-end for diverse tasks remains challenging due to limited labeled data and computational constraints. To address this, we introduce Brain2Vec, a foundation model pre-trained on 74,425 T1-weighted Magnetic Resonance Imaging (MRI) scans pooled from multiple studies. Brain2Vec can serve both as a weight initialization backbone for fine-tuning downstream tasks and as a standalone feature extractor. We evaluated Brain2Vec in three tasks: downstream predictive performance, convergence speed, and embedding quality. Using Brain2Vec's pretrained weights as initialization and fine-tuning on labeled data for a brain age prediction task, results demonstrate improved final accuracy and faster convergence compared to random initialization. Additionally, its frozen embeddings outperform anatomically defined volumetric features on four of five downstream classification tasks and perform competitively with a state-of-the-art foundation model pre-trained on substantially more data. Lastly, we show that Brain2Vec encodes information primarily from a spatially localized and neuroanatomically meaningful set of brain regions, and that manipulating a subset of latent features produces monotonic, region-specific effects on the reconstructed anatomy. Brain2Vec code and trained models are available as an open-source package on https://github.com/CBICA/Brain2Vec. Users can apply Brain2Vec locally by installing the package, or directly through the NiChart cloud platform https://cloud.neuroimagingchart.com, where Brain2Vec features can be derived without requiring local installation or specialized infrastructure.

PR本紙発行元 EmplifAI