Stars
[AISTATS 2024] BOBA: Byzantine-Robust Federated Learning with Label Skewness. Wenxuan Bao, Jun Wu, Jingrui He
Wrapper to easily generate the chat template for Llama2
A python package providing a benchmark with various specified distribution shift patterns.
[Findings of ACL 2023] Bridge the Gap Between CV and NLP! A Optimization-based Textual Adversarial Attack Framework.
A curated list of awesome papers on dataset distillation and related applications.
[ICML 2023] Optimizing the Collaboration Structure in Cross-Silo Federated Learning. Wenxuan Bao, Haohan Wang, Jun Wu, Jingrui He.
Multi-Class Text Classification for products based on their description with Machine Learning algorithms and Neural Networks (MLP, CNN, Distilbert).
Towards reliable rare category analysis on graphs via individual calibration (KDD'23)
Collection of awesome test-time (domain/batch/instance) adaptation methods
Domain Adaptation for Time Series Under Feature and Label Shifts
Implementations of 'RawlsGCN: Towards Rawlsian Difference Principle on Graph Convolutional Network', WWW'22
TextAttack 🐙 is a Python framework for adversarial attacks, data augmentation, and model training in NLP https://textattack.readthedocs.io/en/master/
[TKDD 2023] AdaTime: A Benchmarking Suite for Domain Adaptation on Time Series Data
Reproducing state-of-the-art results
Must-read Papers on Physics-Informed Neural Networks.
Official Implementation of Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reduction (2020)
A collection of libraries to optimise AI model performances
PyGCL: A PyTorch Library for Graph Contrastive Learning
Official repository for CMU Machine Learning Department's 10732: Robustness and Adaptivity in Shifting Environments
A pytorch implementation of graph transformer for node classification
38 traditional FL (tFL) or personalized FL (pFL) algorithms, 3 scenarios, and 24 datasets. www.pfllib.com/
Official code for "Personalized Federated Learning through Local Memorization" (ICML'22)
Virtual Adversarial Training (VAT) implementation for PyTorch
Code for "MixMatch - A Holistic Approach to Semi-Supervised Learning"
Unofficial PyTorch implementation of "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence"