Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Mehdi Jafarnia-Jahromi

dblp:232/3029 · DBLP profile ↗
← Back
8ranked-venue papers
4as first author
5since 2021 · last 2024
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 8 · 4 first-author · 5 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
2 papers
Reinforcement learning · 71% Planning, search and constraint satisfaction · 14% Learning theory · 14%

Topics — the 6 heaviest of 7, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Reinforcement learning
model-free reinforcement learning
0.922021
Implicit Finite-Horizon Approximation and Efficient Optimal Algorithms for Stochastic Shortest Path · NeurIPS 2021
Model-free Reinforcement Learning in Infinite-horizon Average-reward Markov Decision Processes · ICML 2020
Machine learning › Reinforcement learning
regret minimization
0.622021
Implicit Finite-Horizon Approximation and Efficient Optimal Algorithms for Stochastic Shortest Path · NeurIPS 2021
Model-free Reinforcement Learning in Infinite-horizon Average-reward Markov Decision Processes · ICML 2020
Machine learning › Learning theory › online learning › regret bounds
minimax regret
0.512021
Implicit Finite-Horizon Approximation and Efficient Optimal Algorithms for Stochastic Shortest Path · NeurIPS 2021
Machine learning › Reinforcement learning
model-based reinforcement learning
0.512021
Implicit Finite-Horizon Approximation and Efficient Optimal Algorithms for Stochastic Shortest Path · NeurIPS 2021
Knowledge, reasoning and agents › Planning, search and constraint satisfaction › planning under uncertainty
stochastic shortest path
0.512021
Implicit Finite-Horizon Approximation and Efficient Optimal Algorithms for Stochastic Shortest Path · NeurIPS 2021
Machine learning › Reinforcement learning › markov decision process › infinite-horizon markov decision process
average-reward MDP
0.412020
Model-free Reinforcement Learning in Infinite-horizon Average-reward Markov Decision Processes · ICML 2020

Methods — techniques the papers use, named apart from their topics

sparse update · 0.5implicit finite-horizon approximation · 0.5upper confidence bound · 0.4thompson sampling · 0.4adversarial multi-armed bandit · 0.4
YearPublicationVenuePosition
2024 A Bayesian Learning Algorithm for Unknown Zero-sum Stochastic Games with an Arbitrary Opponent
abstract
In this paper, we propose Posterior Sampling Reinforcement Learning for Zero-sum Stochastic Games (PSRL-ZSG), the first online learning algorithm that achieves Bayesian regret bound of $\tilde\mathcal{O}(HS\sqrt{AT})$ in the infinite-horizon zero-sum stochastic games with average-reward criterion. Here $H$ is an upper bound on the span of the bias function, $S$ is the number of states, $A$ is the number of joint actions and $T$ is the horizon. We consider the online setting where the opponent can not be controlled and can take any arbitrary time-adaptive history-dependent strategy. Our regret bound improves on the best existing regret bound of $\tilde\mathcal{O}(\sqrt[3]{DS^2AT^2})$ by Wei et al., (2017) under the same assumption and matches the theoretical lower bound in $T$.
Mehdi Jafarnia-Jahromi, Rahul Jain 0002, Ashutosh Nayyar
AISTATS1
2023 Posterior sampling-based online learning for the stochastic shortest path model
abstract
We consider the problem of online reinforcement learning for the Stochastic Shortest Path (SSP) problem modeled as an unknown MDP with an absorbing state. We propose PSRL-SSP, a simple posterior sampling-based reinforcement learning algorithm for the SSP problem. The algorithm operates in epochs. At the beginning of each epoch, a sample is drawn from the posterior distribution on the unknown model dynamics, and the optimal policy with respect to the drawn sample is followed during that epoch. An epoch completes if either the number of visits to the goal state in the current epoch exceeds that of the previous epoch, or the number of visits to any of the state-action pairs is doubled. We establish a Bayesian regret bound of $\tilde{\mathcal{O}}(B_{\ast} S\sqrt{AK})$, where $B_{\ast}$ is an upper bound on the expected cost of the optimal policy, $S$ is the size of the state space, $A$ is the size of the action space, and $K$ is the number of episodes. The algorithm only requires the knowledge of the prior distribution, and has no hyper-parameters to tune. It is the first such posterior sampling algorithm and outperforms numerically previously proposed optimism-based algorithms.
Mehdi Jafarnia-Jahromi, Liyu Chen, Rahul Jain 0002
UAI1
2022 Online Learning for Unknown Partially Observable MDPs
abstract
Solving Partially Observable Markov Decision Processes (POMDPs) is hard. Learning optimal controllers for POMDPs when the model is unknown is harder. Online learning of optimal controllers for unknown POMDPs, which requires efficient learning using regret-minimizing algorithms that effectively tradeoff exploration and exploitation, is even harder, and no solution exists currently. In this paper, we consider infinite-horizon average-cost POMDPs with unknown transition model, though a known observation model. We propose a natural posterior sampling-based reinforcement learning algorithm (PSRL-POMDP) and show that it achieves a regret bound of $O(\log T)$, where $T$ is the time horizon, when the parameter set is finite. In the general case (continuous parameter set), we show that the algorithm achieves $O(T^{2/3})$ regret under two technical assumptions. To the best of our knowledge, this is the first online RL algorithm for POMDPs and has sub-linear regret.
Mehdi Jafarnia-Jahromi, Rahul Jain 0002, Ashutosh Nayyar
AISTATS1
2021 Learning Infinite-horizon Average-reward MDPs with Linear Function Approximation
abstract
We develop several new algorithms for learning Markov Decision Processes in an infinite-horizon average-reward setting with linear function approximation. Using the optimism principle and assuming that the MDP has a linear structure, we first propose a computationally inefficient algorithm with optimal O(\sqrt{T}) regret and another computationally efficient variant with O(T^{3/4}) regret, where T is the number of interactions. Next, taking inspiration from adversarial linear bandits, we develop yet another efficient algorithm with O(\sqrt{T}) regret under a different set of assumptions, improving the best existing result by Hao et al. (2020) with O(T^{2/3}) regret. Moreover, we draw a connection between this algorithm and the Natural Policy Gradient algorithm proposed by Kakade (2002), and show that our analysis improves the sample complexity bound recently given by Agarwal et al. (2020).
Chen-Yu Wei, Mehdi Jafarnia-Jahromi, Rahul Jain 0002
AISTATS2
2021 Implicit Finite-Horizon Approximation and Efficient Optimal Algorithms for Stochastic Shortest Path
abstract
We introduce a generic template for developing regret minimization algorithms in the Stochastic Shortest Path (SSP) model, which achieves minimax optimal regret as long as certain properties are ensured. The key of our analysis is a new technique called implicit finite-horizon approximation, which approximates the SSP model by a finite-horizon counterpart only in the analysis without explicit implementation. Using this template, we develop two new algorithms: the first one is model-free (the first in the literature to our knowledge) and minimax optimal under strictly positive costs; the second one is model-based and minimax optimal even with zero-cost state-action pairs, matching the best existing result from [Tarbouriech et al., 2021b]. Importantly, both algorithms admit highly sparse updates, making them computationally more efficient than all existing algorithms. Moreover, both can be made completely parameter-free.
Liyu Chen, Mehdi Jafarnia-Jahromi, Rahul Jain 0002
NeurIPS2
2020 Model-free Reinforcement Learning in Infinite-horizon Average-reward Markov Decision Processes
abstract
Model-free reinforcement learning is known to be memory and computation efficient and more amendable to large scale problems. In this paper, two model-free algorithms are introduced for learning infinite-horizon average-reward Markov Decision Processes (MDPs). The first algorithm reduces the problem to the discounted-reward version and achieves $\mathcal{O}(T^{2/3})$ regret after $T$ steps, under the minimal assumption of weakly communicating MDPs. To our knowledge, this is the first model-free algorithm for general MDPs in this setting. The second algorithm makes use of recent advances in adaptive algorithms for adversarial multi-armed bandits and improves the regret to $\mathcal{O}(\sqrt{T})$, albeit with a stronger ergodic assumption. This result significantly improves over the $\mathcal{O}(T^{3/4})$ regret achieved by the only existing model-free algorithm by Abbasi-Yadkori et al. (2019) for ergodic MDPs in the infinite-horizon average-reward setting.
Chen-Yu Wei, Mehdi Jafarnia-Jahromi, Hiteshi Sharma, Rahul Jain 0002
ICML2
2019 PPD: Permutation Phase Defense Against Adversarial Examples in Deep Learning
abstract
Deep neural networks have demonstrated cutting edge performance on various tasks including classification. However, it is well known that adversarially designed imperceptible perturbation of the input can mislead advanced classifiers. In this paper, Permutation Phase Defense (PPD), is proposed as a novel method to resist adversarial attacks. PPD combines random permutation of the image with phase component of its Fourier transform. The basic idea behind this approach is to turn adversarial defense problems analogously into symmetric cryptography, which relies solely on safekeeping of the keys for security. In PPD, safe keeping of the selected permutation ensures effectiveness against adversarial attacks. Testing PPD on MNIST and CIFAR-10 datasets yielded state-of-the-art robustness against the most powerful adversarial attacks currently available.
Mehdi Jafarnia-Jahromi, Tasmin Chowdhury, Hsin-Tai Wu, Sayandev Mukherjee
ICMLA1
2019 Approximate Relative Value Learning for Average-reward Continuous State MDPs
Hiteshi Sharma, Mehdi Jafarnia-Jahromi, Rahul Jain 0002
UAI2