← Search

Yiheng Lin

16 accepted papers

2026

Matting Anything 2: Towards Video Matting for Anything

ICLR 2026poster

Video matting is a crucial task for many applications, but existing methods face significant limitations. They are often domain-specific, focusing primarily on human portraits, and rely on the mask of first frame that is challenging to acquire for transparent or intricate objects like fire or smoke.…

Cited by 0SourceScholar
2026

ThinkGen: Generalized Thinking for Visual Generation

CVPR 2026

Recent progress in Multimodal Large Language Models (MLLMs) demonstrates that Chain-of-Thought (CoT) reasoning enables systematic solutions to complex understanding tasks. However, its extension to generation tasks remains nascent and limited by scenario-specific mechanisms that hinder generalizatio

Cited by 0SourcecodeScholar
2025

Approximate Global Convergence of Independent Learning in Multi-Agent Systems

AISTATS 2025poster

Independent learning (IL) is a popular approach for achieving scalability in large-scale multi-agent systems, yet it typically lacks global convergence guarantees. In this paper, we study two representative algorithms—independent $Q$-learning and independent natural actor-critic—within both value-ba…

Cited by 0SourceScholar
2025

Maximizing the Value of Predictions in Control: Accuracy Is Not Enough

NeurIPS 2025poster

We study the value of stochastic predictions in online optimal control with random disturbances. Prior work provides performance guarantees based on prediction error but ignores the stochastic dependence between predictions and disturbances. We introduce a general framework modeling their joint dist…

Cited by 0SourcecodeScholar
2025

Memory Efficient Matting with Adaptive Token Routing

AAAI 2025technical

Transformer-based models have recently achieved outstanding performance in image matting. However, their application to high-resolution images remains challenging due to the quadratic complexity of global self-attention. To address this issue, we propose MEMatte, a memory-efficient matting framework…

2024

Diffusion for Natural Image Matting

ECCV 2024poster

"Existing natural image matting algorithms inevitably have flaws in their predictions on difficult cases, and their one-step prediction manner cannot further correct these errors. In this paper, we investigate a multi-step iterative approach for the first time to tackle the challenging natural image…

2023

Beyond Black-Box Advice: Learning-Augmented Algorithms for MDPs with Q-Value Predictions

NeurIPS 2023poster

We study the tradeoff between consistency and robustness in the context of a single-trajectory time-varying Markov Decision Process (MDP) with untrusted machine-learned advice. Our work departs from the typical approach of treating advice as coming from black-box sources by instead considering a set…

Cited by 4SourcePDFScholar
2023

Convergence rates for localized actor-critic in networked Markov potential games

UAI 2023poster

We introduce a class of networked Markov potential games where agents are associated with nodes in a network. Each agent has its own local potential function, and the reward of each agent depends only on the states and actions of agents within a neighborhood. In this context, we propose a localized…

2023

Online Adaptive Policy Selection in Time-Varying Systems: No-Regret via Contractive Perturbations

NeurIPS 2023poster

We study online adaptive policy selection in systems with time-varying costs and dynamics. We develop the Gradient-based Adaptive Policy Selection (GAPS) algorithm together with a general analytical framework for online policy selection via online optimization. Under our proposed notion of contracti…

Cited by 16SourcePDFScholar
2022

Bounded-Regret MPC via Perturbation Analysis: Prediction Error, Constraints, and Nonlinearity

NeurIPS 2022accept

We study Model Predictive Control (MPC) and propose a general analysis pipeline to bound its dynamic regret. The pipeline first requires deriving a perturbation bound for a finite-time optimal control problem. Then, the perturbation bound is used to bound the per-step error of MPC, which leads to a…

Cited by 16SourcePDFScholar
2022

Decentralized Online Convex Optimization in Networked Systems

ICML 2022spotlight

We study the problem of networked online convex optimization, where each agent individually decides on an action at every time step and agents cooperatively seek to minimize the total global cost over a finite horizon. The global cost is made up of three types of local costs: convex node costs, temp…

Cited by 11SourcePDFScholar
2021

Multi-Agent Reinforcement Learning in Stochastic Networked Systems

NeurIPS 2021poster

We study multi-agent reinforcement learning (MARL) in a stochastic network of agents. The objective is to find localized policies that maximize the (discounted) global reward. In general, scalability is a challenge in this setting because the size of the global state/action space can be exponential…

Cited by 51SourcePDFScholar
2021

Perturbation-based Regret Analysis of Predictive Control in Linear Time Varying Systems

NeurIPS 2021spotlight

We study predictive control in a setting where the dynamics are time-varying and linear, and the costs are time-varying and well-conditioned. At each time step, the controller receives the exact predictions of costs, dynamics, and disturbances for the future $k$ time steps. We show that when the pre…

Cited by 46SourcePDFScholar
2020

Online Optimization with Memory and Competitive Control

NeurIPS 2020poster

This paper presents competitive algorithms for a novel class of online optimization problems with memory. We consider a setting where the learner seeks to minimize the sum of a hitting cost and a switching cost that depends on the previous $p$ decisions. This setting generalizes Smoothed Online Conv…

Cited by 63SourcePDFScholar
2020

Scalable Multi-Agent Reinforcement Learning for Networked Systems with Average Reward

NeurIPS 2020poster

It has long been recognized that multi-agent reinforcement learning (MARL) faces significant scalability issues due to the fact that the size of the state and action spaces are exponentially large in the number of agents. In this paper, we identify a rich class of networked MARL problems where the m…

Cited by 92SourcePDFScholar
2019

Beyond Online Balanced Descent: An Optimal Algorithm for Smoothed Online Optimization

NeurIPS 2019spotlight

We study online convex optimization in a setting where the learner seeks to minimize the sum of a per-round hitting cost and a movement cost which is incurred when changing decisions between rounds. We prove a new lower bound on the competitive ratio of any online algorithm in the setting where the…

Cited by 79SourcePDFScholar