fedobp

FedOBP: Federated Optimal Brain Personalization with Few Personalized Parameters

https://github.com/uglyghost/fedobp

Science Score: 44.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
    Found CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
    Found .zenodo.json file
  • DOI references
  • Academic publication links
  • Academic email domains
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (6.7%) to scientific vocabulary

Keywords

federated-learning model-pruning optimal-brain-damage
Last synced: 6 months ago · JSON representation ·

Repository

FedOBP: Federated Optimal Brain Personalization with Few Personalized Parameters

Basic Info
  • Host: GitHub
  • Owner: uglyghost
  • License: gpl-3.0
  • Language: Python
  • Default Branch: master
  • Homepage:
  • Size: 8.61 MB
Statistics
  • Stars: 0
  • Watchers: 2
  • Forks: 1
  • Open Issues: 0
  • Releases: 0
Topics
federated-learning model-pruning optimal-brain-damage
Created about 1 year ago · Last pushed about 1 year ago
Metadata Files
Readme Funding License Citation

README.md

FedOBP

Official repository for FedOBP

This repository is based on the FL-bench implementation.

Installation

sh pip install -r .env/requirements.txt

Step 1. Generate FL Dataset

Partition the MNIST according to Dir(0.1) for 100 clients shell python generate_data.py -d mnist -a 0.1 -cn 100 About methods of generating federated dastaset, go check data/README.md for full details.

Step 2. Run FedOBP Main Experiment

sh python main_fedobp.py [--config-path, --config-name] [dataset.name=<DATASET_NAME> args...]

Step 4. FedOBP Ablation Experiment

sh python run_script_ablation.py

Step 3. Run Baselines Experiment

sh python main.py [--config-path, --config-name] [method=<METHOD_NAME> args...]

Monitor runs

This implementation supports tensorboard. 1. Run tensorboard --logdir=<your_log_dir> on terminal. 2. Go check localhost:6006 on your browser.

Bibtex

bibtex @inproceedings{chen2025fedobp, title={FedOBP: Federated Optimal Brain Personalization with Few Personalized Parameters}, author={Chen, Xingyan and Du, Tian and Diao, Enmao}, year={2025}, url={https://github.com/uglyghost/FedOBP.git} }

Owner

  • Login: uglyghost
  • Kind: user

Citation (CITATION.cff)

cff-version: 1.2.0
title: 'FL-bench: A federated learning benchmark for solving image classification tasks'
message: >-
  If you use this software, please cite it using the
  metadata from this file.
type: software
authors:
  - given-names: Jiahao
    family-names: Tan
    email: karhoutam@qq.com
    affiliation: Shenzhen University
  - given-names: Xinpeng
    family-names: Wang
    affiliation: 'The Chinese University of Hong Kong, Shenzhen'
    email: 223015056@link.cuhk.edu.cn
repository-code: 'https://github.com/KarhouTam/FL-bench'
abstract: >-
  Benchmark of federated learning that aim solving image
  classification tasks.
keywords:
  - federated learning
license: GNU General Public License v3.0

GitHub Events

Total
  • Member event: 3
  • Push event: 2
  • Fork event: 1
  • Create event: 2
Last Year
  • Member event: 3
  • Push event: 2
  • Fork event: 1
  • Create event: 2

Dependencies

.github/workflows/docker-publish.yml actions
  • actions/checkout v4 composite
  • docker/build-push-action v6 composite
  • docker/login-action v3 composite
  • docker/setup-buildx-action v3 composite
.env/Dockerfile docker
  • ubuntu 22.04 build
.env/poetry.lock pypi
  • 132 dependencies
.env/pyproject.toml pypi
  • Pillow ^10.4.0
  • PyYAML ^6.0.2
  • cvxpy ^1.5.1
  • faiss-cpu ^1.8.0
  • flwr-datasets ^0.4.0
  • hydra-core ^1.3.2
  • matplotlib ^3.9.0
  • numpy 1.26.4
  • pandas ^2.2.3
  • pynvml ^12.0.0
  • python >=3.10, <=3.12
  • pytorch-minimize ^0.0.2
  • ray 2.36.1
  • rich 13.7.1
  • scikit-learn ^1.5.2
  • scipy ^1.14.1
  • statsmodels ^0.14.4
  • tensorboard ^2.17.1
  • torch 2.2.0
  • torchvision ^0.17.0
  • visdom ^0.2.4
.env/requirements.txt pypi
  • Pillow *
  • PyYAML *
  • cvxpy *
  • flwr-datasets *
  • hydra-core *
  • matplotlib *
  • numpy <2.0
  • pandas *
  • pynvml *
  • pytorch-minimize *
  • ray *
  • rich *
  • scikit-learn *
  • scipy *
  • statsmodels *
  • tensorboard *
  • torch *
  • torchvision *
  • visdom ==0.2.4