Submodular Reinforcement Learning
Manish Prajapat, Mojmir Mutny, Melanie N. Zeilinger, Andreas Krause
Abstract
In reinforcement learning (RL), rewards of states are typically considered additive, and following the Markov assumption, they are of states visited previously. In many important applications, such as coverage control, experiment design and informative path planning, rewards naturally have diminishing returns, i.e., their value decreases in light of similar states visited previously. To tackle this, we propose (SubRL), a paradigm which seeks to optimize more general, non-additive (and history-dependent) rewards modelled via submodular set functions which capture diminishing returns. Unfortunately, in general, even in tabular settings, we show that the resulting optimization problem is hard to approximate. On the other hand, motivated by the success of greedy algorithms in classical submodular optimization, we propose SubPO, a simple policy gradient-based algorithm for SubRL that handles non-additive rewards by greedily maximizing marginal gains. Indeed, under some assumptions on the underlying Markov Decision Process (MDP), SubPO recovers optimal constant factor approximations of submodular bandits. Moreover, we derive a natural policy gradient approach for locally optimizing SubRL instances even in large state- and action- spaces. We showcase the versatility of our approach by applying SubPO to several applications, such as biodiversity monitoring, Bayesian experiment design, informative path planning, and coverage maximization. Our results demonstrate sample efficiency, as well as scalability to high-dimensional state-action spaces.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers15
- Flow Density Control: Generative Optimization Beyond Entropy-Regularized Fine-TuningRiccardo De Santi, Marin Vlastelica, Ya-Ping Hsieh, Zebang Shen et al.NeurIPS 2025 · 17 citations
- Global Reinforcement Learning : Beyond Linear and Convex Rewards via Submodular Semi-gradient MethodsRiccardo De Santi, Manish Prajapat, Andreas KrauseICML 2024 · 14 citations
- Global Rewards in Restless Multi-Armed BanditsNaveen Raman, Zheyuan Shi, Fei FangNeurIPS 2024 · 10 citations
- Verifier-Constrained Flow Expansion for Discovery Beyond the DataRiccardo De Santi, Kimon Protopapas, Ya-Ping Hsieh, Andreas KrauseICLR 2026 · 6 citations
- On the Global Optimality of Policy Gradient Methods in General Utility Reinforcement LearningAnas Barakat, Souradip Chakraborty, Peihong Yu, Pratap Tokekar et al.NeurIPS 2025 · 6 citations
Builds on6
- On the Expressivity of Markov RewardDavid Abel, Will Dabney, Anna Harutyunyan, Mark K. Ho et al.NeurIPS 2021 · 107 citations
- Reward is enough for convex MDPsTom Zahavy, Brendan O'Donoghue, Guillaume Desjardins, Satinder SinghNeurIPS 2021 · 96 citations
- On the Theory of Reinforcement Learning with Once-per-Episode FeedbackNiladri S. Chatterji, Aldo Pacchiano, Peter L. Bartlett, Michael I. JordanNeurIPS 2021 · 37 citations
- Challenging Common Assumptions in Convex Reinforcement LearningMirco Mutti, Riccardo De Santi, Piersilvio De Bartolomeis, Marcello RestelliNeurIPS 2022 · 31 citations
- Information Directed Reward Learning for Reinforcement LearningDavid Lindner, Matteo Turchetta, Sebastian Tschiatschek, Kamil Ciosek et al.NeurIPS 2021 · 27 citations
Related papers
- Multi-Agent Reinforcement Learning with Submodular RewardWenjing Chen, Chengyuan Qian, Shuo Xing, Yi Zhou et al.ICML 2026 · 2 citations
- Learning to Make Decisions via Submodular RegularizationAyya Alieva, Aiden Aceves, Jialin Song, Stephen Mayo et al.ICLR 2021 · 5 citations
- Online Nonsubmodular Minimization with Delayed Costs: From Full Information to Bandit FeedbackTianyi Lin, Aldo Pacchiano, Yaodong Yu, Michael I. JordanICML 2022 · 1 citation
- Near-Optimal Multi-Agent Learning for Safe Coverage ControlManish Prajapat, Matteo Turchetta, Melanie N. Zeilinger, Andreas KrauseNeurIPS 2022 · 23 citations
- Submodular Maximization under Supermodular Constraint: Greedy GuaranteesAjitesh Srivastava, Shanghua TengKDD 2026
