Multi-limb Split Learning for Tumor Classification on Vertically Distributed Data

Abstract

Brain tumors are one of the life-threatening forms of cancer. Previous studies have classified brain tumors using deep neural networks. In this paper, we perform the later task using a collaborative deep learning technique, more specifically split learning. Split learning allows collaborative learning via neural networks splitting into two (or more) parts, a client-side network and a server-side network. The client-side is trained to a certain layer called the cut layer. Then, the rest of the training is resumed on the server-side network. Vertical distribution, a method for distributing data among organizations, was implemented where several hospitals hold different attributes of information for the same set of patients. To the best of our knowledge this paper will be the first paper to implement both split learning and vertical distribution for brain tumor classification. Using both techniques, we were able to achieve train and test accuracy greater than 90\% and 70\%, respectively.

References in corpus (9)

• Deep Learning in Neural Networks: An Overview
• An overview of gradient descent optimization algorithms
• Activation Functions: Comparison of trends in Practice and Research for Deep Learning
• A Deep Learning Approach for Brain Tumor Classification and Segmentation Using a Multiscale Convolutional Neural Network
• Private federated learning on vertically partitioned data via entity resolution and additively homomorphic encryption
• Predictive modeling of brain tumor: A Deep learning approach
• Distributed Training of Deep Learning Models: A Taxonomic Perspective
• Split Learning for collaborative deep learning in healthcare
• SplitFed: When Federated Learning Meets Split Learning