Privacy-aware deep vein thrombosis segmentation using a multi-model federated learning framework with the federated averaging algorithm
摘要
Deep Vein Thrombosis (DVT) is the formation of blood clots in the deep veins of the calf, requiring precise Computer Tomography (CT) scan segmentation for accurate diagnosis and treatment. We proposed and developed an efficient Federated Learning (FedL) architecture using the Federated Averaging (FedAvg) algorithm. Seven distinct local models were designed and trained on non-independent and identically distributed (Non-IID) CT images to maintain data privacy and security, enhancing DVT segmentation efficiency and accuracy. The global model was progressively improved by aggregating the local model’s weights using FedAvg algorithm. Our algorithm was evaluated in three phases using datasets of 1000, 2000, and 3000 samples to assess the global model’s performance. Phase 1 involved three clients, each with unique local models (Convolutional Neural Network (CNN), Sequential, and Semantic). While, Phase 2 expanded to five clients, incorporating additional models (U-Net and VGG Net-19). In Phase 3, scaled to seven clients with advanced models (Modified U-Net and Modified-Net). Empirical results across Phases 1–3 showed significant gains with increasing dataset size –attaining higher Accuracy (