Conference Papers
Permanent URI for this collectionhttps://idr.nitk.ac.in/handle/123456789/28506
Browse
Search Results
Item FedPruNet: Federated Learning Using Pruning Neural Network(Institute of Electrical and Electronics Engineers Inc., 2022) Gowtham, L.; Annappa, A.; Sachin, D.N.Federated Learning (FL) is a distributed form of training the machine learning and deep learning models on the data spread over heterogeneous edge devices. The global model at the server learns by aggregating local models sent by the edge devices, maintaining data privacy, and lowering communication costs by just communicating model updates. The edge devices on which the model gets trained usually will have limitations towards power resource, storage, computations to train the model. This paper address the computation overhead issue on the edge devices by presenting a new method named FedPruNet, which trains the model in edge devices using the neural network model pruning method. The proposed method successfully reduced the computation overhead on edge devices by pruning the model. Experimental results show that for the fixed number of communication rounds, the model parameters are pruned up to 41.35% and 65% on MNIST and CIFAR-10 datasets, respectively, without compromising the accuracy compared to training FL edge devices without pruning. © 2022 IEEE.Item Simulating Federated Transfer Learning for Lung Segmentation using Modified UNet Model(Elsevier B.V., 2022) Ambesange, S.; Annappa, A.; Koolagudi, S.G.Lung segmentation helps doctors in analyzing and diagnosing lung diseases effectively. Covid -19 pandemic highlighted the need for such artificial intelligence (AI) model to segment Lung X-ray images and diagnose patient covid conditions, in a short time, which was not possible due to huge number of patient influx at hospitals with the limited radiologist to diagnose based on test report in short time. AI models developed to assist doctors to diagnose faster, faces another challenge of data privacy. Such AI Models, for better performance, need huge data collected from multiple hospitals/diagnostic centres across the globe into single place to train the AI models. Federated Learning (FL) framework, using transfer learning approach addresses these concerns as FL framework doesn't need data to be shared to outside hospital ecosystem, as AI model get trained on local system and AI model get trained on distributed data. FL with Transfer learning doesn't need the parallel training of the model at all participants nodes like other FL. Paper simulates Federated Transfer learning for Image segmentation using transfer learning technique with few participating nodes and each nodes having different size dataset. The proposed method also leverages other healthcare data available at local system to train the proposed model to overcome lack of more data. Paper uses pre-trained weights of U-net Segmentation Model trained for MRI image segmentation to lung segmentation model. Paper demonstrates using such similar healthcare data available at local system helps improving the performance of the model. The paper uses Explainable AI approach to explain the result. Using above three techniques, Lung segmentation AI model gets near perfect segmentation accuracy. © 2023 The Authors. Published by Elsevier B.V.