Search papers, labs, and topics across Lattice.
FedBNN is introduced, a federated learning framework that trains binary neural networks directly on edge devices, addressing the computational limitations of deploying full-precision DNNs in FL settings. By learning binary representations during local training, FedBNN achieves substantial reductions in model size and FLOPs compared to federated methods using real-valued models. Experiments on benchmark datasets show that FedBNN maintains comparable accuracy to existing federated methods while significantly reducing resource consumption.
Binary neural networks can now be trained effectively in federated settings, offering a path to low-cost, privacy-preserving edge inference without sacrificing accuracy.
Federated Learning (FL) preserves privacy by distributing training across devices. However, using DNNs is computationally intensive at the low-powered edge during inference. Edge deployment demands models that simultaneously optimize memory footprint and computational efficiency, a dilemma where conventional DNNs fail by exceeding resource limits. Traditional post-training binarization reduces model size but suffers from severe accuracy loss due to quantization errors. To address these challenges, we propose FedBNN, a rotation-aware binary neural network framework that learns binary representations directly during local training. By encoding each weight as a single bit $\{+1, -1\}$ instead of a $32$-bit float, FedBNN shrinks the model footprint, significantly reducing runtime (during inference) FLOPs and memory requirements in comparison to federated methods using real models. Evaluations across multiple benchmark datasets demonstrate that FedBNN significantly reduces resource consumption while performing similarly to existing federated methods using real-valued models.
