← Search

Mehdi Fatemi

10 accepted papers

2024

Successor Features for Efficient Multi-Subject Controlled Text Generation

ICML 2024poster

While large language models (LLMs) have achieved impressive performance in generating fluent and realistic text, controlling the generated text so that it exhibits properties such as safety, factuality, and non-toxicity remains challenging. Existing decoding-based controllable text generation method…

Cited by 1SourcePDFScholar
2023

Systematic Rectification of Language Models via Dead-end Analysis

ICLR 2023poster

With adversarial or otherwise normal prompts, existing large language models (LLM) can be pushed to generate toxic discourses. One way to reduce the risk of LLMs generating undesired discourses is to alter the training of the LLM. This can be very restrictive due to demanding computation requirement…

2021

Learning to Represent Action Values as a Hypergraph on the Action Vertices

ICLR 2021poster

Action-value estimation is a critical component of many reinforcement learning (RL) methods whereby sample complexity relies heavily on how fast a good estimator for action value can be learned. By viewing this problem through the lens of representation learning, good representations of both state a…

2021

Medical Dead-ends and Learning to Identify High-Risk States and Treatments

NeurIPS 2021poster

Machine learning has successfully framed many sequential decision making problems as either supervised prediction, or optimal decision-making policy identification via reinforcement learning. In data-constrained offline settings, both approaches may fail as they assume fully optimal behavior or rely…

2021

Shortest-Path Constrained Reinforcement Learning for Sparse Reward Tasks

ICML 2021spotlight

We propose the k-Shortest-Path (k-SP) constraint: a novel constraint on the agent’s trajectory that improves the sample efficiency in sparse-reward MDPs. We show that any optimal policy necessarily satisfies the k-SP constraint. Notably, the k-SP constraint prevents the policy from exploring state-a…

2019

Dead-ends and Secure Exploration in Reinforcement Learning

ICML 2019oral

Many interesting applications of reinforcement learning (RL) involve MDPs that include numerous “dead-end" states. Upon reaching a dead-end state, the agent continues to interact with the environment in a dead-end trajectory before reaching an undesired terminal state, regardless of whatever actions…

2019

Using a Logarithmic Mapping to Enable Lower Discount Factors in Reinforcement Learning

NeurIPS 2019oral

In an effort to better understand the different ways in which the discount factor affects the optimization process in reinforcement learning, we designed a set of experiments to study each effect in isolation. Our analysis reveals that the common perception that poor performance of low discount fact…

2017

Hybrid Reward Architecture for Reinforcement Learning

NeurIPS 2017poster

One of the main challenges in reinforcement learning (RL) is generalisation. In typical deep RL methods this is achieved by approximating the optimal value function with a low-dimensional representation using a deep network. While this approach works well in many domains, in domains where the optim…

Cited by 331SourcePDFScholar