alessiochisari / federated-transfer-learning

Federated Transfer Learning using network composition for Sentiment Analysis; it is my thesis of Master's degree in Management Engineering at University of Catania about Deep Learning applied to Market Research.
MIT License
4 stars 0 forks source link

Federated-Transfer-Learning-using-network-composition-for-Sentiment-Analysis

Federated Transfer Learning using network composition for Sentiment Analysis; it is my thesis of Master's degree in Management Engineering at University of Catania about Deep Learning applied to Market Research.

Description

In this thesis, is proposed a new technique of Federated Trans- fer Learning (FTL), a special case of FL, in which two datasets differ in the feature space. This can be applied to different datasets but similar in their nature, e.g. data collected from companies in the same sector, but with differences in the nature of business. These enterprises, also physically far in the world, share only a small over- lap in feature space, and their datasets differ both in samples and in feature space. Our proposed method aims to address the limitations of existing federated learning approaches, and leveraging transfer learning to provide solutions for the entire sample and feature space under a federation. The general idea of our proposed method is to combine the weights of two different neural networks trained on two different machines and datasets, but similar in their nature, and check how the performances change after the aggregation. The experiments and the drafting of the thesis has been carried out together with my colleague of Data Science for Management Bruno Casella, under the supervision of Prof. Sebastiano Battiato, Prof. Alessandro Ancarani and Prof. Valerio Giuffrida, a Lecturer in Data Science at Edinburgh Napier University. In particular, it was expected a period of stay in Scotland at the Edinburgh Napier University, however it was not possible due to the CoViD-19 limitations.

Dataset

We have used MNIST and SVHN

Usage

Upload the dataset provided on your Google Drive.

Load the .ipynb notebook on Google Colaboratory, and you will be able to run the cells.

License

MIT

Contacts

Alessio B. Chisari - alessiochisari@hotmail.com

Bruno Casella - casella0798@gmail.com