← Search

Yibo Wang

39 accepted papers

2026

Convergence Analysis of the Lion Optimizer in Centralized and Distributed Settings

ICML 2026poster

In this paper, we provide a comprehensive convergence analysis for the Lion optimizer. First, we establish that the original Lion achieves a convergence rate of $\mathcal{O}(d^{1/2}T^{-1/4})$, where $d$ denotes the problem dimension and $T$ is the iteration number. To improve this rate, we propose a…

Cited by 0SourceScholar
2026

MM-DeepResearch: A Simple and Effective Multimodal Agentic Search Baseline

ICML 2026poster

We aim to develop a multimodal research agent capable of explicit reasoning and planning, multi-tool invocation, and cross-modal information synthesis, enabling it to conduct deep research tasks. However, we observe three main challenges in developing such agents: (1) scarcity of search-intensive mu…

Cited by 12SourceScholar
2026

MarCon: Max-Margin Contrastive Learning for Imbalanced Domain Adaptation Semantic Segmentation

IJCAI 2026

Unsupervised Domain Adaptation for Semantic Segmentation (UDA-SS) has seen significant progress in recent years. Existing UDA-SS approaches mostly adopt a pseudo-labeling schema to adapt model in the target domain, but they often overlook the inherent long-tailed data distribution in segmentation. W

Cited by 0Scholar
2026

RECODE: A Benchmark for Research Code DEvelopment with Interactive Human Feedback

ICLR 2026poster

Large language models (LLMs) show the promise in supporting scientific research implementation, yet their ability to generate correct and executable code remains limited. Existing works largely adopt one-shot settings, ignoring the iterative and feedback-driven nature of realistic workflows of scien…

Cited by 0SourcecodeScholar
2026

UltraHorizon: Benchmarking LLM-Agent Capabilities in Ultra Long-Horizon Scenarios

ICML 2026poster

Autonomous agents have recently achieved remarkable progress across diverse domains, yet most evaluations focus on short-horizon, fully observable tasks. In contrast, many critical real-world tasks, such as large-scale software development, commercial investment, and scientific discovery, unfold in …

Cited by 0SourceScholar
2025

Ada-R1: Hybrid-CoT via Bi-Level Adaptive Reasoning Optimization

NeurIPS 2025poster

Recently, long-thought reasoning models achieve strong performance on complex reasoning tasks, but often incur substantial inference overhead, making efficiency a critical concern. Our empirical analysis reveals that the benefit of using Long-CoT varies across problems: while some problems require e…

Cited by 0SourceScholar
2025

Dimension-Free Adaptive Subgradient Methods with Frequent Directions

ICML 2025poster

In this paper, we investigate the acceleration of adaptive subgradient methods through frequent directions (FD), a widely-used matrix sketching technique. The state-of-the-art regret bound exhibits a _linear_ dependence on the dimensionality $d$, leading to unsatisfactory guarantees for high-dimensi…

Cited by 0SourcePDFScholar
2025

KGARevion: An AI Agent for Knowledge-Intensive Biomedical QA

ICLR 2025poster

Biomedical reasoning integrates structured, codified knowledge with tacit, experience-driven insights. Depending on the context, quantity, and nature of available evidence, researchers and clinicians use diverse strategies, including rule-based, prototype-based, and case-based reasoning. Effective m…

Cited by 0SourcePDFScholar
2025

Mulberry: Empowering MLLM with o1-like Reasoning and Reflection via Collective Monte Carlo Tree Search

NeurIPS 2025spotlight

In this work, we aim to develop an MLLM that understands and solves questions by learning to create each intermediate step of the reasoning involved till the final answer. To this end, we propose Collective Monte Carlo Tree Search (CoMCTS), a new learning-to-reason method for MLLMs, which introduces…

Cited by 0SourcecodeScholar
2025

Panacea: Mitigating Harmful Fine-tuning for Large Language Models via Post-fine-tuning Perturbation

NeurIPS 2025poster

Harmful fine-tuning attack introduces significant security risks to the fine-tuning services. Main-stream defenses aim to vaccinate the model such that the later harmful fine-tuning attack is less effective. However, our evaluation results show that such defenses are fragile-- with a few fine-tuning…

Cited by 0SourcecodeScholar
2025

R1-ShareVL: Incentivizing Reasoning Capabilities of Multimodal Large Language Models via Share-GRPO

NeurIPS 2025poster

In this work, we aim to incentivize the reasoning ability of Multimodal Large Language Models (MLLMs) via reinforcement learning (RL) and develop an effective approach that mitigates the sparse reward and advantage vanishing issues during RL. To this end, we propose Share-GRPO, a novel RL approach…

Cited by 0SourcecodeScholar
2025

SPACE: Noise Contrastive Estimation Stabilizes Self-Play Fine-Tuning for Large Language Models

NeurIPS 2025poster

Self-play fine-tuning has demonstrated promising abilities in adapting large language models (LLMs) to downstream tasks with limited real-world data. The basic principle is to iteratively refine the model with real samples and synthetic ones generated from itself. However, the existing methods prima…

Cited by 0SourceScholar
2025

TestNUC: Enhancing Test-Time Computing Approaches and Scaling through Neighboring Unlabeled Data Consistency

ACL 2025long

Test-time computing approaches, which leverage additional computational resources during inference, have been proven effective in enhancing large language model performance. This work introduces a novel, linearly scaling approach, TestNUC, that improves test-time predictions by leveraging the local…

2025

Towards Unbiased Information Extraction and Adaptation in Cross-Domain Recommendation

AAAI 2025technical

Cross-Domain Recommendation (CDR) leverages additional knowledge from auxiliary domains to address the long-standing data sparsity issue. However, existing methods typically acquire this knowledge by minimizing the average loss over all domains, overlooking the fact that different domains possess di…

Cited by 0SourcePDFScholar
2025

Triplets Better Than Pairs: Towards Stable and Effective Self-Play Fine-Tuning for LLMs

NeurIPS 2025poster

Recently, self-play fine-tuning (SPIN) has been proposed to adapt large language models to downstream applications with scarce expert-annotated data, by iteratively generating synthetic responses from the model itself. However, SPIN is designed to optimize the current reward advantages of annotated…

Cited by 0SourceScholar
2024

Adaptive Variance Reduction for Stochastic Optimization under Weaker Assumptions

NeurIPS 2024poster

This paper explores adaptive variance reduction methods for stochastic optimization based on the STORM technique. Existing adaptive extensions of STORM rely on strong assumptions like bounded gradients and bounded function values, or suffer an additional $\mathcal{O}(\log T)$ term in the convergence…

Cited by 1SourcePDFScholar
2024

Advancing Tool-Augmented Large Language Models: Integrating Insights from Errors in Inference Trees

NeurIPS 2024poster

Tool-augmented large language models (LLMs) leverage tools, often in the form of APIs, to improve their reasoning capabilities on complex tasks. This enables them to act as intelligent agents interacting with the real world. The recently introduced ToolLLaMA model by Qin et al. [2023] utilizes the d…

Cited by 7SourcePDFScholar
2024

DA3: A Distribution-Aware Adversarial Attack against Language Models

EMNLP 2024main

Language models can be manipulated by adversarial attacks, which introduce subtle perturbations to input data. While recent attack methods can achieve a relatively high attack success rate (ASR), we’ve observed that the generated adversarial examples have a different data distribution compared with…

2024

DetDiffusion: Synergizing Generative and Perceptive Models for Enhanced Data Generation and Perception

CVPR 2024poster

Current perceptive models heavily depend on resource-intensive datasets prompting the need for innovative solutions. Leveraging recent advances in diffusion models synthetic data by constructing image inputs from various annotations proves beneficial for downstream tasks. While prior methods have se…

Cited by 26SourcePDFScholar
2024

GeoDiffusion: Text-Prompted Geometric Control for Object Detection Data Generation

ICLR 2024poster

Diffusion models have attracted significant attention due to the remarkable ability to create content and generate data for tasks like image classification. However, the usage of diffusion models to generate the high-quality object detection data remains an underexplored area, where not only image-l…

Cited by 24SourcePDFScholar
2024

High-Probability Bound for Non-Smooth Non-Convex Stochastic Optimization with Heavy Tails

ICML 2024poster

Recently, Cutkosky et al. introduce the online-to-non-convex framework, which utilizes online learning methods to solve non-smooth non-convex optimization problems, and achieves an $\mathcal{O}(\epsilon^{-3}\delta^{-1})$ gradient complexity for finding $(\delta,\epsilon)$-stationary points. However,…

Cited by 3SourcePDFScholar
2024

Joint RGB-Spectral Decomposition Model Guided Image Enhancement in Mobile Photography

ECCV 2024poster

"The integration of miniaturized spectrometers into mobile devices offers new avenues for image quality enhancement and facilitates novel downstream tasks. However, the broader application of spectral sensors in mobile photography is hindered by the inherent complexity of spectral images and the con…

2024

LLMs Assist NLP Researchers: Critique Paper (Meta-)Reviewing

EMNLP 2024main

Claim: This work is not advocating the use of LLMs for paper (meta-)reviewing. Instead, wepresent a comparative analysis to identify and distinguish LLM activities from human activities. Two research goals: i) Enable better recognition of instances when someone implicitly uses LLMs for reviewing act…

2024

Non-stationary Projection-Free Online Learning with Dynamic and Adaptive Regret Guarantees

AAAI 2024technical

Projection-free online learning has drawn increasing interest due to its efficiency in solving high-dimensional problems with complicated constraints. However, most existing projection-free online methods focus on minimizing the static regret, which unfortunately fails to capture the challenge of ch…

Cited by 12SourcePDFScholar
2024

Online Composite Optimization Between Stochastic and Adversarial Environments

NeurIPS 2024poster

We study online composite optimization under the Stochastically Extended Adversarial (SEA) model. Specifically, each loss function consists of two parts: a fixed non-smooth and convex regularizer, and a time-varying function which can be chosen either stochastically, adversarially, or in a manner th…

Cited by 4SourcePDFScholar
2024

Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimization

ICML 2024poster

This paper investigates projection-free algorithms for stochastic constrained multi-level optimization. In this context, the objective function is a nested composition of several smooth functions, and the decision set is closed and convex. Existing projection-free algorithms for solving this problem…

Cited by 2SourcePDFScholar
2024

Small-loss Adaptive Regret for Online Convex Optimization

ICML 2024poster

To deal with changing environments, adaptive regret has been proposed to minimize the regret over every interval. Previous studies have established a small-loss adaptive regret bound for general convex functions under the smoothness condition, offering the advantage of being much tighter than minima…

Cited by 3SourcePDFScholar
2024

Universal Online Convex Optimization with $1$ Projection per Round

NeurIPS 2024poster

To address the uncertainty in function types, recent progress in online convex optimization (OCO) has spurred the development of universal algorithms that simultaneously attain minimax rates for multiple types of convex functions. However, for a $T$-round online problem, state-of-the-art methods typ…

Cited by 4SourcePDFScholar
2024

kNN-ICL: Compositional Task-Oriented Parsing Generalization with Nearest Neighbor In-Context Learning

NAACL 2024long

Task-Oriented Parsing (TOP) enables conversational assistants to interpret user commands expressed in natural language, transforming them into structured outputs that combine elements of both natural language and intent/slot tags. Recently, Large Language Models (LLMs) have achieved impressive perfo…

Cited by 4SourcePDFScholar
2023

Aperture Diffraction for Compact Snapshot Spectral Imaging

ICCV 2023poster

We demonstrate a compact, cost-effective snapshot spectral imaging system named Aperture Diffraction Imaging Spectrometer (ADIS), which consists only of an imaging lens with an ultra-thin orthogonal aperture mask and a mosaic filter sensor, requiring no additional physical footprint compared to comm…

Cited by 4PDFcodeScholar
2023

Distributed Projection-Free Online Learning for Smooth and Convex Losses

AAAI 2023technical

We investigate the problem of distributed online convex optimization with complicated constraints, in which the projection operation could be the computational bottleneck. To avoid projections, distributed online projection-free methods have been proposed and attain an O(T^{3/4}) regret bound for ge…

Cited by 11SourcePDFScholar
2022

Explore Spatio-Temporal Aggregation for Insubstantial Object Detection: Benchmark Dataset and Baseline

CVPR 2022poster

We endeavor on a rarely explored task named Insubstan-tial Object Detection (IOD), which aims to localize the object with following characteristics: (1) amorphous shape with indistinct boundary; (2) similarity to surroundings; (3) absence in color. Accordingly, it is far more challenging to distingu…

Cited by 23PDFcodeScholar
2022

Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimization

NeurIPS 2022accept

Variance reduction techniques such as SPIDER/SARAH/STORM have been extensively studied to improve the convergence rates of stochastic non-convex optimization, which usually maintain and update a sequence of estimators for a single function across iterations. What if we need to track multiple functi…

Cited by 22SourcePDFScholar
2022

Optimal Algorithms for Stochastic Multi-Level Compositional Optimization

ICML 2022spotlight

In this paper, we investigate the problem of stochastic multi-level compositional optimization, where the objective function is a composition of multiple smooth but possibly non-convex functions. Existing methods for solving this problem either suffer from sub-optimal sample complexities or need a h…

Cited by 21SourcePDFScholar