← Search

Mingyi Hong

109 accepted papers

2026

GUI-Spotlight: Adaptive Iterative Focus Refinement for Enhanced GUI Visual Grounding

ICML 2026poster

Multimodal large language models (MLLMs) have markedly expanded the competence of graphical user-interface (GUI) systems, propelling them beyond controlled simulations into complex, real-world environments across diverse platforms. However, practical usefulness is still bounded by the reliability of…

Cited by 0SourceScholar
2026

HiPER: Hierarchical Plan–Execute RL for Multi-Turn LLM Agents

ICML 2026poster

Training LLMs as interactive agents for multi-turn decision-making remains challenging, particularly in long-horizon tasks with sparse and delayed rewards, where agents must execute extended sequences of actions before receiving meaningful feedback. Most existing reinforcement learning (RL) methods …

Cited by 0SourceScholar
2026

Leak@$k$: Unlearning Does Not Make LLMs Forget Under Probabilistic Decoding

ICML 2026poster

Unlearning in large language models (LLMs) is critical for regulatory compliance and for building ethical generative AI systems that avoid producing private, toxic, illegal, or copyrighted content. Despite rapid progress, in this work we show that \textit{almost all} existing unlearning methods fail…

Cited by 0SourceScholar
2026

Muon Outperforms Adam in Tail-End Associative Memory Learning

ICLR 2026poster

The Muon optimizer is consistently faster than Adam in training Large Language Models (LLMs), yet the mechanism underlying its success remains unclear. This paper demystifies this mechanism through the lens of associative memory. By ablating the transformer components optimized by Muon, we reveal th…

Cited by 0SourceScholar
2026

StitchCUDA: An Automated Multi-Agents End-to-End GPU Programing Framework with Rubric-based Agentic Reinforcement Learning

ICML 2026poster

Modern machine learning (ML) workloads increasingly rely on GPUs, yet achieving high end-to-end performance remains challenging due to dependencies on both GPU kernel efficiency and host-side settings. Although LLM-based methods show promise on automated GPU kernel generation, prior works mainly foc…

Cited by 0SourceScholar
2025

AssistedDS: Benchmarking How External Domain Knowledge Assists LLMs in Automated Data Science

EMNLP 2025

Large language models (LLMs) have advanced the automation of data science workflows. Yet it remains unclear whether they can critically leverage external domain knowledge as human data scientists do in practice. To answer this question, we introduce AssistedDS (Assisted Data Science), a benchmark de

2025

BRiTE: Bootstrapping Reinforced Thinking Process to Enhance Language Model Reasoning

ICML 2025poster

Large Language Models (LLMs) have demonstrated remarkable capabilities in complex reasoning tasks, yet generating reliable reasoning processes remains a significant challenge. We present a unified probabilistic framework that formalizes LLM reasoning through a novel graphical model incorporating lat…

Cited by 1SourcePDFScholar
2025

DiSK: Differentially Private Optimizer with Simplified Kalman Filter for Noise Reduction

ICLR 2025poster

Differential privacy (DP) offers a robust framework for safeguarding individual data privacy. To utilize DP in training modern machine learning models, differentially private optimizers have been widely used in recent years. A popular approach to privatize an optimizer is to clip the individual gra…

Cited by 1SourcePDFScholar
2025

Do LLMs Recognize Your Preferences? Evaluating Personalized Preference Following in LLMs

ICLR 2025oral

Large Language Models (LLMs) are increasingly deployed as chatbots, yet their ability to personalize responses to user preferences remains limited. We introduce PrefEval, a benchmark for evaluating LLMs' ability to infer, memorize and adhere to user preferences in long-context conversational setting…

2025

InfantAgent-Next: A Multimodal Generalist Agent for Automated Computer Interaction

NeurIPS 2025poster

This paper introduces \textsc{InfantAgent-Next}, a generalist agent capable of interacting with computers in a multimodal manner, encompassing text, images, audio, and video. Unlike existing approaches that either build intricate workflows around a single large model or only provide workflow modular…

Cited by 0SourcecodeScholar
2025

Inference-Time Alignment of Diffusion Models with Direct Noise Optimization

ICML 2025poster

In this work, we focus on the alignment problem of diffusion models with a continuous reward function, which represents specific objectives for downstream tasks, such as increasing darkness or improving the aesthetics of images. The central goal of the alignment problem is to adjust the distribution…

Cited by 0SourcePDFScholar
2025

Joint Reward and Policy Learning with Demonstrations and Human Feedback Improves Alignment

ICLR 2025spotlight

Aligning to human preferences and/or intentions is an important requirement for contemporary foundation models. To ensure alignment, popular approaches such as reinforcement learning with human feedback (RLHF) break down the task into three stages: (i) a model is computed with supervised fine-tuning…

Cited by 0SourcePDFScholar
2025

LUME: LLM Unlearning with Multitask Evaluations

EMNLP 2025

Unlearning aims to remove copyrighted, sensitive, or private content from large language models (LLMs) without a full retraining. In this work, we develop a multi-task unlearning benchmark LUME that features three tasks: (1) unlearn synthetically generated creative short novels, (2) unlearn syntheti

2025

On the Vulnerability of Applying Retrieval-Augmented Generation within Knowledge-Intensive Application Domains

ICML 2025poster

Retrieval-Augmented Generation (RAG) has been empirically shown to enhance the performance of large language models (LLMs) in knowledge-intensive domains such as healthcare, finance, and legal contexts. Given a query, RAG retrieves relevant documents from a corpus and integrates them into the LLMs’…

Cited by 5SourcePDFScholar
2025

RoSTE: An Efficient Quantization-Aware Supervised Fine-Tuning Approach for Large Language Models

ICML 2025poster

Supervised fine-tuning is a standard method for adapting pre-trained large language models (LLMs) to downstream tasks. Quantization has been recently studied as a post-training technique for efficient LLM deployment. To obtain quantized fine-tuned LLMs, conventional pipelines would first fine-tune t…

2025

Split-Merge: Scalable and Memory-Efficient Merging of Expert LLMs

EMNLP 2025

We introduce a zero-shot merging framework for large language models (LLMs) that consolidates specialized domain experts into a single model without any further training. Our core contribution lies in leveraging relative task vectors—difference representations encoding each expert’s unique traits wi

2025

Towards LLM Unlearning Resilient to Relearning Attacks: A Sharpness-Aware Minimization Perspective and Beyond

ICML 2025poster

The LLM unlearning technique has recently been introduced to comply with data regulations and address the safety and ethical concerns of LLMs by removing the undesired data-model influence. However, state-of-the-art unlearning methods face a critical vulnerability: they are susceptible to ``relearn…

2025

Understanding Inverse Reinforcement Learning under Overparameterization: Non-Asymptotic Analysis and Global Optimality

AISTATS 2025poster

The goal of the Inverse reinforcement learning (IRL) task is to identify the underlying reward function and the corresponding optimal policy from a set of expert demonstrations. While most IRL algorithms’ theoretical guarantees rely on a linear reward structure, we aim to extend the theoretical unde…

Cited by 0SourceScholar
2025

Unlearning as multi-task optimization: A normalized gradient difference approach with an adaptive learning rate

NAACL 2025long

Machine unlearning has been used to remove unwanted knowledge acquired by large language models (LLMs). In this paper, we examine machine unlearning from an optimization perspective, framing it as a regularized multi-task optimization problem, where one task optimizes a forgetting objective and anot…

Cited by 6SourcePDFScholar
2024

A Smoothed Bregman Proximal Gradient Algorithm for Decentralized Nonconvex Optimization

ICASSP 2024accepted

Decentralized computation has received considerable research interest lately, due to its wide applications in information processing systems. However, one key requirement to establish convergence for almost all decentralized algorithms, for convex and non-convex problems alike, is that the loss func…

Cited by 0SourceScholar
2024

DOPPLER: Differentially Private Optimizers with Low-pass Filter for Privacy Noise Reduction

NeurIPS 2024poster

Privacy is a growing concern in modern deep-learning systems and applications. Differentially private (DP) training prevents the leakage of sensitive information in the collected training data from the trained machine learning models. DP optimizers, including DP stochastic gradient descent (DPSGD) a…

Cited by 2SourcePDFScholar
2024

Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models

NeurIPS 2024poster

Diffusion models (DMs) have achieved remarkable success in text-to-image generation, but they also pose safety risks, such as the potential generation of harmful content and copyright violations. The techniques of machine unlearning, also known as concept erasing, have been developed to address thes…

2024

Demystifying Poisoning Backdoor Attacks from a Statistical Perspective

ICLR 2024poster

Backdoor attacks pose a significant security risk to machine learning applications due to their stealthy nature and potentially serious consequences. Such attacks involve embedding triggers within a learning model with the intention of causing malicious behavior when an active trigger is present whi…

2024

Differentially Private SGD Without Clipping Bias: An Error-Feedback Approach

ICLR 2024poster

Differentially Private Stochastic Gradient Descent with Gradient Clipping (DPSGD-GC) is a powerful tool for training deep learning models using sensitive data, providing both a solid theoretical privacy guarantee and high efficiency. However, existing research has shown that DPSGD-GC only converges…

2024

EMC$^2$: Efficient MCMC Negative Sampling for Contrastive Learning with Global Convergence

ICML 2024poster

A key challenge in contrastive learning is to generate negative samples from a large sample set to contrast with positive samples, for learning better encoding of the data. These negative samples often follow a softmax distribution which are dynamically updated during the training process. However,…

2024

Getting More Juice Out of the SFT Data: Reward Learning from Human Demonstration Improves SFT for LLM Alignment

NeurIPS 2024poster

Aligning human preference and value is an important requirement for contemporary foundation models. State-of-the-art techniques such as Reinforcement Learning from Human Feedback (RLHF) often consist of two stages: 1) supervised fine-tuning (SFT), where the model is fine-tuned by learning from human…

2024

Krylov Cubic Regularized Newton: A Subspace Second-Order Method with Dimension-Free Convergence Rate

AISTATS 2024poster

Second-order optimization methods, such as cubic regularized Newton methods, are known for their rapid convergence rates; nevertheless, they become impractical in high-dimensional problems due to their substantial memory requirements and computational costs. One promising approach is to execute seco…

Cited by 1SourcePDFScholar
2024

MADA: Meta-Adaptive Optimizers Through Hyper-Gradient Descent

ICML 2024poster

Following the introduction of Adam, several novel adaptive optimizers for deep learning have been proposed. These optimizers typically excel in some tasks but may not outperform Adam uniformly across all tasks. In this work, we introduce Meta-Adaptive Optimizers (MADA), a unified optimizer framework…

Cited by 3SourcePDFScholar
2024

Pre-training Differentially Private Models with Limited Public Data

NeurIPS 2024poster

The superior performance of large foundation models can be attributed to the use of massive amounts of high-quality data. However, such datasets often contain sensitive, private and copyrighted material that requires formal protection. While differential privacy (DP) is a prominent method used to ga…

2024

RAW: A Robust and Agile Plug-and-Play Watermark Framework for AI-Generated Images with Provable Guarantees

NeurIPS 2024poster

Safeguarding intellectual property and preventing potential misuse of AI-generated images are of paramount importance. This paper introduces a robust and agile plug-and-play watermark detection framework, referred to as RAW. As a departure from existing encoder-decoder methods, which incorporate fix…

2024

Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmark

ICML 2024poster

In the evolving landscape of natural language processing (NLP), fine-tuning pre-trained Large Language Models (LLMs) with first-order (FO) optimizers like SGD and Adam has become standard. Yet, as LLMs grow in size, the substantial memory overhead from back-propagation (BP) for FO gradient computati…

2024

SLTrain: a sparse plus low rank approach for parameter and memory efficient pretraining

NeurIPS 2024poster

Large language models (LLMs) have shown impressive capabilities across various tasks. However, training LLMs from scratch requires significant computational power and extensive memory capacity. Recent studies have explored low-rank structures on weights for efficient fine-tuning in terms of paramete…

2024

Unraveling the Gradient Descent Dynamics of Transformers

NeurIPS 2024poster

While the Transformer architecture has achieved remarkable success across various domains, a thorough theoretical foundation explaining its optimization dynamics is yet to be fully developed. In this study, we aim to bridge this understanding gap by answering the following two core questions: (1) Wh…

Cited by 1SourcePDFScholar
2023

A Bayesian Approach to Robust Inverse Reinforcement Learning

CoRL 2023poster

We consider a Bayesian approach to offline model-based inverse reinforcement learning (IRL). The proposed framework differs from existing offline model-based IRL approaches by performing simultaneous estimation of the expert's reward function and subjective model of environment dynamics. We make use…

Cited by 7SourcecodeScholar
2023

A Unified Detection Framework for Inference-Stage Backdoor Defenses

NeurIPS 2023poster

Backdoor attacks involve inserting poisoned samples during training, resulting in a model containing a hidden backdoor that can trigger specific behaviors without impacting performance on normal samples. These attacks are challenging to detect, as the backdoored model appears normal until activated…

Cited by 14SourcePDFScholar
2023

An Implicit Gradient Method for Constrained Bilevel Problems Using Barrier Approximation

ICASSP 2023accepted

In this work, we propose algorithms for solving a class of Bilevel Optimization (BLO) problems, with applications in areas such as signal processing, networking and machine learning. Specifically, we develop a novel barrier-based gradient approximation algorithm that transforms the constrained BLO p…

Cited by 0SourceScholar
2023

FedAvg Converges to Zero Training Loss Linearly for Overparameterized Multi-Layer Neural Networks

ICML 2023poster

Federated Learning (FL) is a distributed learning paradigm that allows multiple clients to learn a joint model by utilizing privately held data at each client. Significant research efforts have been devoted to develop advanced algorithms that deal with the situation where the data at individual clie…

Cited by 8SourcePDFScholar
2023

Linearly Constrained Bilevel Optimization: A Smoothed Implicit Gradient Approach

ICML 2023poster

This work develops analysis and algorithms for solving a class of bilevel optimization problems where the lower-level (LL) problems have linear constraints. Most of the existing approaches for constrained bilevel problems rely on value function-based approximate reformulations, which suffer from iss…

Cited by 22SourcePDFScholar
2023

Selectivity Drives Productivity: Efficient Dataset Pruning for Enhanced Transfer Learning

NeurIPS 2023poster

Massive data is often considered essential for deep learning applications, but it also incurs significant computational and infrastructural costs. Therefore, dataset pruning (DP) has emerged as an effective way to improve data efficiency by identifying and removing redundant training samples without…

2023

Towards Efficient and Optimal Joint Beamforming and Antenna Selection: A Machine Learning Approach

ICASSP 2023accepted

This work revisits the joint transmit beamforming and antenna selection problem. Existing approaches find approximate solutions to this NP-hard problem via various heuristics, e.g., convex/nonconvex relaxation, greedy method, and (deep) supervised learning. However, optimality (or even feasibility)…

Cited by 0SourceScholar
2023

Understanding Backdoor Attacks through the Adaptability Hypothesis

ICML 2023poster

A poisoning backdoor attack is a rising security concern for deep learning. This type of attack can result in the backdoored model functioning normally most of the time but exhibiting abnormal behavior when presented with inputs containing the backdoor trigger, making it difficult to detect and prev…

Cited by 14SourcePDFScholar
2023

VCC: Scaling Transformers to 128K Tokens or More by Prioritizing Important Tokens

NeurIPS 2023poster

Transformers are central in modern natural language processing and computer vision applications. Despite recent works devoted to reducing the quadratic cost of such models with respect to sequence length, dealing with ultra long sequences (e.g., $>$16K tokens) remains challenging. Applications such…

2023

What Is Missing in IRM Training and Evaluation? Challenges and Solutions

ICLR 2023poster

Invariant risk minimization (IRM) has received increasing attention as a way to acquire environment-agnostic data representations and predictions, and also a principled solution for preventing spurious correlations from being learned and improving models’ out-of-distribution generalization. Yet, rec…

Cited by 8SourcePDFScholar
2023

When Demonstrations meet Generative World Models: A Maximum Likelihood Framework for Offline Inverse Reinforcement Learning

NeurIPS 2023oral

Offline inverse reinforcement learning (Offline IRL) aims to recover the structure of rewards and environment dynamics that underlie observed actions in a fixed, finite set of demonstrations from an expert agent. Accurate models of expertise in executing a task has applications in safety-sensitive a…

2022

A Stochastic Linearized Augmented Lagrangian Method for Decentralized Bilevel Optimization

NeurIPS 2022accept

Bilevel optimization has been shown to be a powerful framework for formulating multi-task machine learning problems, e.g., reinforcement learning (RL) and meta-learning, where the decision variables are coupled in both levels of the minimization problems. In practice, the learning tasks would be loc…

Cited by 17SourcePDFScholar
2022

A Stochastic Multi-Rate Control Framework For Modeling Distributed Optimization Algorithms

ICML 2022spotlight

In modern machine learning systems, distributed algorithms are deployed across applications to ensure data privacy and optimal utilization of computational resources. This work offers a fresh perspective to model, analyze, and design distributed optimization algorithms through the lens of stochastic…

Cited by 1SourcePDFScholar
2022

Advancing Model Pruning via Bi-level Optimization

NeurIPS 2022accept

The deployment constraints in practical applications necessitate the pruning of large-scale deep learning models, i.e., promoting their weight sparsity. As illustrated by the Lottery Ticket Hypothesis (LTH), pruning also has the potential of improving their generalization ability. At the core of LTH…

2022

An Implicit Gradient-Type Method for Linearly Constrained Bilevel Problems

ICASSP 2022accepted

In this work, we develop an implicit gradient-type (IG-AL) algorithm for bilevel optimization with strongly convex linear inequality constrained lower-level problems. Many learning problems of interest, including problems in distributed optimization, machine learning, economics, and transport resear…

Cited by 0SourceScholar
2022

Decentralized Learning for Overparameterized Problems: A Multi-Agent Kernel Approximation Approach

ICLR 2022poster

This work develops a novel framework for communication-efficient distributed learning where the models to be learned are overparameterized. We focus on a class of kernel learning problems (which includes the popular neural tangent kernel (NTK) learning as a special case) and propose a novel {\it mul…

Cited by 0SourcePDFScholar
2022

Distributed Optimization for Overparameterized Problems: Achieving Optimal Dimension Independent Communication Complexity

NeurIPS 2022accept

Decentralized optimization are playing an important role in applications such as training large machine learning models, among others. Despite its superior practical performance, there has been some lack of fundamental understanding about its theoretical properties. In this work, we address the foll…

Cited by 5SourcePDFScholar
2022

Distributed adversarial training to robustify deep neural networks at scale

UAI 2022poster

Current deep neural networks (DNNs) are vulnerable to adversarial attacks, where adversarial perturbations to the inputs can change or manipulate classification. To defend against such attacks, an effective and popular approach, known as adversarial training (AT), has been shown to mitigate the nega…

2022

How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspective

ICLR 2022spotlight

The lack of adversarial robustness has been recognized as an important issue for state-of-the-art machine learning (ML) models, e.g., deep neural networks (DNNs). Thereby, robustifying ML models against adversarial attacks is now a major focus of research. However, nearly all existing defense method…

2022

Inducing Equilibria via Incentives: Simultaneous Design-and-Play Ensures Global Convergence

NeurIPS 2022accept

To regulate a social system comprised of self-interested agents, economic incentives are often required to induce a desirable outcome. This incentive design problem naturally possesses a bilevel structure, in which a designer modifies the payoffs of the agents with incentives while anticipating the…

Cited by 19SourcePDFScholar
2022

Maximum-Likelihood Inverse Reinforcement Learning with Finite-Time Guarantees

NeurIPS 2022accept

Inverse reinforcement learning (IRL) aims to recover the reward function and the associated optimal policy that best fits observed sequences of states and actions implemented by an expert. Many algorithms for IRL have an inherent nested structure: the inner loop finds the optimal policy given parame…

Cited by 44SourcePDFScholar
2022

Mismatched Supervised Learning

ICASSP 2022accepted

Supervised learning scenarios, where labels and features are possibly mismatched, have been an emerging concern in machine learning applications. For example, researchers often need to align heterogeneous data from multiple resources to the same entities without a unique identifier in the socioecono…

Cited by 0SourceScholar
2022

Revisiting and Advancing Fast Adversarial Training Through The Lens of Bi-Level Optimization

ICML 2022spotlight

Adversarial training (AT) is a widely recognized defense mechanism to gain the robustness of deep neural networks against adversarial attacks. It is built on min-max optimization (MMO), where the minimizer (i.e., defender) seeks a robust model to minimize the worst-case training loss in the presence…

2022

Understanding Clipping for Federated Learning: Convergence and Client-Level Differential Privacy

ICML 2022spotlight

Providing privacy protection has been one of the primary motivations of Federated Learning (FL). Recently, there has been a line of work on incorporating the formal privacy notion of differential privacy with FL. To guarantee the client-level differential privacy in FL algorithms, the clients’ trans…

Cited by 133SourcePDFScholar
2021

A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentum

NeurIPS 2021poster

This paper proposes a new algorithm -- the \underline{S}ingle-timescale Do\underline{u}ble-momentum \underline{St}ochastic \underline{A}pprox\underline{i}matio\underline{n} (SUSTAIN) -- for tackling stochastic unconstrained bilevel optimization problems. We focus on bilevel problems where the lower…

Cited by 147SourcePDFScholar
2021

Decentralized Riemannian Gradient Descent on the Stiefel Manifold

ICML 2021spotlight

We consider a distributed non-convex optimization where a network of agents aims at minimizing a global function over the Stiefel manifold. The global function is represented as a finite sum of smooth local functions, where each local function is associated with one agent and agents communicate with…

2021

Deep Generative Model Learning For Blind Spectrum Cartography with NMF-Based Radio Map Disaggregation

ICASSP 2021accepted

Spectrum cartography (SC) aims at estimating the multi-aspect (e.g., space, frequency, and time) interference level caused by multiple emitters from limited measurements. Early SC approaches rely on model assumptions about the radio map, e.g., sparsity and smoothness, which may be grossly violated u…

Cited by 0SourceScholar
2021

Fiber-Sampled Stochastic Mirror Descent for Tensor Decomposition with β-Divergence

ICASSP 2021accepted

Canonical polyadic decomposition (CPD) has been a workhorse for multimodal data analytics. This work puts forth a stochastic algorithmic framework for CPD under β-divergence, which is well-motivated in statistical learning—where the Euclidean distance is typically not preferred. Despite the existenc…

Cited by 0SourceScholar
2021

Finding First-Order Nash Equilibria of Zero-Sum Games with the Regularized Nikaido-Isoda Function

AISTATS 2021poster

Efficiently finding First-order Nash Equilibria (FNE) in zero-sum games can be challenging, even in a two-player setting. This work proposes an algorithm for finding the FNEs of a two-player zero-sum game, in which the local cost functions can be non-convex, and the players only have access to local…

Cited by 5SourcePDFScholar
2021

Generalization Bounds for Stochastic Saddle Point Problems

AISTATS 2021poster

This paper studies the generalization bounds for the empirical saddle point (ESP) solution to stochastic saddle point (SSP) problems. For SSP with Lipschitz continuous and strongly convex-strongly concave objective functions, we establish an $O\left(1/n\right)$ generalization bound by using a probab…

Cited by 43SourcePDFScholar
2021

Learning to Continuously Optimize Wireless Resource in Episodically Dynamic Environment

ICASSP 2021accepted

There has been a growing interest in developing data-driven, in particular deep neural network (DNN) based methods for modern communication tasks. For a few popular tasks such as power control, beamforming, and MIMO detection, these methods achieve state-of-the-art performance while requiring less c…

Cited by 0SourceScholar
2021

STEM: A Stochastic Two-Sided Momentum Algorithm Achieving Near-Optimal Sample and Communication Complexities for Federated Learning

NeurIPS 2021poster

Federated Learning (FL) refers to the paradigm where multiple worker nodes (WNs) build a joint model by using local data. Despite extensive research, for a generic non-convex FL problem, it is not clear, how to choose the WNs' and the server's update directions, the minibatch sizes, and the local up…

Cited by 73SourcePDFScholar
2021

When Expressivity Meets Trainability: Fewer than $n$ Neurons Can Work

NeurIPS 2021poster

Modern neural networks are often quite wide, causing large memory and computation costs. It is thus of great interest to train a narrower network. However, training narrow neural nets remains a challenging task. We ask two theoretical questions: Can narrow networks have as strong expressivity as wid…

Cited by 14SourcePDFScholar
2020

Decentralized Min-Max Optimization: Formulations, Algorithms and Applications in Network Poisoning Attack

ICASSP 2020accepted

This paper discusses formulations and algorithms which allow a number of agents to collectively solve problems involving both (non-convex) minimization and (concave) maximization operations. These problems have a number of interesting applications in information processing and machine learning, and…

Cited by 0SourceScholar
2020

Distributed Training with Heterogeneous Data: Bridging Median- and Mean-Based Algorithms

NeurIPS 2020poster

Recently, there is a growing interest in the study of median-based algorithms for distributed non-convex optimization. Two prominent examples include signSGD with majority vote, an effective approach for communication reduction via 1-bit compression on the local gradients, and medianSGD, an algorith…

Cited by 83SourcePDFScholar
2020

Finding Second-Order Stationary Points Efficiently in Smooth Nonconvex Linearly Constrained Optimization Problems

NeurIPS 2020spotlight

This paper proposes two efficient algorithms for computing approximate second-order stationary points (SOSPs) of problems with generic smooth non-convex objective functions and generic linear constraints. While finding (approximate) SOSPs for the class of smooth non-convex linearly constrained probl…

Cited by 27SourcePDFScholar
2020

Improving the Sample and Communication Complexity for Decentralized Non-Convex Optimization: Joint Gradient Estimation and Tracking

ICML 2020poster

Many modern large-scale machine learning problems benefit from decentralized and stochastic optimization. Recent works have shown that utilizing both decentralized computing and local stochastic gradient estimates can outperform state-of-the-art centralized algorithms, in applications involving high…

Cited by 90SourcePDFScholar
2020

Min-Max Optimization without Gradients: Convergence and Applications to Black-Box Evasion and Poisoning Attacks

ICML 2020poster

In this paper, we study the problem of constrained min-max optimization in a black-box setting, where the desired optimizer cannot access the gradients of the objective function but may query its values. We present a principled optimization framework, integrating a zeroth-order (ZO) gradient estimat…

Cited by 100SourcePDFScholar
2020

Understanding Gradient Clipping in Private SGD: A Geometric Perspective

NeurIPS 2020spotlight

Deep learning models are increasingly popular in many machine learning applications where the training data may contain sensitive information. To provide formal and rigorous privacy guarantee, many learning systems now incorporate differential privacy by training their models with (differentially) p…

Cited by 246SourcePDFScholar
2019

Block Alternating Optimization for Non-convex Min-max Problems: Algorithms and Applications in Signal Processing and Communications

ICASSP 2019accepted

The min-max problem, also known as the saddle point problem, can be used to formulate a wide range of applications in signal processing and wireless communications. However, existing optimization theory and methods, which mostly deal with problems with certain convex-concave structure, are not appli…

Cited by 0SourceScholar
2019

Fast and Global Optimal Nonconvex Matrix Factorization via Perturbed Alternating Proximal Point

ICASSP 2019accepted

In this paper, we use the perturbed gradient based alternating minimization for solving a class of low-rank matrix factorization problems. Alternating minimization is a simple but popular approach which has been applied to problems in optimization, machine learning, data mining, and signal processin…

Cited by 0SourceScholar
2019

On Fast Convergence of Proximal Algorithms for SQRT-Lasso Optimization: Don’t Worry About its Nonsmooth Loss Function

UAI 2019poster

Many machine learning techniques sacrifice convenient computational structures to gain estimation robustness and modeling flexibility. However, by exploring the modeling structures, we find these “sacrifices” do not always require more computational efforts. To shed light on such a “free-lunch” phen…

Cited by 15SourcePDFScholar
2019

On the Convergence of A Class of Adam-Type Algorithms for Non-Convex Optimization

ICLR 2019poster

This paper studies a class of adaptive gradient based momentum algorithms that update the search directions and learning rates simultaneously using past gradients. This class, which we refer to as the ''``Adam-type'', includes the popular algorithms such as Adam, AMSGrad, AdaGrad. Despite their pop…

Cited by 406SourcePDFScholar
2019

PA-GD: On the Convergence of Perturbed Alternating Gradient Descent to Second-Order Stationary Points for Structured Nonconvex Optimization

ICML 2019oral

Alternating gradient descent (A-GD) is a simple but popular algorithm in machine learning, which updates two blocks of variables in an alternating manner using gradient descent steps. In this paper, we consider a smooth unconstrained nonconvex optimization problem, and propose a perturbed A-GD (PA-G…

Cited by 25SourcePDFScholar
2019

Perturbed Projected Gradient Descent Converges to Approximate Second-order Points for Bound Constrained Nonconvex Problems

ICASSP 2019accepted

In this paper, a gradient-based method for bound constrained non-convex problems is proposed. By leveraging both projected gradient descent and perturbed gradient descent, the proposed algorithm, named perturbed projected gradient descent (PP-GD), converges to some approximate second-order stationar…

Cited by 0SourceScholar
2019

Provably Global Convergence of Actor-Critic: A Case for Linear Quadratic Regulator with Ergodic Cost

NeurIPS 2019poster

Despite the empirical success of the actor-critic algorithm, its theoretical understanding lags behind. In a broader context, actor-critic can be viewed as an online alternating update algorithm for bilevel optimization, whose convergence is known to be fragile. To understand the instability of acto…

Cited by 171SourcePDFScholar
2019

Variance Reduced Policy Evaluation with Smooth Function Approximation

NeurIPS 2019poster

Policy evaluation with smooth and nonlinear function approximation has shown great potential for reinforcement learning. Compared to linear function approxi- mation, it allows for using a richer class of approximation functions such as the neural networks. Traditional algorithms are based on two tim…

Cited by 47SourcePDFScholar
2019

ZO-AdaMM: Zeroth-Order Adaptive Momentum Method for Black-Box Optimization

NeurIPS 2019poster

The adaptive momentum method (AdaMM), which uses past gradients to update descent directions and learning rates simultaneously, has become one of the most popular first-order optimization methods for solving machine learning problems. However, AdaMM is not suited for solving black-box optimization…

2018

Gradient Primal-Dual Algorithm Converges to Second-Order Stationary Solution for Nonconvex Distributed Optimization Over Networks

ICML 2018oral

In this work, we study two first-order primal-dual based algorithms, the Gradient Primal-Dual Algorithm (GPDA) and the Gradient Alternating Direction Method of Multipliers (GADMM), for solving a class of linearly constrained non-convex optimization problems. We show that with random initialization o…

Cited by 64SourcePDFScholar
2018

Large-Scale Regularized Sumcor GCCA via Penalty-Dual Decomposition

ICASSP 2018accepted

The sum-of-correlations (SUMCOR) generalized canonical correlation analysis (GCCA) aims at producing low-dimensional representations of multiview data via enforcing pairwise similarity of the reduced-dimension views. SUMCOR has been applied to a large variety of applications including blind separati…

Cited by 0SourceScholar
2018

Multi-Agent Reinforcement Learning via Double Averaging Primal-Dual Optimization

NeurIPS 2018poster

Despite the success of single-agent reinforcement learning, multi-agent reinforcement learning (MARL) remains challenging due to complex interactions between agents. Motivated by decentralized applications such as sensor networks, swarm robotics, and power grids, we study policy evaluation in MARL,…

Cited by 212SourcePDFScholar
2018

Software Defined Resource Allocation for Service-Oriented Networks

ICASSP 2018accepted

To support multiple on-demand services over several fixed communication networks, the network operators must allow flexible customization and fast provision of their network resources. One effective approach is network virtualization, whereby each service is mapped to a virtual subnetwork providing…

Cited by 0SourceScholar
2017

A Stochastic Nonconvex Splitting Method for Symmetric Nonnegative Matrix Factorization

AISTATS 2017poster

Symmetric nonnegative matrix factorization (SymNMF) plays an important role in applications of many data analytics problems such as community detection, document clustering and image segmentation. In this paper, we consider a stochastic SymNMF problem in which the observation matrix is generated in…

Cited by 6SourcePDFScholar
2017

A nonconvex splitting method for symmetric nonnegative matrix factorization: Convergence analysis and optimality

ICASSP 2017accepted

Symmetric non-negative matrix factorization (SymNMF) has important applications in data analytics problems such as document clustering, community detection and image segmentation. In this paper, we propose a novel nonconvex variable splitting method for solving SymNMF. Different from the existing wo…

Cited by 0SourceScholar
2017

Prox-PDA: The Proximal Primal-Dual Algorithm for Fast Distributed Nonconvex Optimization and Learning Over Networks

ICML 2017poster

In this paper we consider nonconvex optimization and learning over a network of distributed nodes. We develop a Proximal Primal-Dual Algorithm (Prox-PDA), which enables the network nodes to distributedly and collectively compute the set of first-order stationary solutions in a global sublinear manne…

Cited by 178SourcePDFScholar
2017

Scalable and flexible Max-Var generalized canonical correlation analysis via alternating optimization

ICASSP 2017accepted

Unlike dimensionality reduction (DR) tools for single-view data, e.g., principal component analysis (PCA), canonical correlation analysis (CCA) and generalized CCA (GCCA) are able to integrate information from multiple feature spaces of data. This is critical in multi-modal data fusion and analytics…

Cited by 0SourceScholar
2017

Towards K-means-friendly Spaces: Simultaneous Deep Learning and Clustering

ICML 2017poster

Most learning approaches treat dimensionality reduction (DR) and clustering separately (i.e., sequentially), but recent research has shown that optimizing the two tasks jointly can substantially improve the performance of both. The premise behind the latter genre is that the data samples are obtaine…

2017

Traffic engineering for backhaul networks with wireless link scheduling

ICASSP 2017accepted

Traffic engineering (TE) problem is a central component of the next generation cloud-based wireless networks. In this paper, we study a new resource allocation scheme for effective traffic engineering under practical constraints such as the finite buffer size at each node. To reduce the computationa…

Cited by 3SourceScholar
2016

A penalty-BSUM approach for rate optimization in full-duplex MIMO relay networks with relay processing delay

ICASSP 2016accepted

This paper studies joint source transmit beamforming and relay amplification matrix design to achieve rate maximization for full-duplex (FD) MIMO amplify-and-forward (AF) relay systems with consideration of relay processing delay (RPD). The problem is difficult to solve due mainly to the self-interf…

Cited by 0SourceScholar
2016

An Improved Convergence Analysis of Cyclic Block Coordinate Descent-type Methods for Strongly Convex Minimization

AISTATS 2016poster

The cyclic block coordinate descent-type (CBCD-type) methods have shown remarkable computational performance for solving strongly convex minimization problems. Typical applications include many popular statistical machine learning methods such as elastic-net regression, ridge penalized logistic regr…

Cited by 10SourcePDFScholar
2016

Asynchronous distributed alternating direction method of multipliers: Algorithm and convergence analysis

ICASSP 2016accepted

Alternating direction method of multipliers (ADMM) has been recognized as an efficient approach for solving many large-scale learning problems over a computer cluster. However, traditional synchronized computation does not scale well with the problem size, as the speed of the algorithm is limited by…

Cited by 0SourceScholar
2016

NESTT: A Nonconvex Primal-Dual Splitting Method for Distributed and Stochastic Optimization

NeurIPS 2016poster

We study a stochastic and distributed algorithm for nonconvex problems whose objective consists a sum $N$ nonconvex $L_i/N$-smooth functions, plus a nonsmooth regularizer. The proposed NonconvEx primal-dual SpliTTing (NESTT) algorithm splits the problem into $N$ subproblems, and utilizes an augmen…

Cited by 58SourcePDFScholar
2016

Nonnegative matrix factorization using ADMM: Algorithm and convergence analysis

ICASSP 2016accepted

The nonnegative matrix factorization (NMF) has been a popular model for a wide range of signal processing and machine learning problems. It is usually formulated as a nonconvex cost minimization problem. This work settles the convergence issue of a popular algorithm based on the alternating directio…

Cited by 0SourceScholar
2015

Combining sparse NMF with deep neural network: A new classification-based approach for speech enhancement

ICASSP 2015accepted

In this work, we consider enhancing a target speech from a single-channel noisy observation corrupted by non-stationary noises at low signal-to-noise ratios (SNRs). We take a classification-based approach, where the objective is to estimate an Ideal Binary Mask (IBM) that classifies each time-freque…

Cited by 0SourceScholar
2015

Convergence analysis of alternating direction method of multipliers for a family of nonconvex problems

ICASSP 2015accepted

In this paper, we analyze the behavior of the alternating direction method of multipliers (ADMM), for solving a family of nonconvex problems. Our focus is given to the well-known consensus and sharing problems, both of which have wide applications in signal processing. We show that in the presence o…

Cited by 0SourceScholar
2015

Improved Iteration Complexity Bounds of Cyclic Block Coordinate Descent for Convex Problems

NeurIPS 2015poster

The iteration complexity of the block-coordinate descent (BCD) type algorithm has been under extensive investigation. It was recently shown that for convex problems the classical cyclic BCGD (block coordinate gradient descent) achieves an O(1/r) complexity (r is the number of passes of all blocks).…

Cited by 60SourcePDFScholar
2015

Incorporating spatial information in binaural beamforming for noise suppression in hearing aids

ICASSP 2015accepted

In this paper, we propose a beamforming algorithm for binaural hearing aids with enhanced noise suppression capability. The enhancement is based on incorporating a priori spatial information into the conventional multichannel Wiener filtering (MWF) approach for noise suppression. We develop a low co…

Cited by 11SourceScholar
2015

Semi-asynchronous routing for large scale hierarchical networks

ICASSP 2015accepted

We consider the distributed network routing problem in a large-scale hierarchical network whereby the nodes are partitioned into subnetworks, each managed by a network controller (NC), and there is a central NC to coordinate the operation of the distributed NCs. We propose a semi-asynchronous routin…

Cited by 0SourceScholar