← Search

Akifumi Wachi

12 accepted papers

2026

Cost-Minimized Label-Flipping Poisoning Attack to LLM Alignment

AAAI 2026technical

Large language models (LLMs) are increasingly deployed in real-world systems, making it critical to understand their vulnerabilities. While data poisoning attacks during RLHF/DPO alignment have been studied empirically, their theoretical foundations remain unclear. We investigate the minimum-cost po

Cited by 0SourcePDFScholar
2026

Inference-Aware Meta-Alignment of LLMs via Non-Linear GRPO

ICML 2026poster

Aligning large language models (LLMs) to diverse human preferences is fundamentally challenging since criteria can often conflict with each other. Inference-time alignment methods have recently gained popularity as they allow LLMs to be aligned to multiple criteria via different alignment algorithms…

Cited by 0SourceScholar
2025

A Provable Approach for End-to-End Safe Reinforcement Learning

NeurIPS 2025poster

A longstanding goal in safe reinforcement learning (RL) is a method to ensure the safety of a policy throughout the entire process, from learning to operation. However, existing safe RL paradigms inherently struggle to achieve this objective. We propose a method, called Provably Lifetime Safe RL (PL…

Cited by 0SourceScholar
2025

Offline Guarded Safe Reinforcement Learning for Medical Treatment Optimization Strategies

NeurIPS 2025spotlight

When applying offline reinforcement learning (RL) in healthcare scenarios, the out-of-distribution (OOD) issues pose significant risks, as inappropriate generalization beyond clinical expertise can result in potentially harmful recommendations. While existing methods like conservative Q-learning (C…

Cited by 0SourceScholar
2024

A Survey of Constraint Formulations in Safe Reinforcement Learning

IJCAI 2024poster

Safety is critical when applying reinforcement learning (RL) to real-world problems. As a result, safe RL has emerged as a fundamental and powerful paradigm for optimizing an agent’s policy while incorporating notions of safety. A prevalent safe RL approach is based on a constrained criterion, which…

2024

Flipping-based Policy for Chance-Constrained Markov Decision Processes

NeurIPS 2024poster

Safe reinforcement learning (RL) is a promising approach for many real-world decision-making problems where ensuring safety is a critical necessity. In safe RL research, while expected cumulative safety constraints (ECSCs) are typically the first choices, chance constraints are often more pragmatic…

Cited by 1SourcePDFScholar
2024

Long-Term Safe Reinforcement Learning with Binary Feedback

AAAI 2024technical

Safety is an indispensable requirement for applying reinforcement learning (RL) to real problems. Although there has been a surge of safe RL algorithms proposed in recent years, most existing work typically 1) relies on receiving numeric safety feedback; 2) does not guarantee safety during the learn…

Cited by 3SourcePDFScholar
2024

Stepwise Alignment for Constrained Language Model Policy Optimization

NeurIPS 2024poster

Safety and trustworthiness are indispensable requirements for real-world applications of AI systems using large language models (LLMs). This paper formulates human value alignment as an optimization problem of the language model policy to maximize reward under a safety constraint, and then proposes…

2023

Safe Exploration in Reinforcement Learning: A Generalized Formulation and Algorithms

NeurIPS 2023poster

Safe exploration is essential for the practical use of reinforcement learning (RL) in many real-world scenarios. In this paper, we present a generalized safe exploration (GSE) problem as a unified formulation of common safe exploration problems. We then propose a solution of the GSE problem in the f…

Cited by 13SourcePDFScholar
2021

Neuro-Symbolic Reinforcement Learning with First-Order Logic

EMNLP 2021main

Deep reinforcement learning (RL) methods often require many trials before convergence, and no direct interpretability of trained policies is provided. In order to achieve fast convergence and interpretability for the policy in RL, we propose a novel RL method for text-based games with a recent neuro…

Cited by 48SourcePDFScholar
2021

Safe Policy Optimization with Local Generalized Linear Function Approximations

NeurIPS 2021poster

Safe exploration is a key to applying reinforcement learning (RL) in safety-critical systems. Existing safe exploration methods guaranteed safety under the assumption of regularity, and it has been difficult to apply them to large-scale real problems. We propose a novel algorithm, SPO-LF, that optim…