Skip to content

Implemented the FedAvg Algorithm in Federated Learning. Completed its uncertainty estimation and confidence calibration.

Notifications You must be signed in to change notification settings

Bozhi-Lyu/FedAvg

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Uncertainty Estimation and Calibration Measurement in Federated Learning

We simulated different data distribution scenarios and then implemented FedAvg algorithm from scratch. We used entropy and VR as indicators of uncertainty, and ECE score to measure calibration. Then we explored the affect of different model parameters in FL architecture on accuracy, uncertainty and calibration quality.

About

Implemented the FedAvg Algorithm in Federated Learning. Completed its uncertainty estimation and confidence calibration.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages