This repository contains list of publications on the application of Graph Neural Network in Wireless Communication

DroneSecurity (NDSS 2023)

Conditional diffusion model to generate MNIST. Minimal script. Based on 'Classifier-Free Diffusion Guidance'.

Federated Learning Benchmark - Federated Learning on Non-IID Data Silos: An Experimental Study (ICDE 2022)

The repo for studying and sharing diffusion models.

Repository for the code user in the paper...

Scripts for Imagenet 32 dataset

Code and data accompanying the FedGen paper

PyTorch implementations of Generative Adversarial Networks.

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Flower: A Friendly Federated AI Framework

Generative models (GAN, VAE, Diffusion Models, Autoregressive Models) implemented with Pytorch, Pytorch_lightning and hydra.

Perception toolkit for sim2real training and validation in Unity

A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561

Vision Transformer (ViT) in PyTorch

[ICLR 2021] HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clients

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

🤖 PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

This is the code repository for the following paper: "Model pruning enables efficient federated learning on edge devices".

This is a code repository for the ACL 2022 paper "ODE Transformer: An Ordinary Differential Equation-Inspired Model for Sequence Generation", which redesigns the Transformer architecture from the O…

Self-supervised federated learning for medical imaging - IEEE TMI

Leaf: A Benchmark for Federated Settings

Deep Networks with Stochastic Depth

Refining continuous-in-depth neural networks