A State Representation for Diminishing Rewards

A common setting in multitask reinforcement learning (RL) demands that an agent rapidly adapt to various stationary reward functions randomly sampled from a fixed distribution. In such situations, the successor representation (SR) is a popular framework which supports rapid policy evaluation by decoupling a policy's expected discounted, cumulative state occupancies from a specific reward function. However, in the natural world, sequential tasks are rarely independent, and instead reflect shifting priorities based on the availability and subjective perception of rewarding stimuli. Reflecting this disjunction, in this paper we study the phenomenon of diminishing marginal utility and introduce a novel state representation, the $\lambda$ representation ($\lambda$R) which, surprisingly, is required for policy evaluation in this setting and which generalizes the SR as well as several other state representations from the literature. We establish the $\lambda$R's formal properties and examine its normative advantages in the context of machine learning, as well as its usefulness for studying natural behaviors, particularly foraging.

Paper

Similar papers

Peer review

Reviewer GnnZ5/10 · confidence 3/52023-07-02

Summary

The authors study a problem of DMU in RL, e.g. if visiting the same state may result in a smaller reward. They introduce a \lambda R representation that considers a particular form of diminishing rewards and provides convergence results. They extend it to continuous domains and perform preliminary experimental studies.

Strengths

- Typically, such problems having non-markovian rewards are hard, but by having assumptions of \lamda rewards representation, the authors are able to recover some near-optimality convergence results. - Extension to the continuous domain

Weaknesses

I think the assumption of decay with \lambda is quite strong (very restrictive), and it's hard to imagine practical problems satisfying this. E.g. Authors have motivated the problem with examples of ice cream, finance and neuroscience, but it is hard(impossible) that a decay of visiting again is exactly a factor of 0<\lambda<1. Arguably a more interesting setting to consider would be: - \lambda R for is a lower or upper bound on the true rewards that one may earn after visiting the states multiple times, i.e., the reward will decay at least by that amount or at max by a certain amount if visited again. - I think another interesting setting will be if \lambda changes with state visitation frequency and thus needs to be estimated. Both these settings will consider quite a large part of DMU-type problems. Lemma 3.1 is perhaps not the right way to impose Impossibility/hardness results, and the proof needs to be rigorous. I think it can be interpreted in multiple ways, and also, the proof considers a specific format in which the reward is written and a specific definition of the value function. For the lemma statement to be true, the proof shall eradicate all the ways in which value function and rewards can be defined. For e.g. consider a reward function $r(s) = F(\tau \cup s) - F(\tau)$, where $F(\tau)$ is a general set function, $\tau$ is the agent's trajectory. $F(\tau)$ denotes the cumulative value earned for a trajectory $\tau$ as per the reward function (eqn 3.1). This can be written as Bellman recursion with $r(s_i) = F(s_{0:i}) - F(s_{0:i-1})$ (Imagine $\tau = \{s_0, s_1, s_2 ..... \}$). I would suggest not having it as a lemma but just in the text mentioning it is difficult to solve it with the Bellman equation directly. Experiments: I think authors should consider more diverse reward functions, e.g., coverage functions, experiment design, and informative path planning objectives. For e.g., consider a camera-type measurement model, and if we observe the same region again, I will not get much information (there can be some noise model to establish \lamda type of decay). Similarly, in experiment design/ informative planning objectives: Information gained on the nearby states will decrease if I gather/collect measurements at my current location.

Questions

Can you mention the assumptions behind Proposition 4.1? It is counterintuitive for me. $\lambda=0$ (FR) should be a much harder problem as compared to $\lamda=1$ (SR). However, error bound is better for $\lambda=0$? I would suggest authors add an algorithm block for a high-level understanding of the algorithm. I did not fully understand the experiments. - Are your environments deterministic? - In fig 5.2 b), why does \lambda =1 perform well? I would expect it to stay at one of the blue cells forever since it assumes that it gets the same reward always, and on the plot, it shall have an exponential decay. - Is there an assumption of finite state-action spaces, or can you extend it to continuous dynamics as well? Typos: line 268 the the, 156 when when,

Rating

5: Borderline accept: Technically solid paper where reasons to accept outweigh reasons to reject, e.g., limited evaluation. Please use sparingly.

Confidence

3: You are fairly confident in your assessment. It is possible that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work. Math/other details were not carefully checked.

Soundness

3 good

Presentation

3 good

Contribution

2 fair

Limitations

no societal negative impact

Reviewer 2Ebj6/10 · confidence 3/52023-07-05

Summary

This paper studies the phenomenon of diminishing marginal utility in RL, where the a state-based reward r(s) decays as the agent visits it more often. To solve this problem setting, this work introduced a novel state representation, named the λ representation (λR). The author showed that we fail to define a Bellman equation in the λR setting and introduced a recursive relationship.

Strengths

- The probelm setting of DMU in RL is interesting. - The writing is clear and easy to follow. - The paper provides theoretical analysis of the λ representations.

Weaknesses

- Typos and grammar error: line 156 (when when ...), line 156 (Instead of keeping ...), line 177 (The mechanism) - Missing baselines: the current experiments do not compare to other SOTA exploration baselines with intrinsic rewards. So it's hard to evaluate the effectiveness of the proposed methods. - Experiments for continuous control is only conducted on Halfcheetah, a toy example that most RL baselines can easily solve. Experiments on more complex tasks, i.e., humanoid-run, hopper-hop, acrobot-swingup, fish-swim, from dm_control would be more persuasive.

Questions

- Since exploration is the natural setting for λR, why there is no comparisons to different exploration baselines, i.e., RND, state entropy, pseudo-count ...? - I am curious of how will the λR perform in more complex pixel-based tasks.

Rating

6: Weak Accept: Technically solid, moderate-to-high impact paper, with no major concerns with respect to evaluation, resources, reproducibility, ethical considerations.

Confidence

3: You are fairly confident in your assessment. It is possible that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work. Math/other details were not carefully checked.

Soundness

3 good

Presentation

3 good

Contribution

2 fair

Limitations

- The application of the proposed method seems to be narrow, and there is a lack of comparisons to SOTA exploration baselines.

Reviewer jk487/10 · confidence 3/52023-07-06

Summary

The paper considers the case of non-Markov rewards, where rewards at states decay over time (motivated by the diminishing marginal utility phenomenon). While successor representations (SR) have been used in standard MDPs, the decaying of rewards means that they cannot be used here, and so the paper develops a "successor-like" representation that can handle this case, which generalises SR and similar other representations. Theoretical results ensure that these representations can be learned using a Bellman backup (as in the case of SR), which can be used to evaluate and improve policies. Further extensions to the function approximation case (where features are required) are also demonstrated, and bounds related to composition (similar to the successor features bound) are presented. Results in tabular and pixel-based domains showcase how the representation can be used with model-free RL algorithms to solve this class of tasks, where taking into account the decaying rewards (instead of assuming rewards are stationary) is shown to be beneficial.

Strengths

The paper considers an interesting extension to the typical MDPs considered in reinforcement learning. The motivation regarding natural behaviours is an interesting one, but I think on a more fundamental level, any approach that attacks decision problems in RL that aren't Markovian in some way is of interest to the field. This paper packs a lot into it - it develops many different ideas, ranging from the $\lambda$ representation in the tabular case to the function approximation "feature" case. It also shows how these can be used to evaluate and learn policies, compose existing value functions, and be applied in a deep RL context. Aside from the extensive theory presented, the paper also does a good job of relating the approach to existing work in successor representations and demonstrating under what conditions that approach here simplifies to existing work. This helps to unify these otherwise disparate methods.

Weaknesses

As presented, the experiments demonstrate that the method can be applied in both tabular and function approximation settings, that using the right value for $\lambda$ is necessary to achieve the best performance, and that composition can be achieved just as in the SR/SF case. While these experiments confirm the utility of the representation and go hand in hand with the theory, it would have been nice to see the performance of the approach compared to different baselines. This could answer questions like how it compares to SR/FR/SM under a variety of conditions: when $\lambda < 1$ or when $\lambda =1$ or when states are only ever visited once (because it's continuous/very high dimensional). The pixel domain was helpful to showcase that the approach can be applied in high-dimensional domains like this, but it would have been nice to see a more "compositional" domain as has been demonstrated in previous work to investigate composition (e.g. the domains where the agent collects objects of different colours/shapes with different priorities) [1,2,3]. I realise space is an issue, but it would have been helpful to see a very brief discussion on related work that focuses on non-Markov rewards (outside of the context of successor representations). e.g. [4] Minor: Some of the drawbacks of composition with SF are also inherited by this approach. For instance, the bound in Theorem 5.1 is pretty loose (see [2]), but I understand the point is to show it's the same bound with an extra term that is positive only when the $\lambda$ estimates are wrong, so it's not very important While I found the motivation for considering these forms of non-Markovian rewards interesting, it may be the case that many RL practitioners who are not interested in modelling animal/real-world behaviour will not adopt this since for many practical applications (or popular benchmarks like Atari or MuJoCo), such a formulation is not natural. Some of the contractions used made it hard to parse sentences (e.g. L74). I'd recommend replacing these with the full words. [1] Barreto, André, et al. "Fast reinforcement learning with generalised policy updates." Proceedings of the National Academy of Sciences 117.48 (2020): 30079-30087. [2] Nangue Tasse, Geraud, et al. "Generalisation in lifelong reinforcement learning through logical composition." International Conference on Learning Representations. 2022. [3] Alver, Safa, and Doina Precup. "Constructing a Good Behavior Basis for Transfer using Generalised Policy Updates." International Conference on Learning Representations. 2022. [4] Gaon, Maor, and Ronen Brafman. "Reinforcement learning with non-markovian rewards." Proceedings of the AAAI conference on artificial intelligence. Vol. 34. No. 04. 2020.

Questions

On line 291, it states that there is not yet a measure-theoretic version of the representation. Having said that, while there may not be strict theoretical justification, would there be any issue in practice with applying this approach to continuous state/action domains? I'm slightly confused about the composition in the Four Rooms domain. There are four base policies but three goals. Is each policy to reach the centre of each room (and in three of the rooms, a goal exists)?

Rating

7: Accept: Technically solid paper, with high impact on at least one sub-area, or moderate-to-high impact on more than one areas, with good-to-excellent evaluation, resources, reproducibility, and no unaddressed ethical considerations.

Confidence

3: You are fairly confident in your assessment. It is possible that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work. Math/other details were not carefully checked.

Soundness

4 excellent

Presentation

4 excellent

Contribution

3 good

Limitations

The limitations are well described in Section 7, including many that I would not have even considered.

Reviewer 7pTK6/10 · confidence 4/52023-07-12

Summary

The authors explore diminishing marginal utility in the context of reinforcement learning. Specifically, they study reinforcement learning when reward obtained at a state diminishes with each visit to that state (following a particular mathematical form). The authors show that, under such diminishing rewards, agent policies can be evaluated by using a novel state representation introduced in the paper. This representation generalises earlier state representations in the literature, such as the successor representation.

Strengths

Originally: The concept of diminishing rewards is well known and explored in other fields. The paper presents new results within the context of reinforcement learning. Quality: The authors provide a thoughtful and useful analysis under the assumptions they make. This includes the discussion of how the proposed representation relates to existing state representations in RL. The recursion they have identified is interesting and potentially useful. The paper is well structured. The writing is clear.

Weaknesses

The problem is not strongly motivated: why is studying diminishing marginal utility within the context of RL, with the specific assumptions made in this paper, important? Reasoning that goes beyond "people have diminishing marginal utility" would be useful. Why and how is this concept, in the form studied in the paper, useful for reinforcement learning agents? The writing should clearly distinguish between two cases: (1) an objective quantity in the environment (e.g., available food) gets smaller, (2) the subjective value attached by the agent to an outside quantity gets smaller (e.g., the second ice cream cone does not taste as good). In Section 6, the authors write that naturalistic environments often exhibit diminishing reward and give the example of foraging with diminishing food patches. I see a discrepancy between this example and the problem addressed in the rest of the paper. In the foraging example, the available food decreases because other agents consume it, and the amount of reduction should generally be a function of the time that passed between visits to the site. In contrast, the core assumption in the paper is that the reduction in reward in a given state is a function of how many times the agent itself visited that particular state, regardless of how much time passed since its last visit to that state. Additional comments: Forward-backward representation should be explained in Section 2 (Preliminaries).

Questions

I do not have any particular questions but, as I noted above, I do not find the paper particularly well-motivated in its present form. Why and how is diminishing marginal utility, in the form studied in the paper, useful for reinforcement learning agents? I have read the rebuttal and thank the authors for their additional thoughts on the subject.

Rating

6: Weak Accept: Technically solid, moderate-to-high impact paper, with no major concerns with respect to evaluation, resources, reproducibility, ethical considerations.

Confidence

4: You are confident in your assessment, but not absolutely certain. It is unlikely, but not impossible, that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work.

Soundness

3 good

Presentation

3 good

Contribution

3 good

Limitations

Yes, the limitations have been discussed.

Reviewer jk482023-08-12

Thank you for the response. While there are still some fair questions about the motivation for the setting considered, and (as others have pointed out) many ways that the paper can be extended to incorporate other kinds of DMU-type problems, I think the paper packs a lot into it as it currently stands and am happy to stand by my original rating.

Authorsrebuttal2023-08-16

Thank you!

We appreciate your feedback, and thank you for the time spent reviewing the paper! Please let us know if any additional questions arise.

Reviewer GnnZ2023-08-13

Works that solve the same problem and a few follow up questions

Thank you for the response, I would like to bring the authors' attention towards non-markovian rewards in RL, which deals with a similar problem. Submodularity is a known property of a set function and is an equivalent characterization of diminishing returns property. A good read for this topic could be Krause A. and Golovin D., "Submodular Function Maximization". Chekuri and Pal, "A Recursive Greedy Algorithm for Walks in Directed Graphs" study planning on graphs providing both theoretical upper and lower bounds, and there are works which deal with planning under tabular MDP's under submodularity, Wang, et al. "Planning with Submodular Objective Functions" and, in fact also in the reinforcement learning setting with scalable policy gradient method, Prajapat, et al. Submodular reinforcement learning. There is also very related work which deals with convex RL where the objective is defined over state visitation distribution induced by the policy, Zahavy et al "Reward is enough for convex MDPs", general utilities RL, Kumar et al "Policy Gradient for Reinforcement Learning with General Utilities". A few follow-up questions: - Can you please point me to a specific application and a section/line in the reference paper (or from neurology, economics) that deals with constant decay rate rewards? - Regarding Proposition 4.1, Did I understand correctly that $\lambda=1$ corresponds to the usual RL setting where the rewards are fixed? So $\lambda=0$ shall be a hard problem. In fact, there is quite some research demonstrating that hardness stems from a non-repeatable reward structure; please see Blum et al. Approximation Algorithms for Orienteering and Discounted-Reward TSP. I am not sure why you are getting a faster convergence for $\lambda=0$, while the TSP type of problems are known to be NP-hard. Do you also recover optimality guarantees for $\lambda=0$, or Maybe I missing something? - Regarding Lemma 3.1, sorry, it is difficult to parse for me. Are you proving it by contradiction? If you are confident, you may have it as a proposition for being an independent, interesting result. (feel free to make multiple posts below to explain the proof below) Thanks,

Authorsrebuttal2023-08-14

Response Part 1/2

Regarding Non-Markovian Rewards: Thank you for the references. Indeed, there is a connection to submodular functions, which is partly what motivated our discussion of the $\lambda$ Operator (Section 4.1 and Appendix G). In particular, we note that submodularity is not as well-defined for continuous spaces, in that definitions which are equivalent in the finite case are no longer so in the infinite case. We appreciate the pointer to Wang, et al. "Planning with Submodular Objective Functions." It is certainly related, though differs in that it focuses solely on planning and not the specific form of diminishment we study. Similarly, we appreciate the pointer to Prajapat, et al. “Submodular Reinforcement Learning,” though we note that as it was only uploaded to arXiv a few weeks ago (after the submission date for this conference) we don’t think it affects the novelty of our work. However, we would be more than happy to include a discussion of this work as well. With respect to other examples of non-stationary RL, such as convex MDPs, we will absolutely include a discussion, as we have noted to Reviewer jk48. Something that we would like to emphasize is that, separate from DMU itself, we view the introduction of the $\lambda$R (and by extension, the $\lambda$F and $\lambda$O) as a useful contribution in that it unifies the SR, FR, and FB representations from the broader RL literature. Q1: Here is a list of several references with exponentially decaying utility (with the equivalent of constant $\lambda$), either in discrete or continuous time [1, 2, 3, 4, 5, 6]. One interesting note is that it is common in economics/decision theory papers to use one (constant) $\lambda$ for positive outcomes and a different $\lambda$ for negative outcomes. All of the rewards we study in our experiments are positive, so this specific situation does not directly apply in our settings, but extending our results to this setting would be an interesting direction for the future! Q2: Indeed, this is a subtle but important point. Proposition 4.1 is a convergence rate for policy evaluation under DMU (which includes the stationary case and the $\lambda=0$ case), not optimal control. Indeed, [7], which introduces the FR, describes the connection to TSP problems, and introduces a method “FR Planning” which finds provably shortest paths to a given goal given a fixed set of policies. In the case that there are multiple goals, this devolves to a TSP and is indeed NP-hard. However, computing the FR/$\lambda$R is more like computing the distance of a fixed route (policy) rather than finding the optimal policy. Thank you for bringing this up—we will emphasize it in the paper. [1] Pine, A., Seymour, B., Roiser, J. P., Bossaerts, P., Friston, K. J., Curran, H. V., & Dolan, R. J. (2009). Encoding of marginal utility across time in the human brain. Journal of Neuroscience, 29(30), 9575-9581. Figure 3b. [2] Wispinski, N., Butcher, A., Mathewson, K. W., Chapman, C. S., Botvinick, M., & Pilarski, P. M. (2022). Adaptive patch foraging in deep reinforcement learning agents. Transactions on Machine Learning Research. Figure 3c. [3] Chateauneuf, Alain, and Michèle Cohen. "Risk seeking with diminishing marginal utility in a non-expected utility model." Journal of Risk and Uncertainty 9 (1994): 77-91. Corollary 4. [4] Pratt, J. W. (1978). Risk aversion in the small and in the large. In Uncertainty in economics (pp. 59-79). Academic Press. Section 4. [5] Pine, A., Shiner, T., Seymour, B., & Dolan, R. J. (2010). Dopamine, time, and impulsivity in humans. Journal of Neuroscience, 30(26), 8888-8896. Eq. 2. [6] Rachlin, H. (1992). Diminishing marginal value as delay discounting. Journal of the Experimental Analysis of Behavior, 57(3), 407-415. Eq. 1. [7] Moskovitz, T., Wilson, S. R., & Sahani, M. (2021, October). A First-Occupancy Representation for Reinforcement Learning. In International Conference on Learning Representations.

Authorsrebuttal2023-08-14

Response Part 2/2

Q3: **Definition:** A Bellman operator is a contractive operator $\\mathbb{R}^{|\\mathcal{S}|} \\rightarrow \\mathbb{R}^{|\\mathcal{S}|}$ that depends solely on $\\bar{r}$, one-step expectations under $p^\\pi$, and learning hyperparameters (in our case $\\gamma$ and $\\lambda$). For simplicity, below we assume the state-space $\\mathcal{S}$ is finite, and we let $\\mathbf{\\bar{r}}$ and $\\mathbf{V^\\pi}$ denote vectors of $\\bar r(s’)$ and $V^\\pi(s’)$, respectively, over $s’ \\in \\mathcal{S}$. Let $\\mathbf{v}^\\bot$ denote the orthogonal complement of a vector $\\mathbf{v}$. **Lemma 3.1 (revised):** Assume that $|\\mathcal{S}|>1$. Then, there does not exist a Bellman operator $T$ with fixed point $V^\\pi$. Proof: Assume for a contradiction that $T$ is a Bellman operator. By the Banach fixed-point theorem, $V^\\pi$ must be the unique fixed point of $T$. Hence, $TV^\\pi$ must take on the following form (see the proof of Lemma 3.1 in Appendix B): for $s \\in \\mathcal{S}$, $$(TV^\\pi)(s)= \\bar r(s) + \\gamma \\mathbb{E}\_{s’ \\sim p^\\pi(\\cdot|s)} V^\\pi(s’) +\\gamma (\\lambda-1) \\bar r(s) \\mathbb{E}\_{s’ \\sim p^\\pi(\\cdot|s)}\\Phi\_\\lambda^\\pi (s’,s).$$ For the assumption to hold, there must exist a function $f$ (possibly depending on one-step expectations under $p^\\pi$) such that, for any $s \\in \\mathcal{S}$, $$\\mathbb{E}\_{s’ \\sim p^\\pi(\\cdot|s)} \\Phi\_\\lambda^\\pi(s’,s) = f(\\mathbf{\\bar r}, \\mathbf{V^\\pi}, \\gamma, \\lambda, s).$$ Now, by definition, $$ V^\\pi(s) = \\sum_{s’ \\in \\mathcal{S}} \\Phi_\\lambda^\\pi(s,s’) \\bar r(s’). $$ $\\mathbf{\\bar{r}}$ is a vector in $\\mathbb{R}^{|\\mathcal{S}|}$, so as long as $|\\mathcal{S}|>1$, $\\mathbf{\\bar{r}}^\\bot$ is non-trivial. Fix any $\\mathbf{\\bar{r}}, \\mathbf{V^\\pi}, s$. Pick a vector $\\mathbf{w} \\in \\mathbf{\\bar{r}}^\\bot \\setminus \\{\\mathbf{0}\\}$ and define $$ \\tilde{\\Phi}\_\\lambda^\\pi(s,s’) := \\Phi\_\\lambda^\\pi(s,s’) + w(s’) $$ for any $s,s’ \\in \\mathcal{S}$. Note that $$ \\sum\_{s’ \\in \\mathcal{S}} \\tilde{\\Phi}\_\\lambda^\\pi(s,s’) \\bar r(s’) = \\sum\_{s’ \\in \\mathcal{S}} \\Phi\_\\lambda^\\pi(s,s’) \\bar r(s’) + \\sum\_{s’ \\in \\mathcal{S}} w(s’)\\bar r(s’) = V^\\pi(s), $$ as $\\mathbf{w} \\perp \\mathbf{\\bar{r}}$. However, $$ \\mathbb{E}\_{s’ \\sim p^\\pi(\\cdot|s)} \\tilde{\\Phi}\_\\lambda^\\pi(s’,s) = \\sum\_{s’ \\in \\mathcal{S}} p^\\pi(s’|s) \\Phi\_\\lambda^\\pi(s’,s) + \\sum\_{s’ \\in \\mathcal{S}} p^\\pi(s’|s) w(s). $$ The final term evaluates to $w(s)$. Because $\\mathbf{w} \\neq \\mathbf{0}$, there must exist some $s$ such that $w(s) \\neq 0$. For this $s$, we have a single input $(\\mathbf{\\bar r}, \\mathbf{V^\\pi}, \\gamma, \\lambda, s)$ to $f$ that corresponds to two distinct outputs: $\\mathbb{E}\_{s’ \\sim p^\\pi(\\cdot|s)} \\Phi\_\\lambda^\\pi(s’,s)$ and $\\mathbb{E}\_{s’ \\sim p^\\pi(\\cdot|s)} \\tilde{\\Phi}\_\\lambda^\\pi(s’,s)$. Hence, $f$ is a one-to-many mapping: for fixed input, there is more than one output. Therefore, $f$ is not a function, yielding a contradiction.

Authorsrebuttal2023-08-16

Brief Addendum

Thank you once again for your feedback! We would just like to briefly add to our response regarding the concurrent work on submodular RL. While it is certainly true that diminishing rewards can be characterized as a type of submodular reward, we believe the contributions of these works are complementary. Prajapat, et al. focus on this general class of problems and introduce a policy-based, REINFORCE-like method which, while targeted at a general class of problems is on-policy and may suffer from high variance. In contrast, we focus on a particular sub-class of problems especially relevant to natural behavior and introduce a family of approaches which exploit this reward structure, are value-based (and which can be used to modify the critic in policy-based methods), and are compatible with off-policy learning. In short, while this concurrent work is focused on a more general setting, we consider a specific (but we believe especially interesting) instance of this problem class to derive a solution that directly exploits this structure. In the process, we show that it's possible unify several state representations from the literature into the $\lambda$R. We think both perspectives have value. Thank you once again for your time and consideration.

Reviewer GnnZ2023-08-18

Response to Authors

Thank you for the proof and the clarification. I am still not able to convince myself of how do you get around the hardness of the problem. If I understood the reply, the authors are saying that they are not converging to the optimal policy but some state representation? Does it correspond to local convergence, but that defeats the Bellman type operator's purpose, doesn't it? But theorem 5.1 is, in fact, Guaranteeing global governance with some additive error. One possible explanation I can think of is if line 231 has $\lambda=0$, then the additive error for $\gamma~1$ is actually larger than the $Q$ function and thus, the bound is not meaningful anymore. A minor thing: eqn 2.4 shall have $s_t =s$

Authorsrebuttal2023-08-19

Response to Reviewer

Thank you for your response! Indeed, we are not attempting to claim a fast convergence rate to the optimal policy in a DMU setting. - Proposition 4.1 gives the convergence rate for learning the lambda representation associated with a *fixed* policy $\pi$. This representation can then be used to evaluate $\pi$ (i.e. compute $V^\pi$) for any DMU-type task with non-Markov reward scheme defined in the same environment. - Proposition 5.1 is a generalization of the GPI result of Barreto et al. (2019) [1]. It is not about convergence, and shows improvement over a set of fixed policies rather than optimality. GPI is a mechanism for composing *previously learned* policies, that are each near-optimal for a different task $M_j$. The result then gives a performance bound for the composed policy on a new task $M$, as a function of how close to optimal the old policies are for their respective tasks, as well as how close the $\lambda$ value used to evaluate these policies is from the true decay rate. None of these policies are necessarily optimal for the new task $M$. For additional work on GPI, we refer to [2, 3, 4, 5, 6]. We provide proofs for both results in Appendix B (as well as several additional results), and would be happy to discuss any details that are unclear. Re: Eq. 2.4 -- We actually explain this notation (using $\mathbb E[\cdot|s_t]$ as shorthand for $\mathbb E[\cdot|s_t=s]$) in Section 2, Lines 66-68, but we agree it could be confusing! We can adjust it. Thank you once again for the helpful discussion, and please let us know if anything remains unclear! [1] Barreto, A., Borsa, D., Quan, J., Schaul, T., Silver, D., Hessel, M., ... & Munos, R. (2018, July). Transfer in deep reinforcement learning using successor features and generalised policy improvement. In International Conference on Machine Learning (pp. 501-510). PMLR. [2] Barreto, A., Dabney, W., Munos, R., Hunt, J. J., Schaul, T., van Hasselt, H. P., & Silver, D. (2017). Successor features for transfer in reinforcement learning. Advances in neural information processing systems, 30. [3] Thakoor, S., Rowland, M., Borsa, D., Dabney, W., Munos, R., & Barreto, A. (2022, June). Generalised policy improvement with geometric policy composition. In International Conference on Machine Learning (pp. 21272-21307). PMLR. [4] Barreto, A., Hou, S., Borsa, D., Silver, D., & Precup, D. (2020). Fast reinforcement learning with generalized policy updates. Proceedings of the National Academy of Sciences, 117(48), 30079-30087. [5] Borsa, D., Barreto, A., Quan, J., Mankowitz, D. J., van Hasselt, H., Munos, R., ... & Schaul, T. (2018, September). Universal Successor Features Approximators. In International Conference on Learning Representations. [6] Nemecek, M., & Parr, R. (2021, July). Policy caches with successor features. In International Conference on Machine Learning (pp. 8025-8033). PMLR.

Reviewer GnnZ2023-08-20

Response to Authors

Thanks for the reply. I have no more pending questions.

Authorsrebuttal2023-08-21

Thank you!

Thank you very much for your engagement and feedback--we will certainly integrate it into the paper!

Reviewer 2Ebj2023-08-18

Thanks for the response. I have updated the score.

Program Chairsdecision2023-09-21

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC