arXiv reaDer
眼底糖尿病性網膜症の病期分類のためのフェデレーテッド不確実性認識集約
Federated Uncertainty-Aware Aggregation for Fundus Diabetic Retinopathy Staging
深層学習モデルは、糖尿病性網膜症 (DR) のステージングの分野で有望なパフォーマンスを示しています。ただし、複数の機関間で DR ステージング モデルを共同でトレーニングすることは、iid 以外のデータ、クライアントの信頼性、および予測の信頼性評価のため、依然として課題です。これらの問題に対処するために、各クライアントの信頼性を考慮し、DR ステージングの信頼性の推定を生成する、新しい連合不確実性認識集約パラダイム (FedUAA) を提案します。私たちの FedUAA では、眼底画像のグローバルな表現を学習するために集約されたエンコーダーがすべてのクライアントによって共有されますが、ローカルのパーソナライズされたステージング基準のために、各クライアントに対して新しい温度加温不確実性ヘッド (TWEU) が利用されます。当社の TWEU は、クライアントの信頼性評価のための DR ステージング結果を含む不確実性スコアを生成するために、証拠となるディープ レイヤーを採用しています。さらに、各クライアントの不確実性スコア分布に基づいてモデル集約の重みを動的に調整するための新しい不確実性認識重み付けモジュール (UAW) を開発しました。私たちの実験では、さまざまな機関から公開されている 5 つのデータセットを収集して、フェデレーテッド DR ステージング用のデータセットを実行し、実際の非 iid 条件を満たすようにしました。実験結果は、FedUAA が他のフェデレーテッド ラーニング方法と比較して、より高い信頼性でより優れた DR ステージング パフォーマンスを達成することを示しています。私たちが提案する FedUAA パラダイムは、複数の機関にわたって DR ステージング モデルを共同でトレーニングするという課題に効果的に対処し、実際の臨床シナリオで DR 診断モデルを展開するための堅牢で信頼性の高いソリューションを提供します。
Deep learning models have shown promising performance in the field of diabetic retinopathy (DR) staging. However, collaboratively training a DR staging model across multiple institutions remains a challenge due to non-iid data, client reliability, and confidence evaluation of the prediction. To address these issues, we propose a novel federated uncertainty-aware aggregation paradigm (FedUAA), which considers the reliability of each client and produces a confidence estimation for the DR staging. In our FedUAA, an aggregated encoder is shared by all clients for learning a global representation of fundus images, while a novel temperature-warmed uncertainty head (TWEU) is utilized for each client for local personalized staging criteria. Our TWEU employs an evidential deep layer to produce the uncertainty score with the DR staging results for client reliability evaluation. Furthermore, we developed a novel uncertainty-aware weighting module (UAW) to dynamically adjust the weights of model aggregation based on the uncertainty score distribution of each client. In our experiments, we collect five publicly available datasets from different institutions to conduct a dataset for federated DR staging to satisfy the real non-iid condition. The experimental results demonstrate that our FedUAA achieves better DR staging performance with higher reliability compared to other federated learning methods. Our proposed FedUAA paradigm effectively addresses the challenges of collaboratively training DR staging models across multiple institutions, and provides a robust and reliable solution for the deployment of DR diagnosis models in real-world clinical scenarios.
updated: Thu Mar 23 2023 04:41:44 GMT+0000 (UTC)
published: Thu Mar 23 2023 04:41:44 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アソシエイト