← Search

Stephanie Milani

6 accepted papers

2025

LICORICE: Label-Efficient Concept-Based Interpretable Reinforcement Learning

ICLR 2025poster

Recent advances in reinforcement learning (RL) have predominantly leveraged neural network policies for decision-making, yet these models often lack interpretability, posing challenges for stakeholder comprehension and trust. Concept bottleneck models offer an interpretable alternative by integratin…

Cited by 0SourcePDFScholar
2024

PATIENT-𝜓: Using Large Language Models to Simulate Patients for Training Mental Health Professionals

EMNLP 2024main

Mental illness remains one of the most critical public health issues. Despite its importance, many mental health professionals highlight a disconnect between their training and actual real-world patient practice. To help bridge this gap, we propose PATIENT-𝜓, a novel patient simulation framework for…

2024

When is Transfer Learning Possible?

ICML 2024poster

We present a general framework for transfer learning that is flexible enough to capture transfer in supervised, reinforcement, and imitation learning. Our framework enables new insights into the fundamental question of *when* we can successfully transfer learned information across problems. We model…

Cited by 0SourcePDFScholar
2023

BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks

NeurIPS 2023oral

The MineRL BASALT competition has served to catalyze advances in learning from human feedback through four hard-to-specify tasks in Minecraft, such as create and photograph a waterfall. Given the completion of two years of BASALT competitions, we offer to the community a formalized benchmark through…

2022

Uni[MASK]: Unified Inference in Sequential Decision Problems

NeurIPS 2022accept

Randomly masking and predicting word tokens has been a successful approach in pre-training language models for a variety of downstream tasks. In this work, we observe that the same idea also applies naturally to sequential decision making, where many well-studied tasks like behavior cloning, offline…

2021

Iterative Bounding MDPs: Learning Interpretable Policies via Non-Interpretable Methods

AAAI 2021technical

Current work in explainable reinforcement learning generally produces policies in the form of a decision tree over the state space. Such policies can be used for formal safety verification, agent behavior prediction, and manual inspection of important features. However, existing approaches fit a dec…

Cited by 44SourcePDFScholar