arXiv reaDer
医用画像におけるランドマーク位置特定のための非同期分散型フェデレーテッド生涯学習
Asynchronous Decentralized Federated Lifelong Learning for Landmark Localization in Medical Imaging
フェデレーテッド ラーニングは機械学習分野で最近開発されたもので、データを 1 つの場所またはデバイスと共有することなく、デバイスのシステムが 1 つまたは複数のタスクでトレーニングできるようにします。ただし、このフレームワークでは、個々のモデルを 1 つに統合するための集中化されたグローバル モデルが依然として必要であり、デバイスは同期的にトレーニングされます。これらは両方とも、連合学習を使用する際の潜在的なボトルネックになる可能性があります。この論文では、連合学習のメリットを継承し、中央ノードや同期トレーニングを必要とせずに複数のタスクで同時にトレーニングできる非同期分散型連合生涯学習(ADFLL)メソッドの新しい方法を提案します。したがって、従来の連合学習の潜在的な欠点を克服します。複数の画像シーケンスと画像の向きで左心室をローカライズするための脳腫瘍セグメンテーション (BRATS) データセットで優れたパフォーマンスを示します。私たちのフレームワークにより、エージェントは平均距離誤差 7.81 で最高のパフォーマンスを達成できます。これは、従来の全知エージェントの平均距離誤差 11.78 よりも優れており、距離誤差が従来の生涯学習エージェントよりも有意に (p=0.01) 優れています。 8ラウンドのトレーニングの後、15.17。さらに、すべての ADFLL エージェントは、従来の LL エージェントと同等またはそれ以上の性能を備えています。結論として、従来の RL エージェントと比較して優れたパフォーマンスと高速化を備えた ADFLL フレームワークを開発しました。
Federated learning is a recent development in the machine learning area that allows a system of devices to train on one or more tasks without sharing their data to a single location or device. However, this framework still requires a centralized global model to consolidate individual models into one, and the devices train synchronously, which both can be potential bottlenecks for using federated learning. In this paper, we propose a novel method of asynchronous decentralized federated lifelong learning (ADFLL) method that inherits the merits of federated learning and can train on multiple tasks simultaneously without the need for a central node or synchronous training. Thus, overcoming the potential drawbacks of conventional federated learning. We demonstrate excellent performance on the brain tumor segmentation (BRATS) dataset for localizing the left ventricle on multiple image sequences and image orientation. Our framework allows agents to achieve the best performance with a mean distance error of 7.81, better than the conventional all-knowing agent's mean distance error of 11.78, and significantly (p=0.01) better than a conventional lifelong learning agent with a distance error of 15.17 after eight rounds of training. In addition, all ADFLL agents have comparable or better performance than a conventional LL agent. In conclusion, we developed an ADFLL framework with excellent performance and speed-up compared to conventional RL agents.
updated: Sun Mar 12 2023 23:51:51 GMT+0000 (UTC)
published: Sun Mar 12 2023 23:51:51 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アソシエイト