← Search

Yiran Chen

59 accepted papers

2026

DPad: Efficient Diffusion Language Models with Suffix Dropout

ICLR 2026poster

Diffusion-based Large Language Models (dLLMs) parallelize text generation by framing decoding as a denoising process, but suffer from high computational overhead since they predict all future suffix tokens at each step while retaining only a small fraction. We propose $\textbf{Diffusion Scratchpad}…

Cited by 0SourcecodeScholar
2026

DecodeShare: Tracing the Shared Pathways of LLM Decode-Time Decisions

ICML 2026spotlight

Large language models (LLMs) handle many tasks with one set of parameters, but under KV-cached inference it is unclear what task-general structure, if any, is used at $\textit{decode time}$ rather than during $\textit{prefill}$. We propose $\textbf{DecodeShare}$, a protocol that identifies a low-dim…

Cited by 0SourceScholar
2026

FlashSVD: Memory-Efficient Inference with Streaming for Low-Rank Models

AAAI 2026technical

Singular Value Decomposition (SVD) has recently gained traction as an effective compression technique for large language models (LLMs), with many studies reporting 20-80% parameter reduction at minimal accuracy cost. However, despite reducing weight memory, existing SVD-based approaches still rely o

Cited by 0SourcePDFScholar
2026

IncVGGT: Incremental VGGT for Memory-Bounded Long-Range 3D Reconstruction

ICLR 2026poster

We present IncVGGT, a training-free incremental variant of VGGT that makes transformer-based 3D reconstruction feasible for long sequences in real-world applications. Vanilla VGGT relies on dense global attention, which causes memory to grow quadratically and requires excessive computation, making i…

Cited by 0SourceScholar
2026

Mitigating Non-IID Drift in Zeroth-Order Federated LLM Fine-Tuning with Transferable Sparsity

ICLR 2026poster

Federated Learning enables collaborative fine-tuning of Large Language Models (LLMs) across decentralized Non-Independent and Identically Distributed (Non-IID) clients, but such models' massive parameter sizes lead to significant memory and communication challenges. This work introduces Meerkat, a s…

Cited by 0SourceScholar
2026

PrivAct: Internalizing Contextual Privacy Preservation via Multi-Agent Preference Training

ICML 2026poster

Large language model (LLM) agents are increasingly deployed in personalized tasks involving sensitive, context-dependent information, where privacy violations may arise in agents' action due to the implicitness of contextual privacy. Existing approaches rely on *external*, inference-time interventio…

Cited by 0SourceScholar
2026

Seeing is Solving: Unlocking Efficient Multimodal RL via View Alignment

ICML 2026poster

Although Reinforcement Learning Fine-Tuning (RLFT) applied to Vision-Language Models (VLMs) substantially enhances multimodal reasoning capabilities, their prohibitive training cost limits broad adoption. Surprisingly, most existing methods simply port Large Language Model (LLM) RLFT techniques to V…

Cited by 0SourceScholar
2026

SparseInfer: Accelerating Large Language Model Inference with Semantics-Inspired Adaptive Sparse Activation

ICML 2026poster

Large Language Models (LLMs) with billions of parameters have transformed AI applications but require immense computational and memory resources during inference. Adaptive sparse activation inference, which activates only a small number of neurons for each token, offers a novel way to accelerate mod…

Cited by 0SourceScholar
2026

Vision-Zero: Scalable VLM Self-Improvement via Strategic Gamified Self-Play

ICLR 2026poster

Although reinforcement learning (RL) can effectively enhance the reasoning capabilities of vision–language models (VLMs), current methods remain heavily dependent on labor-intensive datasets that require extensive manual construction and verification, leading to extremely high training costs and con…

Cited by 0SourcecodeScholar
2025

Angles Don’t Lie: Unlocking Training‑Efficient RL Through the Model’s Own Signals

NeurIPS 2025spotlight

Current Reinforcement Fine-tuning (RFT) paradigms for Large Language Models (LLMs) suffer from sample inefficiency due to the redundant exposure of identical queries under uniform data sampling. While previous work has explored curriculum learning via heuristic difficulty metrics, these strategies e…

Cited by 0SourceScholar
2025

Boosting Adversarial Robustness with CLAT: Criticality Leveraged Adversarial Training

ICML 2025poster

Adversarial training (AT) enhances neural network robustness. Typically, AT updates all trainable parameters, but can lead to overfitting and increased errors on clean data. Research suggests that fine-tuning specific parameters may be more effective; however, methods for identifying these essential…

Cited by 0SourcePDFScholar
2025

CoreMatching: A Co-adaptive Sparse Inference Framework with Token and Neuron Pruning for Comprehensive Acceleration of Vision-Language Models

ICML 2025poster

Vision-Language Models (VLMs) excel across diverse tasks but suffer from high inference costs in time and memory. Token sparsity mitigates inefficiencies in token usage, while neuron sparsity reduces high-dimensional computations, both offering promising solutions to enhance efficiency. Recently, th…

2025

Group Distributionally Robust Dataset Distillation with Risk Minimization

ICLR 2025poster

Dataset distillation (DD) has emerged as a widely adopted technique for crafting a synthetic dataset that captures the essential information of a training dataset, facilitating the training of accurate neural models. Its applications span various domains, including transfer learning, federated learn…

2025

KVCOMM: Online Cross-context KV-cache Communication for Efficient LLM-based Multi-agent Systems

NeurIPS 2025poster

Multi-agent large language model (LLM) systems are increasingly adopted for complex language processing tasks that require communication and coordination among agents. However, these systems often suffer substantial overhead from repeated reprocessing of overlapping contexts across agents. In typica…

Cited by 0SourcecodeScholar
2025

Keyframe-oriented Vision Token Pruning: Enhancing Efficiency of Large Vision Language Models on Long-Form Video Processing

ICCV 2025poster

Vision language models (VLMs) demonstrate strong capabilities in jointly processing visual and textual data. However, they often incur substantial computational overhead due to redundant visual information, particularly in long-form video scenarios. Existing approaches predominantly focus on either…

2025

LaMAGIC2: Advanced Circuit Formulations for Language Model-Based Analog Topology Generation

ICML 2025poster

Automation of analog topology design is crucial due to customized requirements of modern applications with heavily manual engineering efforts. The state-of-the-art work applies a sequence-to-sequence approach and supervised finetuning on language models to generate topologies given user specificati…

Cited by 0SourcePDFScholar
2025

Proactive Privacy Amnesia for Large Language Models: Safeguarding PII with Negligible Impact on Model Utility

ICLR 2025poster

With the rise of large language models (LLMs), increasing research has recognized their risk of leaking personally identifiable information (PII) under malicious attacks. Although efforts have been made to protect PII in LLMs, existing methods struggle to balance privacy protection with maintaining…

Cited by 3SourcePDFScholar
2025

SADA: Stability-guided Adaptive Diffusion Acceleration

ICML 2025poster

Diffusion models have achieved remarkable success in generative tasks but suffer from high computational costs due to their iterative sampling process and quadratic‐attention costs. Existing training-free acceleration strategies that reduce per-step computation cost, while effectively reducing samp…

2025

SAFER: Sharpness Aware layer-selective Finetuning for Enhanced Robustness in vision transformers

ICCV 2025poster

Vision transformers (ViTs) have become essential backbones in advanced computer vision applications and multi-modal foundation models. Despite their strengths, ViTs remain vulnerable to adversarial perturbations, comparable to or even exceeding the vulnerability of convolutional neural networks (CNN…

Cited by 0SourcePDFScholar
2025

Which Agent Causes Task Failures and When? On Automated Failure Attribution of LLM Multi-Agent Systems

ICML 2025spotlight

Failure attribution in LLM multi-agent systems—identifying the agent and step responsible for task failures—provides crucial clues for systems debugging but remains underexplored and labor-intensive. In this paper, we propose and formulate a new research area: automated failure attribution for LLM…

2024

Efficient Dataset Distillation via Minimax Diffusion

CVPR 2024poster

Dataset distillation reduces the storage and computational consumption of training a network by generating a small surrogate dataset that encapsulates rich information of the original large-scale one. However previous distillation methods heavily rely on the sample-wise iterative optimization scheme…

2024

FedBPT: Efficient Federated Black-box Prompt Tuning for Large Language Models

ICML 2024poster

Pre-trained language models (PLM) have revolutionized the NLP landscape, achieving stellar performances across diverse tasks. These models, while benefiting from vast training data, often require fine-tuning on specific data to cater to distinct downstream tasks. However, this data adaptation proces…

Cited by 34SourcePDFScholar
2024

LaMAGIC: Language-Model-based Topology Generation for Analog Integrated Circuits

ICML 2024poster

In the realm of electronic and electrical engineering, automation of analog circuit is increasingly vital given the complexity and customized requirements of modern applications. However, existing methods only develop search-based algorithms that require many simulation iterations to design a custom…

Cited by 11SourcePDFScholar
2024

Reimagining Mutual Information for Enhanced Defense against Data Leakage in Collaborative Inference

NeurIPS 2024poster

Edge-cloud collaborative inference empowers resource-limited IoT devices to support deep learning applications without disclosing their raw data to the cloud server, thus protecting user's data. Nevertheless, prior research has shown that collaborative inference still results in the exposure of inpu…

Cited by 0SourcePDFScholar
2024

SLED: Self Logits Evolution Decoding for Improving Factuality in Large Language Models

NeurIPS 2024poster

Large language models (LLMs) have demonstrated remarkable capabilities, but their outputs can sometimes be unreliable or factually incorrect. To address this, we introduce Self Logits Evolution Decoding (SLED), a novel decoding framework that enhances the truthfulness of LLMs without relying on exte…

2024

Towards Building The Federatedgpt: Federated Instruction Tuning

ICASSP 2024accepted

While "instruction-tuned" generative large language models (LLMs) have demonstrated an impressive ability to generalize to new tasks, the training phases heavily rely on large amounts of diverse and high-quality instruction data (such as ChatGPT and GPT-4). Unfortunately, acquiring high-quality data…

Cited by 0SourceScholar
2024

Unlocking the Potential of Federated Learning: The Symphony of Dataset Distillation via Deep Generative Latents

ECCV 2024poster

"Data heterogeneity presents significant challenges for federated learning (FL). Recently, dataset distillation techniques have been introduced, and performed at the client level, to attempt to mitigate some of these challenges. In this paper, we propose a highly efficient FL dataset distillation fr…

2023

Communication-Efficient Vertical Federated Learning with Limited Overlapping Samples

ICCV 2023poster

Federated learning is a popular collaborative learning approach that enables clients to train a global model without sharing their local data. Vertical federated learning (VFL) deals with scenarios in which the data on clients have different feature spaces but share some overlapping samples. Existin…

Cited by 18PDFScholar
2023

Fed-CBS: A Heterogeneity-Aware Client Sampling Mechanism for Federated Learning via Class-Imbalance Reduction

ICML 2023poster

Due to the often limited communication bandwidth of edge devices, most existing federated learning (FL) methods randomly select only a subset of devices to participate in training at each communication round. Compared with engaging all the available clients, such a random-selection mechanism could l…

Cited by 58SourcePDFScholar
2023

LISSNAS: Locality-based Iterative Search Space Shrinkage for Neural Architecture Search

IJCAI 2023poster

Search spaces hallmark the advancement of Neural Architecture Search (NAS). Large and complex search spaces with versatile building operators and structures provide more opportunities to brew promising architectures, yet pose severe challenges on efficient exploration and exploitation. Subsequently,…

Cited by 6SourcePDFScholar
2023

ReAugKD: Retrieval-Augmented Knowledge Distillation For Pre-trained Language Models

ACL 2023short

Knowledge Distillation (KD) is one of the most effective approaches to deploying large-scale pre-trained language models in low-latency environments by transferring the knowledge contained in the large-scale models to smaller student models. Prior KD approaches use the soft labels and intermediate a…

Cited by 23SourcePDFScholar
2022

FedCor: Correlation-Based Active Client Selection Strategy for Heterogeneous Federated Learning

CVPR 2022poster

Client-wise data heterogeneity is one of the major issues that hinder effective training in federated learning (FL). Since the data distribution on each client may vary dramatically, the client selection strategy can significantly influence the convergence rate of the FL process. Active client selec…

Cited by 138PDFScholar
2022

Language-specific Effects on Automatic Speech Recognition Errors for World Englishes

COLING 2022main

Despite recent advancements in automated speech recognition (ASR) technologies, reports of unequal performance across speakers of different demographic groups abound. At the same time, the focus on performance metrics such as the Word Error Rate (WER) in prior studies limit the specificity and scope…

2022

MTG: A Benchmark Suite for Multilingual Text Generation

NAACL 2022findings

We introduce MTG, a new benchmark suite for training and evaluating multilingual text generation. It is the first-proposed multilingual multiway text generation dataset with the largest human-annotated data (400k). It includes four generation tasks (story generation, question generation, title gener…

2022

Why do We Need Large Batchsizes in Contrastive Learning? A Gradient-Bias Perspective

NeurIPS 2022accept

Contrastive learning (CL) has been the de facto technique for self-supervised representation learning (SSL), with impressive empirical success such as multi-modal representation learning. However, traditional CL loss only considers negative samples from a minibatch, which could cause biased gradient…

Cited by 40SourcePDFScholar
2021

Are Factuality Checkers Reliable? Adversarial Meta-evaluation of Factuality in Summarization

EMNLP 2021finding

With the continuous upgrading of the summarization systems driven by deep neural networks, researchers have higher requirements on the quality of the generated summaries, which should be not only fluent and informative but also factually correct. As a result, the field of factual evaluation has deve…

2021

BSQ: Exploring Bit-Level Sparsity for Mixed-Precision Neural Network Quantization

ICLR 2021poster

Mixed-precision quantization can potentially achieve the optimal tradeoff between performance and compression rate of deep neural networks, and thus, have been widely investigated. However, it lacks a systematic method to determine the exact quantization scheme. Previous methods either examine only…

2021

Enhancing Scientific Papers Summarization with Citation Graph

AAAI 2021technical

Previous work for text summarization in scientific domain mainly focused on the content of the input document, but seldom considering its citation network. However, scientific papers are full of uncommon domain-specific terms, making it almost impossible for the model to understand its true meaning…

2021

FL-WBC: Enhancing Robustness against Model Poisoning Attacks in Federated Learning from a Client Perspective

NeurIPS 2021poster

Federated learning (FL) is a popular distributed learning framework that trains a global model through iterative communications between a central server and edge devices. Recent works have demonstrated that FL is vulnerable to model poisoning attacks. Several server-based defense approaches (e.g. ro…

2021

Improving Gradient Regularization using Complex-Valued Neural Networks

ICML 2021spotlight

Gradient regularization is a neural network defense technique that requires no prior knowledge of an adversarial attack and that brings only limited increase in training computational complexity. A form of complex-valued neural network (CVNN) is proposed to improve the performance of gradient regula…

2021

NASGEM: Neural Architecture Search via Graph Embedding Method

AAAI 2021technical

Neural Architecture Search (NAS) automates and prospers the design of neural networks. Estimator-based NAS has been proposed recently to model the relationship between architectures and their performance to enable scalable and flexible search. However, existing estimator-based methods encode the arc…

Cited by 24SourcePDFScholar
2021

Soteria: Provable Defense Against Privacy Leakage in Federated Learning From Representation Perspective

CVPR 2021poster

Federated learning (FL) is a popular distributed learning framework that can reduce privacy risks by not explicitly sharing private data. However, recent works have demonstrated that sharing model updates makes FL vulnerable to inference attack. In this work, we show our key observation that the dat…

Cited by 204PDFcodeScholar
2020

Accelerating CNN Training by Pruning Activation Gradients

ECCV 2020poster

Sparsification is an efficient approach to accelerate CNN inference, but it is challenging to take advantage of sparsity in training procedure because the involved gradients are dynamically changed. Actually, an important observation shows that most of the activation gradients in back-propagation ar…

Cited by 52SourcePDFScholar
2020

Neural Predictor for Neural Architecture Search

ECCV 2020poster

Neural Architecture Search methods are effective but often use complex algorithms to come up with the best architecture. We propose an approach with three basic steps that is conceptually much simpler. First we train $N$ random architectures to generate $N$ (architecture, validation accuracy) pairs…

Cited by 246SourcePDFScholar
2020

Perturbing Across the Feature Hierarchy to Improve Standard and Strict Blackbox Attack Transferability

NeurIPS 2020poster

We consider the blackbox transfer-based targeted adversarial attack threat model in the realm of deep neural network (DNN) image classifiers. Rather than focusing on crossing decision boundaries at the output layer of the source model, our method perturbs representations throughout the extracted fea…

Cited by 95SourcePDFScholar
2020

Structural Sparsification for Far-Field Speaker Recognition with Intel® Gna

ICASSP 2020accepted

Recently, deep neural networks (DNN) have been widely used in speaker recognition area. In order to achieve fast response time and high accuracy, the requirements for hardware resources increase rapidly. However, as the speaker recognition application is often implemented on mobile devices, it is ne…

Cited by 0SourceScholar
2020

TRP: Trained Rank Pruning for Efficient Deep Neural Networks

IJCAI 2020poster

To enable DNNs on edge devices like mobile phones, low-rank approximation has been widely adopted because of its solid theoretical rationale and efficient implementations. Several previous works attempted to directly approximate a pre-trained model by low-rank decomposition; however, small approxima…

Cited by 0SourcePDFScholar
2019

Feature Space Perturbations Yield More Transferable Adversarial Examples

CVPR 2019poster

Many recent works have shown that deep learning models are vulnerable to quasi-imperceptible input perturbations, yet practitioners cannot fully explain this behavior. This work describes a transfer-based blackbox targeted adversarial attack of deep feature space representations that also provides i…

Cited by 239PDFScholar
2019

Learning Efficient Sparse Structures in Speech Recognition

ICASSP 2019accepted

Recurrent neural networks (RNNs), especially long short-term memories (LSTMs) have been widely used in speech recognition and natural language processing. As the sizes of RNN models grow for better performance, the computation cost and therefore the required hardware resource increase rapidly. We pr…

Cited by 0SourceScholar
2018

Learning Intrinsic Sparse Structures within Long Short-Term Memory

ICLR 2018poster

Model compression is significant for the wide adoption of Recurrent Neural Networks (RNNs) in both user devices possessing limited resources and business clusters requiring quick responses to large-scale service requests. This work aims to learn structurally-sparse Long Short-Term Memory (LSTM) by r…

Cited by 161SourcePDFScholar
2017

A Compact DNN: Approaching GoogLeNet-Level Accuracy of Classification and Domain Adaptation

CVPR 2017poster

Recently, DNN model compression based on network architecture design, e.g., SqueezeNet, attracted a lot attention. No accuracy drop on image classification is observed on these extremely compact networks, compared to well-known models. An emerging question, however, is whether these model compressio…

Cited by 71PDFScholar
2017

Coordinating Filters for Faster Deep Neural Networks

ICCV 2017poster

Very large-scale Deep Neural Networks (DNNs) have achieved remarkable successes in a large variety of computer vision tasks. However, the high computation intensity of DNNs makes it challenging to deploy these models on resource-limited systems. Some studies used low-rank approaches that approximate…

Cited by 188PDFcodeScholar
2017

Faster CNNs with Direct Sparse Convolutions and Guided Pruning

ICLR 2017poster

Phenomenally successful in practical inference problems, convolutional neural networks (CNN) are widely deployed in mobile devices, data centers, and even supercomputers. The number of parameters needed in CNNs, however, are often large and undesirable. Consequently, various methods have been develo…

Cited by 307SourcecodeScholar
2017

TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep Learning

NeurIPS 2017oral

High network communication cost for synchronizing gradients and parameters is the well-known bottleneck of distributed training. In this work, we propose TernGrad that uses ternary gradients to accelerate distributed deep learning in data parallelism. Our approach requires only three numerical level…

2016

Learning Structured Sparsity in Deep Neural Networks

NeurIPS 2016poster

High demand for computation resources severely hinders deployment of large-scale Deep Neural Networks (DNN) in resource constrained devices. In this work, we propose a Structured Sparsity Learning (SSL) method to regularize the structures (i.e., filters, channels, filter shapes, and layer depth) of…