← Search

Chaochao Lu

36 accepted papers

2026

Advancing LLM Reasoning with Natural Language and Numerical Feedback

ICML 2026spotlight

Recent advances in reinforcement learning (RL) using numerical rewards have significantly enhanced the complex reasoning capabilities of large language models (LLMs). However, we identify three fundamental limitations of purely numerical feedback: performance plateaus, ineffective spontaneous self-r…

Cited by 0SourceScholar
2026

Causal-Adapter: Taming Text-to-Image Diffusion for Faithful Counterfactual Generation

ICML 2026poster

We present Causal-Adapter, a modular framework that adapts frozen text-to-image diffusion backbones for counterfactual image generation. Our method enables causal interventions on target attributes while preserving all other aspects of the image, including the core identity. In contrast to prior app…

Cited by 0SourceScholar
2026

DEPO: Dual-Efficiency Preference Optimization for LLM Agents

AAAI 2026technical

Recent advances in large language models (LLMs) have greatly improved their reasoning and decision-making abilities when deployed as agents. Richer reasoning, however, often comes at the cost of longer chain of thought (CoT), hampering interaction efficiency in real-world scenarios. Nevertheless, th

Cited by 0SourcePDFScholar
2026

Internalizing Safety Understanding in Large Reasoning Models via Verification

ICML 2026poster

While explicit Chain-of-Thought (CoT) empowers large reasoning models (LRMs), it enables the generation of riskier final answers. Current alignment paradigms primarily rely on externally enforced compliance, optimizing models to detect malicious prompts rather than evaluating the safety of their own…

Cited by 0SourceScholar
2026

MAGIC: A Co-Evolving Attacker–Defender Adversarial Game for Robust LLM Safety

ICML 2026poster

Ensuring robust safety alignment is crucial for Large Language Models (LLMs), yet existing defenses often lag behind evolving adversarial attacks due to their \textbf{reliance on static, pre-collected data distributions}. In this paper, we introduce \textbf{MAGIC}, a novel multi-turn multi-agent rei…

Cited by 0SourceScholar
2026

Native Reasoning Models: Training Language Models to Reason on Unverifiable Data

ICLR 2026poster

The dominant paradigm for training large reasoning models—combining Supervised Fine-Tuning (SFT) with Reinforcement Learning with Verifiable Rewards (RLVR)—is fundamentally constrained by its reliance on high-quality, human-annotated reasoning data and external verifiers. This dependency incurs sign…

Cited by 0SourceScholar
2026

Position: Preparing for AI Systems That Deceive Developers

ICML 2026poster

AI systems may exhibit deceptive behaviors that mislead developers about their capabilities, propensities, or actions. Such deception can take distinct forms across the development lifecycle: training subversion, evaluation gaming, and control evasion. We argue that the AI community should prioritiz…

Cited by 0SourceScholar
2026

Position: Safe AI Should be Resistant and Resilient in an Evolving World

ICML 2026poster

In this position paper, we address the persistent gap between rapidly growing AI capabilities and lagging safety progress. Existing paradigms divide into "Make AI Safe", which applies post-hoc alignment and guardrails but remains brittle and reactive, and "Make Safe AI", which emphasizes intrinsic s…

Cited by 0SourceScholar
2026

Reflector: Internalizing Step-wise Reflection against Indirect Jailbreaks

ICML 2026poster

While Large Language Models (LLMs) demonstrate remarkable capabilities, they remain susceptible to sophisticated, multi-step jailbreak attacks that circumvent conventional surface-level safety alignment by exploiting the internal generation process. To address these vulnerabilities, we propose Refle…

Cited by 0SourceScholar
2026

The Devil behind the mask: An emergent safety vulnerability of Diffusion LLMs

ICLR 2026poster

Diffusion-based large language models (dLLMs) have recently emerged as a powerful alternative to autoregressive LLMs, offering faster inference and greater interactivity via parallel decoding and bidirectional modeling. However, despite strong performance in code generation and text infilling, we i…

Cited by 0SourcecodeScholar
2026

UniCoD: Enhancing Robot Policy via Unified Continuous and Discrete Representation Learning

ICML 2026poster

Building generalist robot policies that can handle diverse tasks in open-ended environments is a central challenge in robotics. To leverage knowledge from large-scale pretraining, prior work (VLA) has typically built generalist policies either on top of vision-language models (VLMs) or generative mo…

Cited by 0SourceScholar
2026

VAST: Video Ability-Stratified Taxonomy for Data-Efficient Video Reasoning

CVPR 2026

Reinforcement learning (RL) has emerged as an effective approach for improving video reasoning in multimodal large language models (MLLMs). However, existing methods remain inefficient for two reasons. First, training data are typically organized by task formats rather than underlying reasoning abil

Cited by 0SourcecodeScholar
2025

ARise: Towards Knowledge-Augmented Reasoning via Risk-Adaptive Search

ACL 2025long

Large language models (LLMs) have demonstrated impressive capabilities and are receiving increasing attention to enhance their reasoning through scaling test-time compute. However, their application in open-ended, knowledge-intensive, complex reasoning scenarios is still limited. Reasoning-oriented…

2025

Adversarial Preference Learning for Robust LLM Alignment

ACL 2025finding

Modern language models often rely on Reinforcement Learning from Human Feedback (RLHF) to encourage safe behaviors. However, they remain vulnerable to adversarial attacks due to three key limitations: (1) the inefficiency and high cost of human annotation, (2) the vast diversity of potential adversa…

2025

Beyond Surface Structure: A Causal Assessment of LLMs' Comprehension ability

ICLR 2025poster

Large language models (LLMs) have shown remarkable capability in natural language tasks, yet debate persists on whether they truly comprehend deep structure (i.e., core semantics) or merely rely on surface structure (e.g., presentation format). Prior studies observe that LLMs' performance declines w…

2025

Can Diffusion Models Learn Hidden Inter-Feature Rules Behind Images?

ICML 2025poster

Despite the remarkable success of diffusion models (DMs) in data generation, they exhibit specific failure cases with unsatisfactory outputs. We focus on one such limitation: the ability of DMs to learn hidden rules between image features. Specifically, for image data with dependent features ($\math…

Cited by 1SourcePDFScholar
2025

From Imitation to Introspection: Probing Self-Consciousness in Language Models

ACL 2025finding

Self-consciousness, the introspection of one’s existence and thoughts, represents a high-level cognitive process. As language models advance at an unprecedented pace, a critical question arises: Are these models becoming self-conscious? Drawing upon insights from psychological and neural science, th…

2025

IP-Dialog: Evaluating Implicit Personalization in Dialogue Systems with Synthetic Data

EMNLP 2025

In modern dialogue systems, the ability to implicitly infer user backgrounds from conversations and leverage this information for personalized assistance is crucial. However, the scarcity of high-quality data remains a fundamental challenge to evaluating and improving this capability. Traditional da

2025

Video Prediction Policy: A Generalist Robot Policy with Predictive Visual Representations

ICML 2025spotlight

Visual representations play a crucial role in developing generalist robotic policies. Previous vision encoders, typically pre-trained with single-image reconstruction or two-image contrastive learning, tend to capture static information, often neglecting the dynamic aspects vital for embodied tasks.…

2024

ACAMDA: Improving Data Efficiency in Reinforcement Learning through Guided Counterfactual Data Augmentation

AAAI 2024technical

Data augmentation plays a crucial role in improving the data efficiency of reinforcement learning (RL). However, the generation of high-quality augmented data remains a significant challenge. To overcome this, we introduce ACAMDA (Adversarial Causal Modeling for Data Augmentation), a novel framework…

Cited by 6SourcePDFScholar
2024

CLEAR: Can Language Models Really Understand Causal Graphs?

EMNLP 2024finding

Causal reasoning is a cornerstone of how humans interpret the world. To model and reason about causality, causal graphs offer a concise yet effective solution. Given the impressive advancements in language models, a crucial question arises: can they really understand causal graphs? To this end, we p…

2024

ConditionVideo: Training-Free Condition-Guided Video Generation

AAAI 2024technical

Recent works have successfully extended large-scale text-to-image models to the video domain, producing promising results but at a high computational cost and requiring a large amount of video data. In this work, we introduce ConditionVideo, a training-free approach to text-to-video generation based…

Cited by 13SourcePDFScholar
2024

Prediction with Action: Visual Policy Learning via Joint Denoising Process

NeurIPS 2024poster

Diffusion models have demonstrated remarkable capabilities in image generation tasks, including image editing and video creation, representing a good understanding of the physical world. On the other line, diffusion models have also shown promise in robotic control tasks by denoising actions, known…

Cited by 4SourcePDFScholar
2024

Quantifying and Mitigating Unimodal Biases in Multimodal Large Language Models: A Causal Perspective

EMNLP 2024finding

Recent advancements in Large Language Models (LLMs) have facilitated the development of Multimodal LLMs (MLLMs). Despite their impressive capabilities, MLLMs often suffer from over-reliance on unimodal biases (e.g., language bias and vision bias), leading to incorrect answers in complex multimodal t…

2023

Few-Shot Composition Learning for Image Retrieval with Prompt Tuning

AAAI 2023technical

We study the problem of composition learning for image retrieval, for which we learn to retrieve target images with search queries in the form of a composition of a reference image and a modification text that describes desired modifications of the image. Existing models of composition learning for…

Cited by 10SourcePDFScholar
2023

InfoPrompt: Information-Theoretic Soft Prompt Tuning for Natural Language Understanding

NeurIPS 2023poster

Soft prompt tuning achieves superior performances across a wide range of few-shot tasks. However, the performances of prompt tuning can be highly sensitive to the initialization of the prompts. We have also empirically observed that conventional prompt tuning methods cannot encode and learn sufficie…

Cited by 31SourcePDFScholar
2022

Action-Sufficient State Representation Learning for Control with Structural Constraints

ICML 2022spotlight

Perceived signals in real-world scenarios are usually high-dimensional and noisy, and finding and using their representation that contains essential and sufficient information required by downstream decision-making tasks will help improve computational efficiency and generalization ability in the ta…

Cited by 48SourcePDFScholar
2022

AdaRL: What, Where, and How to Adapt in Transfer Reinforcement Learning

ICLR 2022spotlight

One practical challenge in reinforcement learning (RL) is how to make quick adaptations when faced with new environments. In this paper, we propose a principled framework for adaptive RL, called AdaRL, that adapts reliably and efficiently to changes across domains with a few samples from the target…

2022

Invariant Causal Representation Learning for Out-of-Distribution Generalization

ICLR 2022poster

Due to spurious correlations, machine learning systems often fail to generalize to environments whose distributions differ from the ones used at training time. Prior work addressing this, either explicitly or implicitly, attempted to find a data representation that has an invariant relationship with…

Cited by 116SourcePDFScholar