arXiv reaDer
パーソナライズされたニューラルアーキテクチャを使用したMRIでのフェデレーション全前立腺セグメンテーション
Federated Whole Prostate Segmentation in MRI with Personalized Neural Architectures
堅牢な深層学習ベースのモデルを構築するには、理想的には複数のソースからの多様なトレーニングデータが必要です。ただし、特に医療データが関係している場合は、患者のプライバシーの懸念や規制上のハードルのために、これらのデータセットを簡単に組み合わせることができません。連合学習(FL)は、一元化されたデータセットを必要とせずに機械学習モデルをトレーニングする方法です。各FLクライアントは、すべてのクライアントからのパラメーターを集約するグローバルサーバーとモデルパラメーターを共有するだけで、ローカルデータをトレーニングします。同時に、各クライアントのデータは、患者集団、画像機器、および取得プロトコルの局所的な変動により、差異や不整合を示す可能性があります。したがって、フェデレーション学習モデルは、クライアントのデータのローカルな特殊性に適応できる必要があります。この作業では、「スーパーネット」をトレーニングすることにより、FLをローカルニューラルアーキテクチャ検索に基づくAutoML手法と組み合わせます。さらに、各FLクライアントのサイトでパーソナライズされたモデルアーキテクチャを可能にする適応スキームを提案します。提案された方法は、3D前立腺MRIからの4つの異なるデータセットで評価され、AutoMLスーパーネットを介して最適なパスを選択することにより、適応後のローカルモデルのパフォーマンスを向上させることが示されています。
Building robust deep learning-based models requires diverse training data, ideally from several sources. However, these datasets cannot be combined easily because of patient privacy concerns or regulatory hurdles, especially if medical data is involved. Federated learning (FL) is a way to train machine learning models without the need for centralized datasets. Each FL client trains on their local data while only sharing model parameters with a global server that aggregates the parameters from all clients. At the same time, each client's data can exhibit differences and inconsistencies due to the local variation in the patient population, imaging equipment, and acquisition protocols. Hence, the federated learned models should be able to adapt to the local particularities of a client's data. In this work, we combine FL with an AutoML technique based on local neural architecture search by training a "supernet". Furthermore, we propose an adaptation scheme to allow for personalized model architectures at each FL client's site. The proposed method is evaluated on four different datasets from 3D prostate MRI and shown to improve the local models' performance after adaptation through selecting an optimal path through the AutoML supernet.
updated: Fri Jul 16 2021 20:35:29 GMT+0000 (UTC)
published: Fri Jul 16 2021 20:35:29 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アソシエイト