← Search

Juntao Dai

14 accepted papers

2026

Align Once, Benefit Multilingually: Enforcing Multilingual Consistency for LLM Safety Alignment

ICLR 2026poster

The widespread deployment of large language models (LLMs) across linguistic communities necessitates reliable multilingual safety alignment. However, recent efforts to extend alignment to other languages often require substantial resources, either through large-scale, high-quality supervision in the…

Cited by 0SourceScholar
2026

Debate with Images: Detecting Deceptive Behaviors in Multimodal Large Language Models

ICML 2026poster

As frontier AI systems become increasingly capable, concerns about deceptive behaviors have intensified. Unlike hallucinations, which stem from capability limitations, deception involves strategically misleading responses despite correct internal representations. While prior work has primarily studi…

Cited by 0SourceScholar
2026

VLA-Arena: An Open-Source Framework for Benchmarking Vision-Language-Action Models

ICML 2026poster

While Vision-Language-Action models (VLAs) are rapidly advancing toward generalist robot policies, quantitatively characterizing their capability boundaries and failure modes remains challenging. To address this, we introduce **VLA-Arena**, a comprehensive benchmark. It features a novel structured t…

Cited by 0SourceScholar
2026

What, Whether and How? Unveiling Process Reward Models for Thinking with Images Reasoning

AAAI 2026technical

The rapid advancement of Large Vision Language Models (LVLMs) has demonstrated excellent abilities in various visual tasks. Building upon these developments, the thinking with images paradigm has emerged, enabling models to dynamically edit and re-encode visual information at each reasoning step, mi

Cited by 0SourcePDFScholar
2025

Automate Strategy Finding with LLM in Quant Investment

EMNLP 2025

We present a novel three-stage framework leveraging Large Language Models (LLMs) within a risk-aware multi-agent system for automate strategy finding in quantitative finance. Our approach addresses the brittleness of traditional deep learning models in financial applications by: employing prompt-eng

Cited by 0SourcePDFScholar
2025

Mitigating Reward Over-Optimization in RLHF via Behavior-Supported Regularization

ICLR 2025poster

Reinforcement learning from human feedback (RLHF) is an effective method for aligning large language models (LLMs) with human values. However, reward over-optimization remains an open challenge leading to discrepancies between the performance of LLMs under the reward model and the true human objecti…

Cited by 0SourcePDFScholar
2025

Safe RLHF-V: Safe Reinforcement Learning from Multi-modal Human Feedback

NeurIPS 2025poster

Multimodal large language models (MLLMs) are essential for building general-purpose AI assistants; however, they pose increasing safety risks. How can we ensure safety alignment of MLLMs to prevent undesired behaviors? Going further, it is critical to explore how to fine-tune MLLMs to preserve capab…

Cited by 0SourceScholar
2025

Towards Advanced Mathematical Reasoning for LLMs via First-Order Logic Theorem Proving

EMNLP 2025

Large language models (LLMs) have shown promising first-order logic (FOL) reasoning capabilities with applications in various areas. However, their effectiveness in complex mathematical reasoning involving multi-step FOL deductions is still under-researched. While LLMs perform competitively on estab

2024

Aligner: Efficient Alignment by Learning to Correct

NeurIPS 2024oral

With the rapid development of large language models (LLMs) and ever-evolving practical requirements, finding an efficient and effective alignment method has never been more critical. However, the tension between the complexity of current alignment methods and the need for rapid iteration in deployme…

Cited by 20SourcePDFScholar
2024

Safe Reinforcement Learning using Finite-Horizon Gradient-based Estimation

ICML 2024poster

A key aspect of Safe Reinforcement Learning (Safe RL) involves estimating the constraint condition for the next policy, which is crucial for guiding the optimization of safe policy updates. However, the existing *Advantage-based Estimation* (ABE) method relies on the infinite-horizon discounted adva…

Cited by 1SourcePDFScholar
2023

Augmented Proximal Policy Optimization for Safe Reinforcement Learning

AAAI 2023technical

Safe reinforcement learning considers practical scenarios that maximize the return while satisfying safety constraints. Current algorithms, which suffer from training oscillations or approximation errors, still struggle to update the policy efficiently with precise constraint satisfaction. In this a…

Cited by 21SourcePDFScholar
2023

BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Dataset

NeurIPS 2023poster

In this paper, we introduce the BeaverTails dataset, aimed at fostering research on safety alignment in large language models (LLMs). This dataset uniquely separates annotations of helpfulness and harmlessness for question-answering pairs, thus offering distinct perspectives on these crucial attribu…

Cited by 400SourcePDFScholar
2022

Constrained Update Projection Approach to Safe Policy Optimization

NeurIPS 2022accept

Safe reinforcement learning (RL) studies problems where an intelligent agent has to not only maximize reward but also avoid exploring unsafe areas. In this study, we propose CUP, a novel policy optimization method based on Constrained Update Projection framework that enjoys rigorous safety guarantee…