notesum.ai

Published at November 27

Federated Learning with Uncertainty and Personalization via Efficient Second-order Optimization

cs.LG

cs.CV

stat.ML

Released Date: November 27, 2024

Authors: Shivam Pal¹, Aishwarya Gupta¹, Saqib Sarwar¹, Piyush Rai¹

Aff.: ¹Department of Computer Science and Engineering, IIT Kanpur, India

Arxiv: http://arxiv.org/abs/2411.18385v1

Refer to caption

Models	EMNIST				CIFAR-10				SVHN
Models	ACC( $\uparrow$ )	ECE( $\downarrow$ )	NLL( $\downarrow$ )	BS( $\downarrow$ )	ACC( $\uparrow$ )	ECE( $\downarrow$ )	NLL( $\downarrow$ )	BS( $\downarrow$ )	ACC( $\uparrow$ )	ECE( $\downarrow$ )	NLL( $\downarrow$ )	BS( $\downarrow$ )
FedAvg	91.66	0.0405	0.3355	0.1303	62.25	0.0981	1.199	0.5191	82.14	0.0311	0.6857	0.2640
FedLaplace	91.33	0.0381	0.3255	0.1314	61.80	0.1072	1.233	0.5284	81.99	0.0211	0.6423	0.2627
FedIvon@mean	93.14	0.0349	0.2821	0.1075	62.92	0.0983	1.1500	0.5114	84.54	0.0241	0.5624	0.2256
FedIvon	93.09	0.0188	0.2341	0.1019	62.54	0.0312	1.0790	0.5021	84.76	0.0148	0.5303	0.2210