Edge.AI - Distributed Learning Subsystem

This repository is designed mainly for research, and pytorch implementation of various Federated Learning (FL) area.

Target FL Task

• Image Classification
  • Global Federated Learning (GFL)
  • Personalized Federated Learning (PFL)
• Re-Identification
• NLP (Multi-Lingual Problem)

Contact

Feel free to contact us if you have any questions:)

• Seongyoon Kim: curisam@kaist.ac.kr
• Woojin Chung: gartland@kaist.ac.kr
• Yongjin Yang: dyyjkd@kaist.ac.kr
• Sungwoo Cho: peter8256@kaist.ac.kr

Acknowledgements

This work was supported by Institute of Information & communications Technology Planning & Evaluation (IITP) grant funded by the Korea government(MSIT) (No. 2021-0-00907, Development of Adaptive and Lightweight Edge-Collaborative Analysis Technology for Enabling Proactively Immediate Response and Rapid Learning).

The GitHub repositories we referred to for code implementations for each target task are as follows:

• Image Classification

  • Global Federated Learning (GFL)
    • https://github.com/FedML-AI/FedML (For overall code structure)
    • https://github.com/Lee-Gihun/FedNTD
  • Personalized Federated Learning (PFL)
    • https://github.com/pliang279/LG-FedAvg
    • https://github.com/jhoon-oh/FedBABU

• Re-Identification

  • https://github.com/cap-ntu/FedReID

• NLP (Multi-Lingual Problem)

  • https://github.com/adap/flower