From a70e1abef470703e8886258108e59f3e20938832 Mon Sep 17 00:00:00 2001 From: JQ Liu Date: Thu, 5 Dec 2024 22:35:19 +0800 Subject: [PATCH] 20241205 update --- README.md | 32 ++++++++++++++++++++++++++++++-- SlaugFL_Overview.png | Bin 0 -> 539061 bytes 2 files changed, 30 insertions(+), 2 deletions(-) create mode 100644 SlaugFL_Overview.png diff --git a/README.md b/README.md index f437b3e..28c39a6 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,30 @@ -# SlaugFL -Code for SlaugFL +# SlaugFL: Efficient Edge Federated Learning With Selective GAN-Based Data Augmentation + +This is the pytorch implementation code of the paper "[SlaugFL: Efficient Edge Federated Learning with Selective GAN-based Data Augmentation](https://ieeexplore.ieee.org/abstract/document/10521828)" + +## Overview +The following figure shows an overview of our SlaugFL. The SlaugFL consists of two parts. The left part, in the dotted box, is the preparation phase which aims to obtain a trained ACGAN model. The right part shows the process of our FL collaborative training. In the collaborative training phase, we propose a dual-calibration method on the device side to improve FL performance: during the local training phase of selected devices, the local model is calibrated on the augmented data and further corrected through aligning its local class prototypes with the received p-f global class prototypes. + +![SlaugFL](./SlaugFL_Overview.png) + +We divide the implementation code into two parts: The Preparation Phase and The Collaborative Training Phase. + +## The Preparation Phase + + +## The Collaborative Training Phase + + +## Citation + +If you find this code is useful to your research, please consider to cite our paper. + +``` +@article{liu2024slaugfl, + title={SlaugFL: Efficient Edge Federated Learning with Selective GAN-based Data Augmentation}, + author={Liu, Jianqi and Zhao, Zhiwei and Luo, Xiangyang and Li, Pan and Min, Geyong and Li, Huiyong}, + journal={IEEE Transactions on Mobile Computing}, + year={2024}, + publisher={IEEE} +} +``` \ No newline at end of file diff --git a/SlaugFL_Overview.png b/SlaugFL_Overview.png new file mode 100644 index 0000000000000000000000000000000000000000..8995a156ac739d4a1cd69f6134a39890a4bc4986 GIT binary patch literal 539061 zcmce;byVER(mxD?1t+*Wf#4F{hTy^7g1fuxK!7A@AV`ql9^BpC-QC^Y-pTIX=jQ%) z@45Nw<(xV6nfZ2ib=jw?x@JDf%Zei-;3GglKp;y>h$uopyg-M5fFXSW3x4N1cnbpp z0vg3!SXkatSX|iF%GN>2PT$a2+}Ot0!Q4<${4E3oS6G;ewi%%cra)qCIUVItz(jU- z6v`9(*9t4e(GPPwPsZie3-r-e6+)4{2aCz(uh`ZfGR!0*?xmDfktST^=A+-@Zu`)n^c9xWk8*3fIkE>QvK@rxKX z$Jffcw>mm_oLap{kiAD3=)z-YV_rcLthY7K9UhZ*ElpolUq74f7s-o5eq0Qk`N7pk zzGe*Zgw{1 zDXui3;T%uq!}-{A3tyA>i}rF9B*;>XHwN7MR*5cJcFYIJMU*ppV=hD0T*c8k#I6RT zrKoHyIDIT;G%xQ&dp52$hd-r&(5`O?bU-*2=rC0AY#`im%FE{*ZK2DV_-2Z>9N^~M z%GcL*B=S(nqML-^?4y{3w1r9xv9d}QITk^Ht} zGZl&kBp-LP0_PEU6FZ6t7D*<_sDK+n^5_j_JhBYZDC%A%qfnz!o_8qh-Ld43M(4Go z6Yh`{iYc5S4Iw+OLS0X$M)4~Z!=3L^IMHT?XNE2NVy10$c)V@YLh8j-Q|hN2cbzCr z@hupfdgdmQ9fqIVbhoEVbhkM*k)GgupnUh|Rjb+e-wdcCc-Pr#b%>wy%Nhw}F{oJS zmuIAFu&dqyNcthXYu)N!n8Ip3Y+UyWMp^b=N0GepU0>P|SRHaqaY(UW_UqAJVRX=S z(6uLVOcA%3a6f)yc~qQHim+uKr?8OnR&$1dkH#pFzpcH^NJ?%8E9e8fK9h$FX8}eS zj&?n4cZsc0f%gwToIJ+^n;+#peakUS_-a zNb%1VgSs3~t@vOJymYJ`&oXvQf#rIj=DpUe;znKh0X1@O z*3aQ*y54vU@k1cn?Ml&&y2D@@z%?1r-l`^tKU_evf2rkWL$w4kta*CNeS*wBYdz2xv%r2pI4P68sW?B>2Cv7$gk@)F1By zARvOxA)xfaj{{g9he?9R6Jr(L-V;EBK6A)HU5kFlOyq$!GBLoCC z)z25Cq$1@B1OyO5Qbb7E4f0?S&P!`@7WPDZ{)`x}s{ESt0PpP!N~;&1$CHX;ud^5` zfH&{b0#y3Ub|Exh@S?qKdlkyeD*5&*oGr=CR?E}txHcTCH`Yj_6rM(yunNbjv+EflI|`8 zp#K*i|3$z7mL6%V4x+5WZ}9%-oG2YX5VYhoHtGM6)4$f2*DHpAg1^$V`~1IU`^TKf z3qY4Tbd}Yt|8euPKy1GCo^(7jWP1>@WjWu1zi0@yK6t#UT3LVaa6T++t;cWUruD>^+IKXtU2FvvJ>7GZnyoUtizJZcYm9 z7B8#)jDGNHq%g`S^JsRsn*V%FFGKYPVs*8!k>vd~jX%=3KioqYFYM315#86wr)m zsXLsnHe70O2i=BC6S^+g<8nI~Nhi{4ltJ%L6^7*5?7vgWkrkDdmHoM);M0wvva<3A zZ5->gQL~O1H66UdcRh8x<2mie*TM85#;*hgNpg~1H0f0oG&DX%2;9+R*A>!*5pbf2 z6WWXXTw-!RT_?q@ua`O{6cP%yqhhyu*(W4DJw1M}s>o?KD~_CsD(iBsy=QW-7o7dN9s zbYqm!$9`LC^jsfc8xGB)wXm>YHXTa(Nmix~&gET)B8Sbw$3R4!F2zcjWXb4P`Xi}A z*0I4_%)U=vR!a@s3;8Z_+D@aQ5>ir@)!*}(J+G`Lrlz`-IcuUOEjp><-lI3rtX0e( zb`jV%ytMbOz}h>P_AcsFoN(2;B;x$;aN3DO!OqT}En_;E5O{ZY7rzIqrg*Q`{V9tp zS+?PFN~=`aTFG>!1;n+R89hgjTCF!-q!XtWIjvlQ@d;BcdeO#|$@{ttDkdgW(-dzP z=|r%eu0CEbNBop~hFDlzX*cn_hy9HEC3l8^?b>VVBs)M53K9Nz_MW4Xw@;T6W6cV{ z$jGRpBJa)>_~;FQs5igX z;Hg*>2zRoosDF+{7@NhMw$BV`HpV#=u$`}B055;-=k&4q=V$08=-2k&y{~3t7c1_) z=B=wcy$Rvpn*UrdGlQ&XU4mMrpV7c$YX9;l6?_JB&H0o#bL%`o9;gf6*NZG9+o#r717V_H`3kTcPq?yYEInJY`YQnkxiw0t#b#>i8U6LVqf}v zy5`#R#sCl10e@JFIm|M_v3Jfql`deq&L4i~6;YY%%ee*~ zUDHmbKM{p)bL}lJs)yWN^L;nWwVt^Cs2<`3=<-e?;&tipz-@L)>T)K8e;;y-i%v=~ z&GL1Ydpil^+@3O$@@W0xG{QIZmg{=S({`cz%SvH<1Vt;4J&Cw~vH_T37Tw^9H(}t3 z(%9216P_zuiMHXlT+a{Ng6jf9QcR7Q=k`VQM|D3Ly~{#kda*P5lP^Yw0zrc&A^W&{ zmE4yTN`?aA&@Vf@li-NmNVnEY$o~N{dnwS^DTM6pS-yK7)z%!RMuw2k|DV_XUuZxX z37h}|Vxn$G#YtpRw@#FQ8)0W~K)E33LRb1c68q=JYp}vP`#jv7Don!_H*Cgg84yo% zxbdWD7R}Nv~t05n7?N2Go3*pu_Dt{>E}tYVZmp<0~gPptI`1@W8+Rm znSzWi!ieiiJPs@Mfo_HM!7k4v_3m~RYR7hD&QuED;a232&V27H2Pq=>{K2TPGH95f z>g!1Ky13yCf$~;boMUwF>TsTO%UUcH_vN)E#y;2Bo)IMjJ0*)8tR%|E^yA!~) zpBAAj!Wyx|#k_b^Xl!s;cl(9ON}kpf_CY5O*4ucgL* zeXao8<#DFeu$>XL+HS7GJhI~HCp1e3S(#VMO2OdmA=^~f8*^-Jhc^rxxuw+TCMslz zb$ZF>BsshT>;9^}fxu8<^%8>f54U$hLP7(H{HGllxp{e?z#I|{=^SI7Ju1IL#yeg% z757;u_Bv?=dz+GW6w7L&AO8^10ngxfCTLK;BsA)^0-$wI+)rBJPDg}MMe~O?e_yj} zfCtrMly1WAcfUQSf~f`Qe-)-4@Vhb28*o@Bkr0=Fob{Vw)c+ql=vg%*SpE}GV8BfM z&8eKSKEWxB1uzQ;r|xn0?!ackSTGw>`1@+-0f~m>i7SymyvvxDpz(@R{Pr$ z6#1DbU=kLk|E4@B6Twk!)FGCs=kJoF0F5hw1`}NHyU9iGE3i%(d3Mg~{EaXFOd}8z zoLq>suG0SQ^JVbhTwzwnzAEnTlKi{Kz?-6pf@6OLyPDq)<*k0Id6iny;qR)s$V+B1 zjo`5{#fH-^0uU|6E9n-NZbFRsZRmdhuzwaVzZ9tjsgQ!g%Oh}D0t_MXIuU|JhknEL z4_f$-eB5yd5YDA;H)dPC{LMW{Z9utr@M}wSoSA=LF?AkM`a2(1 zKb@tQ4lL~r@55P#|5thv4j`RI5B;%BNy(y?Ev3H$^N6OOFhK?OD&Oyt`w0eDBC>-? zAW`-orTlltYQ_L2Zi+l9=kLe+J2~L#l%+q7GCP_2I?)z`5dSnNni_(VoKUf5^VOlx8(|2{(cfDUZy{!z^ign*i zn__1=Z>8IR@bRs-MU{*EFYDivr;hLHP3-G2@92%~SDI(o53_36aO~)__qL}8>+F?v zP=k4VvrvQe>w$PwT6Ryc~A6 z@sW(fc-R?NR#ARXu{&T({CLz}`70}T&><6z@NzAjGRj)B_&IyM_(UH`4RVF}92SBA znDPM2SjHBDVL|+snv*+T`wsYMsPm-1mlY`U{pw7zi(2!<@!o?>a`(-@sZuzM%k)Tb z_P0ZaC@>hqV}mh)*(@Vhkm5@~60l)&+-vJ&!*UT=xVX5%Z@00g_ z4?CU++H8;|4G?q%V^Kg`;xX$F?36UsMgJ!gBK-nGzUY@xlruTUKWAtesBLKw3ypoz zTaqf5tpPv^B}5lePKQO`*xFhflVXzDxP}`}{#90R(k-i3X%$1C7SH6r!96@dxzO@4 z|0;`L{C3p4R0__O2}wPWhZU`@KmJv!Tr3`p#_Ku9GL!%`_DIsaB}ps#Unlbjwd|i8 zx)x&a77*n|t$@NG^maeAJL3EY7k;s>7J(Nvn|Dh^_&Hb1@8pG=ps|p8D$`wq@OTyl z4!Bx5j7f$38}<0K|LrV5N3Zqb4xmIgU95`BTB+J`q2naXC;a#A5`I19ulw!j0Eppf zX<}0WmlEs*%oTF9u$$Q}W;?epVUQ?ybdE*8BmesdK!5n$uclgk)@O!oFYR{#CR7q7 zI59a=|Lj3PssN5SoPrR7lC;d10q2S=nGsm3(RG2jrr5CP7X#BVg}>nJjsa-z6(>h5 zd)kko$xiXLqyOeKKu6>&u~wOheTgNNZ9z}9*r5Y*<7HS*ap(I#&cXh`RMJ84mjgHW+xJu(H`O)Rb}N`Sbpe&L~dr;M)k|CzYZk;ke{eW00^H8l5r>KU@{QK^f)M= zBH_TG=+DWNDDiJX(3<_k6=U)s{L|m@PDB8(|1H;lKl7)F7s4n9`4?ij9DGIkxgbkL z92QSZ?;9aRD0uXXz;*Lq0K%!!?{2KwM2#YeSa+HvRQ&I@`)~LDBqh`3`2OY>O;AIaSrmAv$spdmEGptdx|aEWN&FWJE^t93 z7f_cds7=et0YZ(1T)E6du--d&M@t9b3^0+yx(!?gAu($=j?)ebVXUKJ9{!II@Zt?< z4WC-|=*Kj9u&3ty4Do+sirzDPE?jJ1wf44{ITaTnv(gCyr6g({*9gAUCsNY6B ziy=7p8d>=Td4Y21ClY<7Sm?l10KEDy#{K^connE};~}YqWb z2J?a(CS(fH2{%OZ`;x0#;gm+!;%GR;KR9)e=%-8mfOIyZ0pWd8uCb-G@ukl8etQaB zp>0E)nB%)i%SDl4v~8;nME*}$;9YyChn?@wh#EAShFZhno*W5}3Fyv66Piu!E|&tMU8 zA6({zr!6TQ15I`!pXA$~{gLrUN$a(lZ?D}b4b9H)iPTUT9M3xZJ;sZ)1H>TPcr`ZR ziz5%cEe<{6oq&XbuUjkJj6MJq8MKrNqa$NVwE**=k#b)$U&TWAsFq$EasRB1r{*Bu zERNJ}lxy<)hv=dc~A@<8;2_j9V_ zQ7o@|;<1@w3or-5k@}}*)~NzJJp$`}iq0kdKweD+;_U|T2;-7^LXb|cg&_*~1_H~- z1b6GNI#-uokFW#wLhjZWoO7$}ar25^Zt472-(&`$70U1Xs>w7>gGv++@6EYy19nLj zYD{}PDMkb{@Y>@Z@T4=)HwhJUbNlmyIvcrzEARf$=ocg+XhWj@9Zrn!*h&$1$Hu&$ z&?~~B6~(2x{6S@+HD({|c-L4FAILLKr*}7QU z(Vt}|sFM?CgM#O|%RZR^gwa9H)9y5c5iKTM{&AF4YWSN2;P~gQU;$(a-WL*rNX%Oo znW|+Bm^%u9$uJxGC{jRNHMgg7gasji@ZYwSPy}TM@q9g62p?Os&ZS?331NqE_eK~k zuoc8a!KXdX(>KQUI)$Wzbgze&K{rU1o7TTND<(<7Um=yl@}q z9R$dTWQNU3=tp81@gJ5)3G(Zc&Il6B)g{M62!i_7eTa8Q3O_gWN|96ux^G`wT~Qe> zqL}+5(3?n&BC+HjLa9hMK<7Ni>Uymbe)J$QtVirc!-)F-LhSZOm%&fI$B~lc-H_373jIHZl{Y~_=JMw z1m-5f1(o-{u^{NfZv_{V&QVQsrIocN*S)Vsm%cgj|8Z)<7m!$1*+C5o5no$@Q}F2W zBgeob_y{o6LBoHYWQTj{HrZmhcb~+^@{#n#$LN1~=>I!C2>RvC>U^!+b+eq|<7cXq zg5}#o%J