← Search

Ruochen Wang

20 accepted papers

2025

Don’t Think Longer, Think Wisely: Optimizing Thinking Dynamics for Large Reasoning Models

NeurIPS 2025poster

While recent success of large reasoning models (LRMs) significantly advanced LLMs' reasoning capability by optimizing the final answer accuracy using reinforcement learning, they may also drastically increase the output length due to *overthinking*—characterized by unnecessarily complex reasoning pa…

Cited by 0SourceScholar
2025

Is Your Multimodal Language Model Oversensitive to Safe Queries?

ICLR 2025poster

Humans are prone to cognitive distortions — biased thinking patterns that lead to exaggerated responses to specific stimuli, albeit in very different contexts. This paper demonstrates that advanced Multimodal Large Language Models (MLLMs) exhibit similar tendencies. While these models are designed t…

Cited by 0SourcePDFScholar
2025

Large Language Models are Interpretable Learners

ICLR 2025poster

The trade-off between expressiveness and interpretability remains a core challenge when building human-centric models for classification and decision-making. While symbolic rules offer interpretability, they often lack expressiveness, whereas neural networks excel in performance but are known for be…

2025

QG-CoC: Question-Guided Chain-of-Captions for Large Multimodal Models

EMNLP 2025

Recently, Multimodal Large Language Models (MLLMs) encounter two key issues in multi-image contexts: (1) a lack of fine-grained perception across disparate images, and (2) a diminished capability to effectively reason over and synthesize information from multiple visual inputs. However, while variou

Cited by 0SourcePDFScholar
2025

The Crystal Ball Hypothesis in diffusion models: Anticipating object positions from initial noise

ICLR 2025poster

Diffusion models have achieved remarkable success in text-to-image generation tasks, yet the influence of initial noise remains largely unexplored. In this study, we identify specific regions within the initial noise image, termed trigger patches, that play a key role in inducing object generation i…

Cited by 3SourcePDFScholar
2024

DrAttack: Prompt Decomposition and Reconstruction Makes Powerful LLMs Jailbreakers

EMNLP 2024finding

Safety-aligned Large Language Models (LLMs) are still vulnerable to some manual and automated jailbreak attacks, which adversarially trigger LLMs to output harmful content. However, existing jailbreaking methods usually view a harmful prompt as a whole but they are not effective at reducing LLMs’ at…

2024

One Prompt is not Enough: Automated Construction of a Mixture-of-Expert Prompts

ICML 2024poster

Large Language Models (LLMs) exhibit strong generalization capabilities to novel tasks when prompted with language instructions and in-context demos. Since this ability sensitively depends on the quality of prompts, various methods have been explored to automate the instruction design. While these m…

2024

Solving for X and Beyond: Can Large Language Models Solve Complex Math Problems with More-Than-Two Unknowns?

EMNLP 2024finding

Large Language Models have demonstrates remarkable performance in solving math problems, a hallmark of human intelligence.Despite high success rates on current benchmarks, however, these often feature simple problems with only one or two unknowns, which do not sufficiently challenge their reasoning…

2024

When and How do negative prompts take effect?

ECCV 2024poster

"The concept of negative prompts, emerging from conditional generation models like Stable Diffusion, allows users to specify what to exclude from the generated images.Despite the widespread use of negative prompts, their intrinsic mechanisms remain largely unexplored. This paper presents the first c…

2023

FedDM: Iterative Distribution Matching for Communication-Efficient Federated Learning

CVPR 2023poster

Federated learning (FL) has recently attracted increasing attention from academia and industry, with the ultimate goal of achieving collaborative training under privacy and communication constraints. Existing iterative model averaging based FL algorithms require a large number of communication round…

Cited by 122SourcePDFScholar
2023

Scaling Up Dataset Distillation to ImageNet-1K with Constant Memory

ICML 2023poster

Dataset Distillation is a newly emerging area that aims to distill large datasets into much smaller and highly informative synthetic ones to accelerate training and reduce storage. Among various dataset distillation methods, trajectory-matching-based methods (MTT) have achieved SOTA performance in m…

2022

Efficient Non-Parametric Optimizer Search for Diverse Tasks

NeurIPS 2022accept

Efficient and automated design of optimizers plays a crucial role in full-stack AutoML systems. However, prior methods in optimizer search are often limited by their scalability, generability, or sample efficiency. With the goal of democratizing research and application of optimizer search, we prese…

2022

Generalizing Few-Shot NAS with Gradient Matching

ICLR 2022poster

Efficient performance estimation of architectures drawn from large search spaces is essential to Neural Architecture Search. One-Shot methods tackle this challenge by training one supernet to approximate the performance of every architecture in the search space via weight-sharing, thereby drasticall…

2022

Learning to Schedule Learning rate with Graph Neural Networks

ICLR 2022poster

Recent decades have witnessed great development of stochastic optimization in training deep neural networks. Learning rate scheduling is one of the most important factors that influence the performance of stochastic optimizers like Adam. Traditional methods seek to find a relatively proper schedulin…

Cited by 20SourcePDFScholar
2021

DrNAS: Dirichlet Neural Architecture Search

ICLR 2021poster

This paper proposes a novel differentiable architecture search method by formulating it into a distribution learning problem. We treat the continuously relaxed architecture mixing weight as random variables, modeled by Dirichlet distribution. With recently developed pathwise derivatives, the Dirichl…

2021

RANK-NOSH: Efficient Predictor-Based Architecture Search via Non-Uniform Successive Halving

ICCV 2021poster

Predictor-based algorithms have achieved remarkable performance in the Neural Architecture Search (NAS) tasks. However, these methods suffer from high computation costs, as training the performance predictor usually requires training and evaluating hundreds of architectures from scratch. Previous wo…

Cited by 20PDFScholar
2021

Rethinking Architecture Selection in Differentiable NAS

ICLR 2021oral

Differentiable Neural Architecture Search is one of the most popular Neural Architecture Search (NAS) methods for its search efficiency and simplicity, accomplished by jointly optimizing the model weight and architecture parameters in a weight-sharing supernet via gradient-based algorithms. At the e…