activity
20182026
most citedFederated Learning for Mobile Keyboard Prediction

1.1k citations · 1.6k across the 34 of their papers we have counts for

collaborators

36 papers

cs.CL2026

Evo: Evolving the Evolver through Weight-Harness Co-Evolution

Chejian Xu, Shujian Zhang, Wenxuan Zhou +5

Self-evolving agents are increasingly used to improve language model systems without human intervention through either weight updates or harness improvements. Existing approaches t…

cs.CL2026

Steering LLMs for Culturally Localized Generation

Simran Khanuja, Hongbin Liu, Shujian Zhang +4

LLMs are deployed globally, yet produce responses biased towards cultures with abundant training data. Existing cultural localization approaches such as prompting or post-training…

cs.CL2025

MUSIC: MUlti-Step Instruction Contrast for Multi-Turn Reward Models

Wenzhe Li, Shujian Zhang, Wenxuan Zhou +5

Evaluating the quality of multi-turn conversations is crucial for developing capable Large Language Models (LLMs), yet remains a significant challenge, often requiring costly human…

cs.CL2025

Fantastic Reasoning Behaviors and Where to Find Them: Unsupervised Discovery of the Reasoning Process

Zhenyu Zhang, Shujian Zhang, John Lambert +6

Despite the growing reasoning capabilities of recent large language models (LLMs), their internal mechanisms during the reasoning process remain underexplored. Prior approaches oft…

cs.CL2025

Eliciting Behaviors in Multi-Turn Conversations

Jing Huang, Shujian Zhang, Lun Wang +3

Identifying specific and often complex behaviors from large language models (LLMs) in conversational settings is crucial for their evaluation. Recent work proposes novel techniques…

cs.CR2025

Attacker's Noise Can Manipulate Your Audio-based LLM in the Real World

Vinu Sankar Sadasivan, Soheil Feizi, Rajiv Mathews +1

This paper investigates the real-world vulnerabilities of audio-based large language models (ALLMs), such as Qwen2-Audio. We first demonstrate that an adversary can craft stealthy…

eess.AS2024★ 1 cited

Parameter-Efficient Transfer Learning under Federated Learning for Automatic Speech Recognition

Xuan Kan, Yonghui Xiao, Tien-Ju Yang +2

This work explores the challenge of enhancing Automatic Speech Recognition (ASR) model performance across various user-specific domains while preserving user data privacy. We emplo…

cs.LG2024★ 1 cited

Learning from straggler clients in federated learning

Andrew Hard, Antonious M. Girgis, Ehsan Amid +4

How well do existing federated learning algorithms learn from client devices that return model updates with a significant time delay? Is it even possible to learn effectively from…

cs.CL2023

The Gift of Feedback: Improving ASR Model Quality by Learning from User Corrections through Federated Learning

Lillian Zhou, Yuxin Ding, Mingqing Chen +5

Automatic speech recognition (ASR) models are typically trained on large datasets of transcribed speech. As language evolves and new terms come into use, these models can become ou…

cs.LG2023

Unintended Memorization in Large ASR Models, and How to Mitigate It

Lun Wang, Om Thakkar, Rajiv Mathews

It is well-known that neural networks can unintentionally memorize their training examples, causing privacy concerns. However, auditing memorization in large non-auto-regressive au…

cs.LG2023

Heterogeneous Federated Learning Using Knowledge Codistillation

Jared Lichtarge, Ehsan Amid, Shankar Kumar +3

Federated Averaging, and many federated learning algorithm variants which build upon it, have a limitation: all clients must share the same model architecture. This results in unus…

cs.LG2022

Recycling Scraps: Improving Private Learning by Leveraging Intermediate Checkpoints

Virat Shejwalkar, Arun Ganesh, Rajiv Mathews +5

In this work, we focus on improving the accuracy-variance trade-off for state-of-the-art differentially private machine learning (DP ML) methods. First, we design a general framewo…

cs.CL2022★ 2 cited

Large vocabulary speech recognition for languages of Africa: multilingual modeling and self-supervised learning

Sandy Ritchie, You-Chi Cheng, Mingqing Chen +4

Almost none of the 2,000+ languages spoken in Africa have widely available automatic speech recognition systems, and the required data is also only available for a few languages. W…

eess.AS2022

UserLibri: A Dataset for ASR Personalization Using Only Text

Theresa Breiner, Swaroop Ramaswamy, Ehsan Variani +6

Personalization of speech models on mobile devices (on-device personalization) is an active area of research, but more often than not, mobile devices have more text-only data than…

cs.LG2022★ 5 cited

Mixed Federated Learning: Joint Decentralized and Centralized Learning

Sean Augenstein, Andrew Hard, Lin Ning +4

Federated learning (FL) enables learning from decentralized privacy-sensitive data, with computations on raw data confined to take place at edge clients. This paper introduces mixe…

cs.LG2022★ 4 cited

Online Model Compression for Federated Learning with Large Models

Tien-Ju Yang, Yonghui Xiao, Giovanni Motta +3

This paper addresses the challenges of training large neural network models under federated learning settings: high on-device memory usage and communication cost. The proposed Onli…

cs.CL2022★ 1 cited

Scaling Language Model Size in Cross-Device Federated Learning

Jae Hun Ro, Theresa Breiner, Lara McConnaughey +4

Most studies in cross-device federated learning focus on small models, due to the server-client communication and on-device computation bottlenecks. In this work, we leverage vario…

cs.CL2022★ 1 cited

Detecting Unintended Memorization in Language-Model-Fused ASR

W. Ronny Huang, Steve Chien, Om Thakkar +1

End-to-end (E2E) models are often being accompanied by language models (LMs) via shallow fusion for boosting their overall quality as well as recognition of rare words. At the same…

cs.SD2022

Extracting Targeted Training Data from ASR Models, and How to Mitigate It

Ehsan Amid, Om Thakkar, Arun Narayanan +2

Recent work has designed methods to demonstrate that model updates in ASR training can leak potentially sensitive attributes of the utterances used in computing the updates. In thi…

eess.AS2022

Production federated keyword spotting via distillation, filtering, and joint federated-centralized training

Andrew Hard, Kurt Partridge, Neng Chen +9

We trained a keyword spotting model using federated learning on real user devices and observed significant improvements when the model was deployed for inference on phones. To comp…

cs.CL2022

Capitalization Normalization for Language Modeling with an Accurate and Efficient Hierarchical RNN Model

Hao Zhang, You-Chi Cheng, Shankar Kumar +3

Capitalization normalization (truecasing) is the task of restoring the correct case (uppercase or lowercase) of noisy text. We propose a fast, accurate and compact two-level hierar…

cs.LG2021★ 3 cited

Public Data-Assisted Mirror Descent for Private Model Training

Ehsan Amid, Arun Ganesh, Rajiv Mathews +6

In this paper, we revisit the problem of using in-distribution public data to improve the privacy/utility trade-offs for differentially private (DP) model training. (Here, public d…

cs.LG2021★ 5 cited

Jointly Learning from Decentralized (Federated) and Centralized Data to Mitigate Distribution Shift

Sean Augenstein, Andrew Hard, Kurt Partridge +1

With privacy as a motivation, Federated Learning (FL) is an increasingly used paradigm where learning takes place collectively on edge devices, each with a cache of user-generated…

cs.LG2021★ 9 cited

Revealing and Protecting Labels in Distributed Training

Trung Dang, Om Thakkar, Swaroop Ramaswamy +3

Distributed learning paradigms such as federated learning often involve transmission of model updates, or gradients, over a network, thereby avoiding transmission of private data.…