← Search

Chaoyang He

11 accepted papers

2024

ScaleLLM: A Resource-Frugal LLM Serving Framework by Optimizing End-to-End Efficiency

EMNLP 2024industry

Large language models (LLMs) have surged in popularity and are extensively used in commercial applications, where the efficiency of model serving is crucial for the user experience. Most current research focuses on optimizing individual sub-procedures, e.g. local inference and communication, however…

Cited by 7SourcePDFScholar
2024

TensorOpera Router: A Multi-Model Router for Efficient LLM Inference

EMNLP 2024industry

With the rapid growth of Large Language Models (LLMs) across various domains, numerous new LLMs have emerged, each possessing domain-specific expertise. This proliferation has highlighted the need for quick, high-quality, and cost-effective LLM query response methods. Yet, no single LLM exists to ef…

Cited by 10SourcePDFScholar
2023

A Data-Free Approach to Mitigate Catastrophic Forgetting in Federated Class Incremental Learning for Vision Tasks

NeurIPS 2023poster

Deep learning models often suffer from forgetting previously learned information when trained on new data. This problem is exacerbated in federated learning (FL), where the data is distributed and can change independently for each user. Many solutions are proposed to resolve this catastrophic forget…

Cited by 37SourcePDFScholar
2023

FairFed: Enabling Group Fairness in Federated Learning

AAAI 2023technical

Training ML models which are fair across different demographic groups is of critical importance due to the increased integration of ML in crucial decision-making scenarios such as healthcare and recruitment. Federated learning has been viewed as a promising solution for collaboratively training mach…

Cited by 237SourcePDFScholar
2022

FLamby: Datasets and Benchmarks for Cross-Silo Federated Learning in Realistic Healthcare Settings

NeurIPS 2022accept

Federated Learning (FL) is a novel approach enabling several clients holding sensitive data to collaboratively train machine learning models, without centralizing data. The cross-silo FL setting corresponds to the case of few ($2$--$50$) reliable clients, each holding medium to large datasets, and i…

2022

FedNLP: Benchmarking Federated Learning Methods for Natural Language Processing Tasks

NAACL 2022findings

Increasing concerns and regulations about data privacy and sparsity necessitate the study of privacy-preserving, decentralized learning methods for natural language processing (NLP) tasks. Federated learning (FL) provides promising approaches for a large number of clients (e.g., personal devices or…

2022

SpreadGNN: Decentralized Multi-Task Federated Learning for Graph Neural Networks on Molecular Data

AAAI 2022technical

Graph Neural Networks (GNNs) are the first choice methods for graph machine learning problems thanks to their ability to learn state-of-the-art level representations from graph-structured data. However, centralizing a massive amount of real-world graph data for GNN training is prohibitive due to use…

Cited by 52SourcePDFScholar
2021

MEST: Accurate and Fast Memory-Economic Sparse Training Framework on the Edge

NeurIPS 2021spotlight

Recently, a new trend of exploring sparsity for accelerating neural network training has emerged, embracing the paradigm of training on the edge. This paper proposes a novel Memory-Economic Sparse Training (MEST) framework targeting for accurate and fast execution on edge devices. The proposed MEST…

2021

PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models

ICML 2021spotlight

The size of Transformer models is growing at an unprecedented rate. It has taken less than one year to reach trillion-level parameters since the release of GPT-3 (175B). Training such models requires both substantial engineering efforts and enormous computing resources, which are luxuries most resea…

Cited by 36SourcePDFScholar
2020

Group Knowledge Transfer: Federated Learning of Large CNNs at the Edge

NeurIPS 2020poster

Scaling up the convolutional neural network (CNN) size (e.g., width, depth, etc.) is known to effectively improve model accuracy. However, the large model size impedes training on resource-constrained edge devices. For instance, federated learning (FL) may place undue burden on the compute capabilit…

2020

MiLeNAS: Efficient Neural Architecture Search via Mixed-Level Reformulation

CVPR 2020poster

Many recently proposed methods for Neural Architecture Search (NAS) can be formulated as bilevel optimization. For efficient implementation, its solution requires approximations of second-order methods. In this paper, we demonstrate that gradient errors caused by such approximations lead to suboptim…

Cited by 168PDFcodeScholar