yjwong1999/AL-based-FL-for-Multi-Task-Disaster-Detection-Model

We design a multi-task model for joint disaster classification and victim detection. We train the model using both the Centralized Learning (CL) and Federated Learning (FL) methods. We also tried Active Learning (AL) to see how it could help in reducing the labeling workload for disaster dataset. Lastly, we optimized the model using OpenVINO.

Jupyter Notebook